private QueryStateMachine createQueryStateMachine(String query) { return QueryStateMachine.begin( query, TEST_SESSION, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); }
private QueryStateMachine createQueryStateMachine(String query) { return QueryStateMachine.begin( query, TEST_SESSION, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); }
private <T extends Statement> DataDefinitionExecution<T> createDataDefinitionExecution( String query, Session session, ResourceGroupId resourceGroup, T statement, List<Expression> parameters, WarningCollector warningCollector) { @SuppressWarnings("unchecked") DataDefinitionTask<T> task = (DataDefinitionTask<T>) tasks.get(statement.getClass()); checkArgument(task != null, "no task for statement: %s", statement.getClass().getSimpleName()); QueryStateMachine stateMachine = QueryStateMachine.begin( query, session, locationFactory.createQueryLocation(session.getQueryId()), resourceGroup, task.isTransactionControl(), transactionManager, accessControl, executor, metadata, warningCollector); stateMachine.setUpdateType(task.getName()); return new DataDefinitionExecution<>(task, statement, transactionManager, metadata, accessControl, stateMachine, parameters); } }
private <T extends Statement> DataDefinitionExecution<T> createDataDefinitionExecution( String query, Session session, ResourceGroupId resourceGroup, T statement, List<Expression> parameters, WarningCollector warningCollector) { @SuppressWarnings("unchecked") DataDefinitionTask<T> task = (DataDefinitionTask<T>) tasks.get(statement.getClass()); checkArgument(task != null, "no task for statement: %s", statement.getClass().getSimpleName()); QueryStateMachine stateMachine = QueryStateMachine.begin( query, session, locationFactory.createQueryLocation(session.getQueryId()), resourceGroup, task.isTransactionControl(), transactionManager, accessControl, executor, metadata, warningCollector); stateMachine.setUpdateType(task.getName()); return new DataDefinitionExecution<>(task, statement, transactionManager, metadata, accessControl, stateMachine, parameters); } }
private QueryStateMachine createQueryStateMachine(String query, Session session, TransactionManager transactionManager) { return QueryStateMachine.begin( query, session, URI.create("fake://uri"), new ResourceGroupId("test"), true, transactionManager, new AccessControlManager(transactionManager), executor, metadata, WarningCollector.NOOP); }
private QueryStateMachine createQueryStateMachine(String query, Session session, TransactionManager transactionManager) { return QueryStateMachine.begin( query, session, URI.create("fake://uri"), new ResourceGroupId("test"), true, transactionManager, new AccessControlManager(transactionManager), executor, metadata, WarningCollector.NOOP); }
private QueryStateMachine createQueryStateMachine(String query, Session session, TransactionManager transactionManager) { return QueryStateMachine.begin( query, session, URI.create("fake://uri"), new ResourceGroupId("test"), true, transactionManager, new AccessControlManager(transactionManager), executor, metadata, WarningCollector.NOOP); }
private QueryStateMachine createQueryStateMachine(String query, Session session, TransactionManager transactionManager) { return QueryStateMachine.begin( query, session, URI.create("fake://uri"), new ResourceGroupId("test"), true, transactionManager, new AllowAllAccessControl(), executor, metadata, WarningCollector.NOOP); }
private QueryStateMachine createQueryStateMachine(String query, Session session, TransactionManager transactionManager) { return QueryStateMachine.begin( query, session, URI.create("fake://uri"), new ResourceGroupId("test"), true, transactionManager, new AccessControlManager(transactionManager), executor, metadata, WarningCollector.NOOP); }
private QueryStateMachine createQueryStateMachine(String query, Session session, TransactionManager transactionManager) { return QueryStateMachine.begin( query, session, URI.create("fake://uri"), new ResourceGroupId("test"), true, transactionManager, new AllowAllAccessControl(), executor, metadata, WarningCollector.NOOP); }
private void testSetSessionWithParameters(String property, Expression expression, String expectedValue, List<Expression> parameters) { QualifiedName qualifiedPropName = QualifiedName.of(CATALOG_NAME, property); QueryStateMachine stateMachine = QueryStateMachine.begin( format("set %s = 'old_value'", qualifiedPropName), TEST_SESSION, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); getFutureValue(new SetSessionTask().execute(new SetSession(qualifiedPropName, expression), transactionManager, metadata, accessControl, stateMachine, parameters)); Map<String, String> sessionProperties = stateMachine.getSetSessionProperties(); assertEquals(sessionProperties, ImmutableMap.of(qualifiedPropName.toString(), expectedValue)); } }
private void testSetSessionWithParameters(String property, Expression expression, String expectedValue, List<Expression> parameters) { QualifiedName qualifiedPropName = QualifiedName.of(CATALOG_NAME, property); QueryStateMachine stateMachine = QueryStateMachine.begin( format("set %s = 'old_value'", qualifiedPropName), TEST_SESSION, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); getFutureValue(new SetSessionTask().execute(new SetSession(qualifiedPropName, expression), transactionManager, metadata, accessControl, stateMachine, parameters)); Map<String, String> sessionProperties = stateMachine.getSetSessionProperties(); assertEquals(sessionProperties, ImmutableMap.of(qualifiedPropName.toString(), expectedValue)); } }
private void assertSetRole(String statement, Map<String, SelectedRole> expected) { SetRole setRole = (SetRole) parser.createStatement(statement); QueryStateMachine stateMachine = QueryStateMachine.begin( statement, testSessionBuilder() .setCatalog(CATALOG_NAME) .build(), URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); new SetRoleTask().execute(setRole, transactionManager, metadata, accessControl, stateMachine, ImmutableList.of()); QueryInfo queryInfo = stateMachine.getQueryInfo(Optional.empty()); assertEquals(queryInfo.getSetRoles(), expected); } }
private void assertSetRole(String statement, Map<String, SelectedRole> expected) { SetRole setRole = (SetRole) parser.createStatement(statement); QueryStateMachine stateMachine = QueryStateMachine.begin( statement, testSessionBuilder() .setCatalog(CATALOG_NAME) .build(), URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); new SetRoleTask().execute(setRole, transactionManager, metadata, accessControl, stateMachine, ImmutableList.of()); QueryInfo queryInfo = stateMachine.getQueryInfo(Optional.empty()); assertEquals(queryInfo.getSetRoles(), expected); } }
.build(); QueryStateMachine stateMachine = QueryStateMachine.begin( "reset foo", session,
.build(); QueryStateMachine stateMachine = QueryStateMachine.begin( "reset foo", session,
private Set<String> executeDeallocate(String statementName, String sqlString, Session session) { TransactionManager transactionManager = createTestTransactionManager(); AccessControl accessControl = new AccessControlManager(transactionManager); QueryStateMachine stateMachine = QueryStateMachine.begin( sqlString, session, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); Deallocate deallocate = new Deallocate(new Identifier(statementName)); new DeallocateTask().execute(deallocate, transactionManager, metadata, new AllowAllAccessControl(), stateMachine, emptyList()); return stateMachine.getDeallocatedPreparedStatements(); } }
private Set<String> executeDeallocate(String statementName, String sqlString, Session session) { TransactionManager transactionManager = createTestTransactionManager(); AccessControl accessControl = new AccessControlManager(transactionManager); QueryStateMachine stateMachine = QueryStateMachine.begin( sqlString, session, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, accessControl, executor, metadata, WarningCollector.NOOP); Deallocate deallocate = new Deallocate(new Identifier(statementName)); new DeallocateTask().execute(deallocate, transactionManager, metadata, new AllowAllAccessControl(), stateMachine, emptyList()); return stateMachine.getDeallocatedPreparedStatements(); } }
private Map<String, String> executePrepare(String statementName, Statement statement, String sqlString, Session session) { TransactionManager transactionManager = createTestTransactionManager(); QueryStateMachine stateMachine = QueryStateMachine.begin( sqlString, session, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, new AccessControlManager(transactionManager), executor, metadata, WarningCollector.NOOP); Prepare prepare = new Prepare(identifier(statementName), statement); new PrepareTask(new SqlParser()).execute(prepare, transactionManager, metadata, new AllowAllAccessControl(), stateMachine, emptyList()); return stateMachine.getAddedPreparedStatements(); } }
private Map<String, String> executePrepare(String statementName, Statement statement, String sqlString, Session session) { TransactionManager transactionManager = createTestTransactionManager(); QueryStateMachine stateMachine = QueryStateMachine.begin( sqlString, session, URI.create("fake://uri"), new ResourceGroupId("test"), false, transactionManager, new AccessControlManager(transactionManager), executor, metadata, WarningCollector.NOOP); Prepare prepare = new Prepare(identifier(statementName), statement); new PrepareTask(new SqlParser()).execute(prepare, transactionManager, metadata, new AllowAllAccessControl(), stateMachine, emptyList()); return stateMachine.getAddedPreparedStatements(); } }