@Override public boolean updateQueryWithJson(final String json) throws TajoException { final BlockingInterface stub = conn.getTMStub(); final QueryRequest request = buildQueryRequest(json, true); UpdateQueryResponse response; try { response = stub.updateQuery(null, request); } catch (ServiceException e) { throw new RuntimeException(e); } throwIfError(response.getState()); return true; }
@Override public boolean updateQueryWithJson(final String json) throws TajoException { final BlockingInterface stub = conn.getTMStub(); final QueryRequest request = buildQueryRequest(json, true); UpdateQueryResponse response; try { response = stub.updateQuery(null, request); } catch (ServiceException e) { throw new RuntimeException(e); } throwIfError(response.getState()); return true; }
@Override public boolean updateQuery(final String sql) throws TajoException { final BlockingInterface stub = conn.getTMStub(); final QueryRequest request = buildQueryRequest(sql, false); UpdateQueryResponse response; try { response = stub.updateQuery(null, request); } catch (ServiceException e) { throw new RuntimeException(e); } throwIfError(response.getState()); conn.updateSessionVarsCache(ProtoUtil.convertToMap(response.getSessionVars())); return true; }
@Override public boolean updateQuery(final String sql) throws TajoException { final BlockingInterface stub = conn.getTMStub(); final QueryRequest request = buildQueryRequest(sql, false); UpdateQueryResponse response; try { response = stub.updateQuery(null, request); } catch (ServiceException e) { throw new RuntimeException(e); } throwIfError(response.getState()); conn.updateSessionVarsCache(ProtoUtil.convertToMap(response.getSessionVars())); return true; }
@Override public QueryFuture executeSQLAsync(String sql) throws TajoException { ClientProtos.SubmitQueryResponse response = queryClient.executeQuery(sql); ExceptionUtil.throwIfError(response.getState()); QueryId queryId = new QueryId(response.getQueryId()); switch (response.getResultType()) { case ENCLOSED: return new QueryFutureForEnclosed(queryId, TajoClientUtil.createResultSet(this.queryClient, response, 200)); case FETCH: AsyncQueryFuture future = new AsyncQueryFuture(queryId); executor.execute(future); return future; default: return new QueryFutureForNoFetch(queryId); } }
@Override public QueryFuture executeSQLAsync(String sql) throws TajoException { ClientProtos.SubmitQueryResponse response = queryClient.executeQuery(sql); ExceptionUtil.throwIfError(response.getState()); QueryId queryId = new QueryId(response.getQueryId()); switch (response.getResultType()) { case ENCLOSED: return new QueryFutureForEnclosed(queryId, TajoClientUtil.createResultSet(this.queryClient, response, 200)); case FETCH: AsyncQueryFuture future = new AsyncQueryFuture(queryId); executor.execute(future); return future; default: return new QueryFutureForNoFetch(queryId); } }
@Override public ResultSet executeQueryAndGetResult(String sql) throws TajoException { SubmitQueryResponse response = executeQuery(sql); throwIfError(response.getState()); QueryId queryId = new QueryId(response.getQueryId()); switch (response.getResultType()) { case ENCLOSED: return TajoClientUtil.createResultSet(this, response, defaultFetchRows); case FETCH: return this.getQueryResultAndWait(queryId); default: return this.createNullResultSet(queryId); } }
@Override public ResultSet executeJsonQueryAndGetResult(final String json) throws TajoException { SubmitQueryResponse response = executeQueryWithJson(json); throwIfError(response.getState()); QueryId queryId = new QueryId(response.getQueryId()); switch (response.getResultType()) { case ENCLOSED: return TajoClientUtil.createResultSet(this, response, defaultFetchRows); case FETCH: return this.getQueryResultAndWait(queryId); default: return this.createNullResultSet(queryId); } }
@Override public ResultSet executeQueryAndGetResult(String sql) throws TajoException { SubmitQueryResponse response = executeQuery(sql); throwIfError(response.getState()); QueryId queryId = new QueryId(response.getQueryId()); switch (response.getResultType()) { case ENCLOSED: return TajoClientUtil.createResultSet(this, response, defaultFetchRows); case FETCH: return this.getQueryResultAndWait(queryId); default: return this.createNullResultSet(queryId); } }
@Override public ResultSet executeJsonQueryAndGetResult(final String json) throws TajoException { SubmitQueryResponse response = executeQueryWithJson(json); throwIfError(response.getState()); QueryId queryId = new QueryId(response.getQueryId()); switch (response.getResultType()) { case ENCLOSED: return TajoClientUtil.createResultSet(this, response, defaultFetchRows); case FETCH: return this.getQueryResultAndWait(queryId); default: return this.createNullResultSet(queryId); } }
protected TajoMemoryResultSet fetchNextQueryResult(final QueryId queryId, final int fetchRowNum) throws TajoException { final boolean compress = conn.getProperties().getBool(ClientParameters.USE_COMPRESSION); final BlockingInterface stub = conn.getTMStub(); final GetQueryResultDataRequest.Builder request = GetQueryResultDataRequest.newBuilder(); request.setSessionId(conn.sessionId) .setQueryId(queryId.getProto()) .setFetchRowNum(fetchRowNum); if (compress) { request.setCompressCodec(DEFAULT_CODEC); } GetQueryResultDataResponse response; try { response = stub.getQueryResultData(null, request.build()); } catch (ServiceException e) { throw new RuntimeException(e); } throwIfError(response.getState()); if(response.hasResultSet()) { SerializedResultSet resultSet = response.getResultSet(); return new TajoMemoryResultSet(queryId, SchemaFactory.newV1(resultSet.getSchema()), resultSet, getClientSideSessionVars()); } else { return TajoClientUtil.createNullResultSet(queryId); } }
protected TajoMemoryResultSet fetchNextQueryResult(final QueryId queryId, final int fetchRowNum) throws TajoException { final boolean compress = conn.getProperties().getBool(ClientParameters.USE_COMPRESSION); final BlockingInterface stub = conn.getTMStub(); final GetQueryResultDataRequest.Builder request = GetQueryResultDataRequest.newBuilder(); request.setSessionId(conn.sessionId) .setQueryId(queryId.getProto()) .setFetchRowNum(fetchRowNum); if (compress) { request.setCompressCodec(DEFAULT_CODEC); } GetQueryResultDataResponse response; try { response = stub.getQueryResultData(null, request.build()); } catch (ServiceException e) { throw new RuntimeException(e); } throwIfError(response.getState()); if(response.hasResultSet()) { SerializedResultSet resultSet = response.getResultSet(); return new TajoMemoryResultSet(queryId, new Schema(resultSet.getSchema()), resultSet, getClientSideSessionVars()); } else { return TajoClientUtil.createNullResultSet(queryId); } }