nl.topicus.jdbc.CloudSpannerSavepoint Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spanner-jdbc Show documentation

JDBC Driver for Google Cloud Spanner

There is a newer version: 1.1.6

package nl.topicus.jdbc;

import java.sql.SQLException;
import java.sql.Savepoint;
import java.util.concurrent.atomic.AtomicInteger;
import nl.topicus.jdbc.shaded.com.google.common.base.Preconditions;
import nl.topicus.jdbc.shaded.com.google.rpc.Code;
import nl.topicus.jdbc.exception.CloudSpannerSQLException;

/**
 * Savepoint implementation for Google Cloud Spanner JDBC driver
 * 
 * @author loite
 *
 */
class CloudSpannerSavepoint implements Savepoint {
  private static final AtomicInteger CURRENT_ID = new AtomicInteger(0);

  private final String name;

  private final Integer id;

  static CloudSpannerSavepoint named(String name) {
    Preconditions.checkNotNull(name);
    return new CloudSpannerSavepoint(name);
  }

  static CloudSpannerSavepoint generated() {
    return new CloudSpannerSavepoint(CURRENT_ID.incrementAndGet());
  }

  private CloudSpannerSavepoint(int id) {
    this.id = id;
    this.name = null;
  }

  private CloudSpannerSavepoint(String name) {
    this.id = null;
    this.name = name;
  }

  private boolean isNamed() {
    return name != null;
  }

  @Override
  public int hashCode() {
    return isNamed() ? name.hashCode() : id.hashCode();
  }

  @Override
  public boolean equals(Object o) {
    if (!(o instanceof CloudSpannerSavepoint))
      return false;
    CloudSpannerSavepoint other = (CloudSpannerSavepoint) o;
    if (this.isNamed())
      return other.isNamed() && this.name.equals(other.name);
    return this.id.equals(other.id);
  }

  @Override
  public int getSavepointId() throws SQLException {
    if (isNamed())
      throw new CloudSpannerSQLException("This is a named savepoint", Code.FAILED_PRECONDITION);
    return id;
  }

  @Override
  public String getSavepointName() throws SQLException {
    if (!isNamed())
      throw new CloudSpannerSQLException("This is not a named savepoint", Code.FAILED_PRECONDITION);
    return name;
  }

}