package org.postgresql.jdbc;
import org.postgresql.core.Utils;
import org.postgresql.util.GT;
import org.postgresql.util.PSQLException;
import org.postgresql.util.PSQLState;
import java.sql.SQLException;
import java.sql.Savepoint;
public class PSQLSavepoint implements Savepoint {
private boolean isValid;
private final boolean isNamed;
private int id;
private String name;
public PSQLSavepoint(int id) {
this.isValid = true;
this.isNamed = false;
this.id = id;
}
public PSQLSavepoint(String name) {
this.isValid = true;
this.isNamed = true;
this.name = name;
}
@Override
public int getSavepointId() throws SQLException {
if (!isValid) {
throw new PSQLException(GT.tr("Cannot reference a savepoint after it has been released."),
PSQLState.INVALID_SAVEPOINT_SPECIFICATION);
}
if (isNamed) {
throw new PSQLException(GT.tr("Cannot retrieve the id of a named savepoint."),
PSQLState.WRONG_OBJECT_TYPE);
}
return id;
}
@Override
public String getSavepointName() throws SQLException {
if (!isValid) {
throw new PSQLException(GT.tr("Cannot reference a savepoint after it has been released."),
PSQLState.INVALID_SAVEPOINT_SPECIFICATION);
}
if (!isNamed) {
throw new PSQLException(GT.tr("Cannot retrieve the name of an unnamed savepoint."),
PSQLState.WRONG_OBJECT_TYPE);
}
return name;
}
public void invalidate() {
isValid = false;
}
public String getPGName() throws SQLException {
if (!isValid) {
throw new PSQLException(GT.tr("Cannot reference a savepoint after it has been released."),
PSQLState.INVALID_SAVEPOINT_SPECIFICATION);
}
if (isNamed) {
return Utils.escapeIdentifier(null, name).toString();
}
return "JDBC_SAVEPOINT_" + id;
}
}