How to use

SequenceManager

in

org.apache.phoenix.compile

@Override
public void explain(List<String> planSteps) {
  super.explain(planSteps);
  int nSequences = sequenceManager.getSequenceCount();
  planSteps.add("CLIENT RESERVE VALUES FROM " + nSequences + " SEQUENCE" + (nSequences == 1 ? "" : "S"));
}

public StatementContext(PhoenixStatement statement, Scan scan) {
  this(statement, FromCompiler.EMPTY_TABLE_RESOLVER, scan, new SequenceManager(statement));
}

  @Override
  public Expression visit(SequenceValueParseNode node) throws SQLException {
    return context.getSequenceManager().newSequenceReference(node);
  }
}

@SuppressWarnings("unchecked")
@Override
public QueryPlan compilePlan(PhoenixStatement stmt, Sequence.ValueOp seqAction) throws SQLException {
  if(!getUdfParseNodes().isEmpty()) {
    stmt.throwIfUnallowedUserDefinedFunctions(getUdfParseNodes());
  }
  SelectStatement select = SubselectRewriter.flatten(this, stmt.getConnection());
  ColumnResolver resolver = FromCompiler.getResolverForQuery(select, stmt.getConnection());
  select = StatementNormalizer.normalize(select, resolver);
  SelectStatement transformedSelect = SubqueryRewriter.transform(select, resolver, stmt.getConnection());
  if (transformedSelect != select) {
    resolver = FromCompiler.getResolverForQuery(transformedSelect, stmt.getConnection());
    select = StatementNormalizer.normalize(transformedSelect, resolver);
  }
  QueryPlan plan = new QueryCompiler(stmt, select, resolver, Collections.<PDatum>emptyList(), stmt.getConnection().getIteratorFactory(), new SequenceManager(stmt), true, false, null).compile();
  plan.getContext().getSequenceManager().validateSequences(seqAction);
  return plan;
}

Tuple tuple = sequenceManager.getSequenceCount() == 0 ? null :
  sequenceManager.newSequenceTuple(null);
for (Expression constantExpression : constantExpressions) {
  PColumn column = allColumns.get(columnIndexes[nodeIndex]);

  @SuppressWarnings("unchecked")
  @Override
  public MutationPlan compilePlan(PhoenixStatement stmt, Sequence.ValueOp seqAction) throws SQLException {
    if(!getUdfParseNodes().isEmpty()) {
      stmt.throwIfUnallowedUserDefinedFunctions(getUdfParseNodes());
    }
    UpsertCompiler compiler = new UpsertCompiler(stmt, this.getOperation());
    MutationPlan plan = compiler.compile(this);
    plan.getContext().getSequenceManager().validateSequences(seqAction);
    return plan;
  }
}

@Override
public Tuple next() throws SQLException {
  Tuple next = super.next();
  if (next == null) {
    return null;
  }
  next = sequenceManager.newSequenceTuple(next);
  return next;
}

ParseNode numToAllocateNode = node.getNumToAllocateNode();
long numToAllocate = determineNumToAllocate(tableName, numToAllocateNode);
SequenceKey key = new SequenceKey(tenantId, tableName.getSchemaName(), tableName.getTableName(), nSaltBuckets);
SequenceValueExpression expression = sequenceMap.get(key);

long timestamp = scn == null ? HConstants.LATEST_TIMESTAMP : scn;
services.validateSequences(nextSequences, timestamp, srcSequenceValues, sqlExceptions, action);
setSequenceValues(srcSequenceValues, dstSequenceValues, sqlExceptions);

Tuple tuple = sequenceManager.getSequenceCount() == 0 ? null :
  sequenceManager.newSequenceTuple(null);
for (Expression constantExpression : constantExpressions) {
  PColumn column = allColumns.get(columnIndexes[nodeIndex]);

@SuppressWarnings("unchecked")
@Override
public QueryPlan compilePlan(PhoenixStatement stmt, Sequence.ValueOp seqAction) throws SQLException {
  if(!getUdfParseNodes().isEmpty()) {
    stmt.throwIfUnallowedUserDefinedFunctions(getUdfParseNodes());
  }
  SelectStatement select = SubselectRewriter.flatten(this, stmt.getConnection());
  ColumnResolver resolver = FromCompiler.getResolverForQuery(select, stmt.getConnection());
  select = StatementNormalizer.normalize(select, resolver);
  SelectStatement transformedSelect = SubqueryRewriter.transform(select, resolver, stmt.getConnection());
  if (transformedSelect != select) {
    resolver = FromCompiler.getResolverForQuery(transformedSelect, stmt.getConnection());
    select = StatementNormalizer.normalize(transformedSelect, resolver);
  }
  QueryPlan plan = new QueryCompiler(stmt, select, resolver, Collections.<PDatum>emptyList(), stmt.getConnection().getIteratorFactory(), new SequenceManager(stmt), true, false, null).compile();
  plan.getContext().getSequenceManager().validateSequences(seqAction);
  return plan;
}

  @SuppressWarnings("unchecked")
  @Override
  public MutationPlan compilePlan(PhoenixStatement stmt, Sequence.ValueOp seqAction) throws SQLException {
    if(!getUdfParseNodes().isEmpty()) {
      stmt.throwIfUnallowedUserDefinedFunctions(getUdfParseNodes());
    }
    DeleteCompiler compiler = new DeleteCompiler(stmt, this.getOperation());
    MutationPlan plan = compiler.compile(this);
    plan.getContext().getSequenceManager().validateSequences(seqAction);
    return plan;
  }
}

@Override
public Tuple next() throws SQLException {
  Tuple next = super.next();
  if (next == null) {
    return null;
  }
  next = sequenceManager.newSequenceTuple(next);
  return next;
}

ParseNode numToAllocateNode = node.getNumToAllocateNode();
long numToAllocate = determineNumToAllocate(tableName, numToAllocateNode);
SequenceKey key = new SequenceKey(tenantId, tableName.getSchemaName(), tableName.getTableName(), nSaltBuckets);
SequenceValueExpression expression = sequenceMap.get(key);

long timestamp = scn == null ? HConstants.LATEST_TIMESTAMP : scn;
services.validateSequences(nextSequences, timestamp, srcSequenceValues, sqlExceptions, action);
setSequenceValues(srcSequenceValues, dstSequenceValues, sqlExceptions);

@Override
public ExplainPlan getExplainPlan() throws SQLException {
  List<String> planSteps = Lists.newArrayListWithExpectedSize(2);
  if (context.getSequenceManager().getSequenceCount() > 0) {
    planSteps.add("CLIENT RESERVE " + context.getSequenceManager().getSequenceCount() + " SEQUENCES");
  }
  planSteps.add("PUT SINGLE ROW");
  return new ExplainPlan(planSteps);
}

/**
 *  Constructor that lets you override whether or not to collect request level metrics.
 */
public StatementContext(PhoenixStatement statement, boolean collectRequestLevelMetrics) {
  this(statement, FromCompiler.EMPTY_TABLE_RESOLVER, new Scan(), new SequenceManager(statement), collectRequestLevelMetrics);
}

Tuple tuple = sequenceManager.getSequenceCount() == 0 ? null :
  sequenceManager.newSequenceTuple(null);
for (Expression constantExpression : constantExpressions) {
  PColumn column = allColumns.get(columnIndexes[nodeIndex]);

@SuppressWarnings("unchecked")
@Override
public QueryPlan compilePlan(PhoenixStatement stmt, Sequence.ValueOp seqAction) throws SQLException {
  if(!getUdfParseNodes().isEmpty()) {
    stmt.throwIfUnallowedUserDefinedFunctions(getUdfParseNodes());
  }
  SelectStatement select = SubselectRewriter.flatten(this, stmt.getConnection());
  ColumnResolver resolver = FromCompiler.getResolverForQuery(select, stmt.getConnection());
  select = StatementNormalizer.normalize(select, resolver);
  SelectStatement transformedSelect = SubqueryRewriter.transform(select, resolver, stmt.getConnection());
  if (transformedSelect != select) {
    resolver = FromCompiler.getResolverForQuery(transformedSelect, stmt.getConnection());
    select = StatementNormalizer.normalize(transformedSelect, resolver);
  }
  QueryPlan plan = new QueryCompiler(stmt, select, resolver, Collections.<PDatum>emptyList(), stmt.getConnection().getIteratorFactory(), new SequenceManager(stmt), true, false, null).compile();
  plan.getContext().getSequenceManager().validateSequences(seqAction);
  return plan;
}

  @SuppressWarnings("unchecked")
  @Override
  public MutationPlan compilePlan(PhoenixStatement stmt, Sequence.ValueOp seqAction) throws SQLException {
    if(!getUdfParseNodes().isEmpty()) {
      stmt.throwIfUnallowedUserDefinedFunctions(getUdfParseNodes());
    }
    DeleteCompiler compiler = new DeleteCompiler(stmt, this.getOperation());
    MutationPlan plan = compiler.compile(this);
    plan.getContext().getSequenceManager().validateSequences(seqAction);
    return plan;
  }
}