@UsedByGeneratedCode public void killQuery(String queryId) { queryManager.cancelQuery(new QueryId(queryId)); }
public TaskId(StageId stageId, String id) { validateId(id); this.fullId = stageId.getQueryId().getId() + "." + stageId.getId() + "." + id; }
public QueryId getQueryId() { return new QueryId(QueryId.parseDottedId(fullId, 3, "taskId").get(0)); }
public StageId(String queryId, String id) { this.queryId = new QueryId(queryId); this.id = validateId(id); }
@EventField public String getQueryId() { return queryId.toString(); }
public TaskId(String queryId, String stageId, String id) { validateId(id); this.fullId = queryId + "." + stageId + "." + id; }
public String getId() { return QueryId.parseDottedId(fullId, 3, "taskId").get(2); }
query = manager.getQueryInfo(QueryId.valueOf(queryId));
@EventField public String getQueryId() { return queryId.toString(); }
public QueryId(String id) { this.id = validateId(id); }
public Session toSession(SessionPropertyManager sessionPropertyManager) { return new Session( new QueryId(queryId), transactionId, clientTransactionSupport, new Identity(user, Optional.empty()), source, catalog, schema, timeZoneKey, locale, remoteUserAddress, userAgent, startTime, systemProperties, catalogProperties, sessionPropertyManager); } }
@JsonCreator public static QueryId valueOf(String queryId) { List<String> ids = parseDottedId(queryId, 1, "queryId"); return new QueryId(ids.get(0)); }
@EventField public String getQueryId() { return queryId.toString(); }
public StageId(QueryId queryId, String id) { this.queryId = requireNonNull(queryId, "queryId is null"); this.id = validateId(id); }
public static TaskContext createTaskContext(Executor executor, Session session, DataSize maxMemory, DataSize preallocated) { MemoryPool memoryPool = new MemoryPool(new MemoryPoolId("test"), new DataSize(1, GIGABYTE)); MemoryPool systemMemoryPool = new MemoryPool(new MemoryPoolId("testSystem"), new DataSize(1, GIGABYTE)); QueryContext queryContext = new QueryContext(new QueryId("test_query"), maxMemory, memoryPool, systemMemoryPool, executor); return createTaskContext(queryContext, executor, session, preallocated); }
public StageId getStageId() { List<String> ids = QueryId.parseDottedId(fullId, 3, "taskId"); return new StageId(new QueryId(ids.get(0)), ids.get(1)); }
public ConnectorSession toConnectorSession() { return new FullConnectorSession(queryId.toString(), identity, timeZoneKey, locale, startTime); }
private void testSetSession(Expression expression, String expectedValue) throws Exception { TransactionManager transactionManager = createTestTransactionManager(); QueryStateMachine stateMachine = QueryStateMachine.begin(new QueryId("query"), "set foo.bar = 'baz'", TEST_SESSION, URI.create("fake://uri"), false, transactionManager, executor); new SetSessionTask().execute(new SetSession(QualifiedName.of("foo", "bar"), expression), transactionManager, metadata, new AllowAllAccessControl(), stateMachine).join(); Map<String, String> sessionProperties = stateMachine.getSetSessionProperties(); assertEquals(sessionProperties, ImmutableMap.of("foo.bar", expectedValue)); } }
@JsonCreator public static StageId valueOf(String stageId) { List<String> ids = QueryId.parseDottedId(stageId, 2, "stageId"); return new StageId(new QueryId(ids.get(0)), ids.get(1)); }
public ConnectorSession toConnectorSession(String catalog) { requireNonNull(catalog, "catalog is null"); return new FullConnectorSession( queryId.toString(), identity, timeZoneKey, locale, startTime, catalogProperties.getOrDefault(catalog, ImmutableMap.of()), catalog, sessionPropertyManager); }