com.scalar.db.transaction.consensuscommit.TransactionResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scalardb Show documentation
A universal transaction manager that achieves database-agnostic transactions and distributed transactions that span multiple databases
The newest version!
package com.scalar.db.transaction.consensuscommit;

import static com.google.common.base.Preconditions.checkNotNull;

import com.scalar.db.api.Result;
import com.scalar.db.api.TransactionState;
import com.scalar.db.common.AbstractResult;
import com.scalar.db.io.Column;
import com.scalar.db.io.IntColumn;
import com.scalar.db.io.Key;
import com.scalar.db.io.TextColumn;
import java.nio.ByteBuffer;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import javax.annotation.Nullable;
import javax.annotation.concurrent.Immutable;

@Immutable
public class TransactionResult extends AbstractResult {
  private final Result result;

  public TransactionResult(Result result) {
    // assume that all the columns are projected to the result
    this.result = checkNotNull(result);
  }

  /** @deprecated As of release 3.8.0. Will be removed in release 5.0.0 */
  @Deprecated
  @Override
  public Optional getPartitionKey() {
    return result.getPartitionKey();
  }

  /** @deprecated As of release 3.8.0. Will be removed in release 5.0.0 */
  @Deprecated
  @Override
  public Optional getClusteringKey() {
    return result.getClusteringKey();
  }

  @Override
  public boolean isNull(String columnName) {
    return result.isNull(columnName);
  }

  @Override
  public boolean getBoolean(String columnName) {
    return result.getBoolean(columnName);
  }

  @Override
  public int getInt(String columnName) {
    return result.getInt(columnName);
  }

  @Override
  public long getBigInt(String columnName) {
    return result.getBigInt(columnName);
  }

  @Override
  public float getFloat(String columnName) {
    return result.getFloat(columnName);
  }

  @Override
  public double getDouble(String columnName) {
    return result.getDouble(columnName);
  }

  @Nullable
  @Override
  public String getText(String columnName) {
    return result.getText(columnName);
  }

  @Nullable
  @Override
  public ByteBuffer getBlobAsByteBuffer(String columnName) {
    return result.getBlobAsByteBuffer(columnName);
  }

  @Nullable
  @Override
  public byte[] getBlobAsBytes(String columnName) {
    return result.getBlobAsBytes(columnName);
  }

  @Nullable
  @Override
  public Object getAsObject(String columnName) {
    return result.getAsObject(columnName);
  }

  @Override
  public boolean contains(String columnName) {
    return result.contains(columnName);
  }

  @Override
  public Set getContainedColumnNames() {
    return result.getContainedColumnNames();
  }

  @Override
  public Map> getColumns() {
    return result.getColumns();
  }

  @Nullable
  public String getId() {
    return getText(Attribute.ID);
  }

  public TransactionState getState() {
    if (isNull(Attribute.STATE)) {
      // To handle existing databases that do not have transaction metadata, the record is deemed as
      // committed if the state is NULL.
      return TransactionState.COMMITTED;
    }
    return TransactionState.getInstance(getInt(Attribute.STATE));
  }

  public int getVersion() {
    return getInt(Attribute.VERSION);
  }

  public long getPreparedAt() {
    return getBigInt(Attribute.PREPARED_AT);
  }

  public long getCommittedAt() {
    return getBigInt(Attribute.COMMITTED_AT);
  }

  public boolean isCommitted() {
    return getState().equals(TransactionState.COMMITTED);
  }

  public boolean isDeemedAsCommitted() {
    return getId() == null;
  }

  public boolean isMergedResult() {
    return result instanceof MergedResult;
  }

  public boolean hasBeforeImage() {
    // We need to check not only before_id but also before_version to determine if the record has
    // the before image or not since we set before_version to 0 for the prepared record when
    // updating the record deemed as the committed state (cf. PrepareMutationComposer).
    return !getBeforeIdColumn().hasNullValue() || !getBeforeVersionColumn().hasNullValue();
  }

  private TextColumn getBeforeIdColumn() {
    return (TextColumn) result.getColumns().get(Attribute.BEFORE_ID);
  }

  private IntColumn getBeforeVersionColumn() {
    return (IntColumn) result.getColumns().get(Attribute.BEFORE_VERSION);
  }
}