How to use

getGeneratedKey

method

in

org.apache.shardingsphere.core.routing.SQLRouteResult

private Optional<GeneratedKey> getGeneratedKey() {
  if (null != routeResult && routeResult.getSqlStatement() instanceof InsertStatement) {
    return Optional.fromNullable(routeResult.getGeneratedKey());
  }
  return Optional.absent();
}

  private Optional<GeneratedKey> getGeneratedKey() {
    if (null != routeResult && routeResult.getSqlStatement() instanceof InsertStatement) {
      return Optional.fromNullable(routeResult.getGeneratedKey());
    }
    return Optional.absent();
  }
}

private void setGeneratedKeys(final SQLRouteResult sqlRouteResult, final GeneratedKey generatedKey) {
  generatedKeys.addAll(generatedKey.getGeneratedKeys());
  sqlRouteResult.getGeneratedKey().getGeneratedKeys().clear();
  sqlRouteResult.getGeneratedKey().getGeneratedKeys().addAll(generatedKeys);
}

@Override
public ResultSet getGeneratedKeys() throws SQLException {
  Optional<GeneratedKey> generatedKey = getGeneratedKey();
  if (returnGeneratedKeys && generatedKey.isPresent()) {
    return new GeneratedKeysResultSet(routeResult.getGeneratedKey().getGeneratedKeys().iterator(), generatedKey.get().getColumn().getName(), this);
  }
  if (1 == getRoutedStatements().size()) {
    return getRoutedStatements().iterator().next().getGeneratedKeys();
  }
  return new GeneratedKeysResultSet();
}

@Override
public ResultSet getGeneratedKeys() throws SQLException {
  Optional<GeneratedKey> generatedKey = getGeneratedKey();
  if (preparedStatementExecutor.isReturnGeneratedKeys() && generatedKey.isPresent()) {
    return new GeneratedKeysResultSet(routeResult.getGeneratedKey().getGeneratedKeys().iterator(), generatedKey.get().getColumn().getName(), this);
  }
  if (1 == preparedStatementExecutor.getStatements().size()) {
    return preparedStatementExecutor.getStatements().iterator().next().getGeneratedKeys();
  }
  return new GeneratedKeysResultSet();
}