/** * bindShardQuery creates a BoundShardQuery. */ public static BoundShardQuery bindShardQuery(String keyspace, Iterable<String> shards, String query, Map<String, ?> vars) { return bindShardQuery(keyspace, shards, bindQuery(query, vars)); }
/** * bindKeyspaceIdQuery creates a BoundKeyspaceIdQuery. */ public static BoundKeyspaceIdQuery bindKeyspaceIdQuery(String keyspace, Iterable<byte[]> keyspaceIds, String query, Map<String, ?> vars) { return bindKeyspaceIdQuery(keyspace, keyspaceIds, bindQuery(query, vars)); }
public Cursor streamExecuteKeyspaceIds(Context ctx, String query, String keyspace, Iterable<byte[]> keyspaceIds, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { StreamExecuteKeyspaceIdsRequest.Builder requestBuilder = StreamExecuteKeyspaceIdsRequest .newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .addAllKeyspaceIds( Iterables.transform(checkNotNull(keyspaceIds), Proto.BYTE_ARRAY_TO_BYTE_STRING)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new StreamCursor(client.streamExecuteKeyspaceIds(ctx, requestBuilder.build())); }
public Cursor streamExecuteShards(Context ctx, String query, String keyspace, Iterable<String> shards, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { StreamExecuteShardsRequest.Builder requestBuilder = StreamExecuteShardsRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .addAllShards(checkNotNull(shards)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new StreamCursor(client.streamExecuteShards(ctx, requestBuilder.build())); }
public Cursor streamExecuteKeyRanges(Context ctx, String query, String keyspace, Iterable<? extends KeyRange> keyRanges, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { StreamExecuteKeyRangesRequest.Builder requestBuilder = StreamExecuteKeyRangesRequest .newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .addAllKeyRanges(checkNotNull(keyRanges)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new StreamCursor(client.streamExecuteKeyRanges(ctx, requestBuilder.build())); }
public SQLFuture<Cursor> executeEntityIds(Context ctx, String query, String keyspace, String entityColumnName, Map<byte[], ?> entityKeyspaceIds, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { ExecuteEntityIdsRequest.Builder requestBuilder = ExecuteEntityIdsRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .setEntityColumnName(checkNotNull(entityColumnName)) .addAllEntityKeyspaceIds(Iterables .transform(entityKeyspaceIds.entrySet(), Proto.MAP_ENTRY_TO_ENTITY_KEYSPACE_ID)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new SQLFuture<Cursor>( transformAsync( client.executeEntityIds(ctx, requestBuilder.build()), new AsyncFunction<ExecuteEntityIdsResponse, Cursor>() { @Override public ListenableFuture<Cursor> apply(ExecuteEntityIdsResponse response) throws Exception { Proto.checkError(response.getError()); return Futures.<Cursor>immediateFuture(new SimpleCursor(response.getResult())); } }, directExecutor())); }
public Cursor streamExecute(Context ctx, String query, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { StreamExecuteRequest.Builder requestBuilder = StreamExecuteRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspaceShard(keyspace) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new StreamCursor(client.streamExecute(ctx, requestBuilder.build())); }
public SQLFuture<Cursor> executeKeyspaceIds(Context ctx, String query, String keyspace, Iterable<byte[]> keyspaceIds, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { ExecuteKeyspaceIdsRequest.Builder requestBuilder = ExecuteKeyspaceIdsRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .addAllKeyspaceIds( Iterables.transform(checkNotNull(keyspaceIds), Proto.BYTE_ARRAY_TO_BYTE_STRING)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new SQLFuture<Cursor>( transformAsync( client.executeKeyspaceIds(ctx, requestBuilder.build()), new AsyncFunction<ExecuteKeyspaceIdsResponse, Cursor>() { @Override public ListenableFuture<Cursor> apply(ExecuteKeyspaceIdsResponse response) throws Exception { Proto.checkError(response.getError()); return Futures.<Cursor>immediateFuture(new SimpleCursor(response.getResult())); } }, directExecutor())); }
public SQLFuture<Cursor> executeKeyRanges(Context ctx, String query, String keyspace, Iterable<? extends KeyRange> keyRanges, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { ExecuteKeyRangesRequest.Builder requestBuilder = ExecuteKeyRangesRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .addAllKeyRanges(checkNotNull(keyRanges)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new SQLFuture<Cursor>( transformAsync( client.executeKeyRanges(ctx, requestBuilder.build()), new AsyncFunction<ExecuteKeyRangesResponse, Cursor>() { @Override public ListenableFuture<Cursor> apply(ExecuteKeyRangesResponse response) throws Exception { Proto.checkError(response.getError()); return Futures.<Cursor>immediateFuture(new SimpleCursor(response.getResult())); } }, directExecutor())); }
public SQLFuture<Cursor> executeShards(Context ctx, String query, String keyspace, Iterable<String> shards, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { ExecuteShardsRequest.Builder requestBuilder = ExecuteShardsRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspace(checkNotNull(keyspace)) .addAllShards(checkNotNull(shards)) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new SQLFuture<Cursor>( transformAsync( client.executeShards(ctx, requestBuilder.build()), new AsyncFunction<ExecuteShardsResponse, Cursor>() { @Override public ListenableFuture<Cursor> apply(ExecuteShardsResponse response) throws Exception { Proto.checkError(response.getError()); return Futures.<Cursor>immediateFuture(new SimpleCursor(response.getResult())); } }, directExecutor())); }
public SQLFuture<Cursor> execute(Context ctx, String query, @Nullable Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { ExecuteRequest.Builder requestBuilder = ExecuteRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setKeyspaceShard(keyspace) .setTabletType(checkNotNull(tabletType)) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new SQLFuture<Cursor>( transformAsync( client.execute(ctx, requestBuilder.build()), new AsyncFunction<ExecuteResponse, Cursor>() { @Override public ListenableFuture<Cursor> apply(ExecuteResponse response) throws Exception { Proto.checkError(response.getError()); return Futures.<Cursor>immediateFuture(new SimpleCursor(response.getResult())); } }, directExecutor())); }
/** * * @param ctx Context on user and execution deadline if any. * @param query Sql Query to be executed. * @param bindVars Parameters to bind with sql. * @param vtSession Session to be used with the call. * @return * @throws SQLException */ public Cursor streamExecute(Context ctx, String query, @Nullable Map<String, ?> bindVars, VTSession vtSession) throws SQLException { StreamExecuteRequest.Builder requestBuilder = StreamExecuteRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setSession(vtSession.getSession()); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new StreamCursor(client.streamExecute(ctx, requestBuilder.build())); } /**
public SQLFuture<List<SplitQueryResponse.Part>> splitQuery(Context ctx, String keyspace, String query, @Nullable Map<String, ?> bindVars, Iterable<String> splitColumns, int splitCount, int numRowsPerQueryPart, Algorithm algorithm) throws SQLException { SplitQueryRequest.Builder requestBuilder = SplitQueryRequest.newBuilder() .setKeyspace(checkNotNull(keyspace)) .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .addAllSplitColumn(splitColumns) .setSplitCount(splitCount) .setNumRowsPerQueryPart(numRowsPerQueryPart) .setAlgorithm(algorithm); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } return new SQLFuture<List<SplitQueryResponse.Part>>( transformAsync( client.splitQuery(ctx, requestBuilder.build()), new AsyncFunction<SplitQueryResponse, List<SplitQueryResponse.Part>>() { @Override public ListenableFuture<List<SplitQueryResponse.Part>> apply( SplitQueryResponse response) throws Exception { return Futures.<List<SplitQueryResponse.Part>>immediateFuture( response.getSplitsList()); } }, directExecutor())); }
SplitQueryRequest.newBuilder() .setKeyspace(checkNotNull(keyspace)) .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .addAllSplitColumn(splitColumns) .setSplitCount(splitCount)
vtSession.checkCallIsAllowed("execute"); ExecuteRequest.Builder requestBuilder = ExecuteRequest.newBuilder() .setQuery(Proto.bindQuery(checkNotNull(query), bindVars)) .setSession(vtSession.getSession());
checkCallIsAllowed("executeEntityIds"); ExecuteEntityIdsRequest.Builder requestBuilder = ExecuteEntityIdsRequest.newBuilder() .setQuery(Proto.bindQuery(query, bindVars)) .setKeyspace(keyspace) .setEntityColumnName(entityColumnName)
checkCallIsAllowed("executeKeyspaceIds"); ExecuteKeyspaceIdsRequest.Builder requestBuilder = ExecuteKeyspaceIdsRequest.newBuilder() .setQuery(Proto.bindQuery(query, bindVars)) .setKeyspace(keyspace) .addAllKeyspaceIds(Iterables.transform(keyspaceIds, Proto.BYTE_ARRAY_TO_BYTE_STRING))
public synchronized SQLFuture<Cursor> execute(Context ctx, String query, Map<String, ?> bindVars, TabletType tabletType, Query.ExecuteOptions.IncludedFields includedFields) throws SQLException { checkCallIsAllowed("execute"); ExecuteRequest.Builder requestBuilder = ExecuteRequest.newBuilder() .setQuery(Proto.bindQuery(query, bindVars)) .setKeyspaceShard(keyspace) .setTabletType(tabletType) .setSession(session) .setOptions(Query.ExecuteOptions.newBuilder() .setIncludedFields(includedFields)); if (ctx.getCallerId() != null) { requestBuilder.setCallerId(ctx.getCallerId()); } SQLFuture<Cursor> call = new SQLFuture<>( transformAsync( client.execute(ctx, requestBuilder.build()), new AsyncFunction<ExecuteResponse, Cursor>() { @Override public ListenableFuture<Cursor> apply(ExecuteResponse response) throws Exception { setSession(response.getSession()); Proto.checkError(response.getError()); return Futures.<Cursor>immediateFuture(new SimpleCursor(response.getResult())); } }, directExecutor())); lastCall = call; return call; }
checkCallIsAllowed("executeShards"); ExecuteShardsRequest.Builder requestBuilder = ExecuteShardsRequest.newBuilder() .setQuery(Proto.bindQuery(query, bindVars)) .setKeyspace(keyspace) .addAllShards(shards)
checkCallIsAllowed("executeKeyRanges"); ExecuteKeyRangesRequest.Builder requestBuilder = ExecuteKeyRangesRequest.newBuilder() .setQuery(Proto.bindQuery(query, bindVars)) .setKeyspace(keyspace) .addAllKeyRanges(keyRanges)