public ResultMessage.Prepared prepare(String queryString, QueryState queryState) { ClientState cState = queryState.getClientState(); return prepare(queryString, cState, cState instanceof ThriftClientState); }
public ThriftClientState(InetSocketAddress remoteAddress) { super(remoteAddress); this.queryState = new QueryState(this); }
public long getTimestamp(QueryState state) { long tstamp = getSpecificOptions().timestamp; return tstamp != Long.MIN_VALUE ? tstamp : state.getTimestamp(); }
private boolean startSessionIfRequested() { if (state().getQueryState().traceNextQuery()) { state().getQueryState().createTracingSession(Collections.EMPTY_MAP); return true; } return false; }
public UntypedResultSet process(final ConsistencyLevel cl, final ConsistencyLevel serialConsistencyLevel, ClientState clientState, final String query, Long writetime, final Object... values) throws RequestExecutionException, RequestValidationException, InvalidRequestException { if (logger.isDebugEnabled()) logger.debug("processing CL={} SERIAL_CL={} query={}", cl, serialConsistencyLevel, query); ParsedStatement.Prepared prepared = QueryProcessor.getStatement(query, clientState); List<ByteBuffer> boundValues = new ArrayList<ByteBuffer>(values.length); for (int i = 0; i < values.length; i++) { Object v = values[i]; AbstractType type = prepared.boundNames.get(i).type; boundValues.add(v instanceof ByteBuffer || v == null ? (ByteBuffer) v : type.decompose(v)); } QueryState queryState = new QueryState(clientState); QueryOptions queryOptions = QueryOptions.forInternalCalls(cl, serialConsistencyLevel, boundValues); ResultMessage result = QueryProcessor.instance.process(query, queryState, queryOptions, System.nanoTime()); writetime = queryState.getTimestamp(); if (result instanceof ResultMessage.Rows) return UntypedResultSet.create(((ResultMessage.Rows) result).result); else return null; }
public Message.Response execute(QueryState state, long queryStartNanoTime) { try { UUID tracingId = null; if (isTracingRequested()) { tracingId = UUIDGen.getTimeUUID(); state.prepareTracingSession(tracingId); } if (state.traceNextQuery()) { state.createTracingSession(getCustomPayload()); Tracing.instance.begin("Preparing CQL3 query", state.getClientAddress(), ImmutableMap.of("query", query)); } Message.Response response = ClientState.getCQLQueryHandler().prepare(query, state, getCustomPayload()); if (tracingId != null) response.setTracingId(tracingId); return response; } catch (Exception e) { JVMStabilityInspector.inspectThrowable(e); return ErrorMessage.fromException(e); } finally { Tracing.instance.stopSession(); } }
public Message.Response execute(QueryState state) { try { UUID tracingId = null; if (isTracingRequested()) { tracingId = UUIDGen.getTimeUUID(); state.prepareTracingSession(tracingId); } if (state.traceNextQuery()) { state.createTracingSession(); Tracing.instance.begin("Preparing CQL3 query", ImmutableMap.of("query", query)); } Message.Response response = state.getClientState().getCQLQueryHandler().prepare(query, state); if (tracingId != null) response.setTracingId(tracingId); return response; } catch (Exception e) { JVMStabilityInspector.inspectThrowable(e); return ErrorMessage.fromException(e); } finally { Tracing.instance.stopSession(); } }
public ResultMessage execute(ClientState state) throws RequestValidationException, RequestExecutionException { return QueryProcessor.process(String.format("SELECT * FROM %s.%s", Auth.AUTH_KS, Auth.USERS_CF), ConsistencyLevel.QUORUM, QueryState.forInternalCalls()); } }
state.prepareTracingSession(tracingId); if (state.traceNextQuery()) state.createTracingSession();
public ByteBuffer trace_next_query() throws TException { UUID sessionId = UUIDGen.getTimeUUID(); state().getQueryState().prepareTracingSession(sessionId); return TimeUUIDType.instance.decompose(sessionId); }
public IAuthenticator.SaslNegotiator getSaslNegotiator(QueryState queryState) { if (saslNegotiator == null) saslNegotiator = DatabaseDescriptor.getAuthenticator().newSaslNegotiator(queryState.getClientAddress()); return saslNegotiator; } }
public Message.Response execute(QueryState state, long queryStartNanoTime) { try { UUID tracingId = null; if (isTracingRequested()) { tracingId = UUIDGen.getTimeUUID(); state.prepareTracingSession(tracingId); } if (state.traceNextQuery()) { state.createTracingSession(getCustomPayload()); Tracing.instance.begin("Preparing CQL3 query", state.getClientAddress(), ImmutableMap.of("query", query)); } Message.Response response = ClientState.getCQLQueryHandler().prepare(query, state, getCustomPayload()); if (tracingId != null) response.setTracingId(tracingId); return response; } catch (Exception e) { JVMStabilityInspector.inspectThrowable(e); return ErrorMessage.fromException(e); } finally { Tracing.instance.stopSession(); } }
state.prepareTracingSession(tracingId); if (state.traceNextQuery()) state.createTracingSession(); Message.Response response = state.getClientState().getCQLQueryHandler().process(query, state, options); if (options.skipMetadata() && response instanceof ResultMessage.Rows) ((ResultMessage.Rows)response).result.metadata.setSkipMetadata();
private void executeLoggedBatch(List<CQLStatement> statements) throws RequestExecutionException, RequestValidationException { BatchStatement batch = new BatchStatement(0, BatchStatement.Type.LOGGED, Lists.newArrayList(Iterables.filter(statements, ModificationStatement.class)), Attributes.none()); QueryProcessor.instance.processBatch(batch, QueryState.forInternalCalls(), BatchQueryOptions.withoutPerStatementVariables(QueryOptions.DEFAULT), System.nanoTime()); }
private boolean startSessionIfRequested() { if (state().getQueryState().traceNextQuery()) { state().getQueryState().createTracingSession(Collections.EMPTY_MAP); return true; } return false; }
public ByteBuffer trace_next_query() throws TException { UUID sessionId = UUIDGen.getTimeUUID(); state().getQueryState().prepareTracingSession(sessionId); return TimeUUIDType.instance.decompose(sessionId); }
public IAuthenticator.SaslNegotiator getSaslNegotiator(QueryState queryState) { if (saslNegotiator == null) saslNegotiator = DatabaseDescriptor.getAuthenticator().newSaslNegotiator(queryState.getClientAddress()); return saslNegotiator; } }
public Message.Response execute(QueryState state, long queryStartNanoTime) { try { UUID tracingId = null; if (isTracingRequested()) { tracingId = UUIDGen.getTimeUUID(); state.prepareTracingSession(tracingId); } if (state.traceNextQuery()) { state.createTracingSession(getCustomPayload()); Tracing.instance.begin("Preparing CQL3 query", state.getClientAddress(), ImmutableMap.of("query", query)); } Message.Response response = ClientState.getCQLQueryHandler().prepare(query, state, getCustomPayload()); if (tracingId != null) response.setTracingId(tracingId); return response; } catch (Exception e) { JVMStabilityInspector.inspectThrowable(e); return ErrorMessage.fromException(e); } finally { Tracing.instance.stopSession(); } }
QueryHandler handler = state.getClientState().getCQLQueryHandler(); ParsedStatement.Prepared prepared = handler.getPrepared(statementId); if (prepared == null) state.prepareTracingSession(tracingId); if (state.traceNextQuery()) state.createTracingSession();
public ResultMessage.Prepared prepare(String queryString, QueryState queryState) { ClientState cState = queryState.getClientState(); return prepare(queryString, cState, cState instanceof ThriftClientState); }