public StatementClientV1(OkHttpClient httpClient, ClientSession session, String query) { requireNonNull(httpClient, "httpClient is null"); requireNonNull(session, "session is null"); requireNonNull(query, "query is null"); this.httpClient = httpClient; this.timeZone = session.getTimeZone(); this.query = query; this.requestTimeoutNanos = session.getClientRequestTimeout(); this.user = session.getUser(); this.clientCapabilities = Joiner.on(",").join(ClientCapabilities.values()); Request request = buildQueryRequest(session, query); JsonResponse<QueryResults> response = JsonResponse.execute(QUERY_RESULTS_CODEC, httpClient, request); if ((response.getStatusCode() != HTTP_OK) || !response.hasValue()) { state.compareAndSet(State.RUNNING, State.CLIENT_ERROR); throw requestFailedException("starting query", request, response); } processResponse(response.getHeaders(), response.getValue()); }
if (!session.getClientCapabilities().contains(ClientCapabilities.PATH.toString())) { throw new PrestoException(NOT_SUPPORTED, "SET PATH not supported by client");
private ConnectorSession createNewSession(TransactionId transactionId) { return testSessionBuilder() .setCatalog("test_catalog") .setSchema("information_schema") .setClientCapabilities(stream(ClientCapabilities.values()) .map(ClientCapabilities::toString) .collect(toImmutableSet())) .setTransactionId(transactionId) .build() .toConnectorSession(); } }
if (!session.getClientCapabilities().contains(ClientCapabilities.PATH.toString())) { throw new PrestoException(NOT_SUPPORTED, "SET PATH not supported by client");
private ConnectorSession createNewSession(TransactionId transactionId) { return testSessionBuilder() .setCatalog("test_catalog") .setSchema("information_schema") .setClientCapabilities(stream(ClientCapabilities.values()) .map(ClientCapabilities::toString) .collect(toImmutableSet())) .setTransactionId(transactionId) .build() .toConnectorSession(); } }