public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, DataSize maxMemory) { return builder(notificationExecutor, yieldExecutor, session) .setQueryMaxMemory(maxMemory) .build(); }
public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, TaskStateMachine taskStateMachine) { return builder(notificationExecutor, yieldExecutor, session) .setTaskStateMachine(taskStateMachine) .build(); }
private boolean shouldDenyPrivilege(String userName, String entityName, TestingPrivilegeType type) { TestingPrivilege testPrivilege = privilege(userName, entityName, type); for (TestingPrivilege denyPrivilege : denyPrivileges) { if (denyPrivilege.matches(testPrivilege)) { return true; } } return false; }
public static MaterializedResult toMaterializedResult(Session session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
@Override public void close() { runner.close(); } }
@Override protected List<Driver> createDrivers(TaskContext taskContext) { return localQueryRunner.createDrivers(session, query, new NullOutputFactory(), taskContext); } }
@Override public Session getSession() { return queryRunner.getDefaultSession(); }
@Override public WarningCollector create() { return new TestingWarningCollector(config, testConfig); } }
@Override public ConnectorOutputTableHandle beginCreateTable(ConnectorSession session, ConnectorTableMetadata tableMetadata, Optional<ConnectorNewTableLayout> layout) { createTable(session, tableMetadata, false); return TestingHandle.INSTANCE; }
@Override public ConnectorPageSink createPageSink(ConnectorTransactionHandle transactionHandle, ConnectorSession session, ConnectorOutputTableHandle outputTableHandle) { return new TestingPageSink(); }
@Override public OperatorFactory duplicate() { return new PageConsumerOperatorFactory(operatorId, planNodeId, pageConsumer, pagePreprocessor); } }
public static MaterializedResult toMaterializedResult(ConnectorSession session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, DataSize maxMemory) { return builder(notificationExecutor, yieldExecutor, session) .setQueryMaxMemory(maxMemory) .build(); }
public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, TaskStateMachine taskStateMachine) { return builder(notificationExecutor, yieldExecutor, session) .setTaskStateMachine(taskStateMachine) .build(); }
@Override protected List<Driver> createDrivers(TaskContext taskContext) { return localQueryRunner.createDrivers(session, query, new NullOutputFactory(), taskContext); } }
private boolean shouldDenyPrivilege(String userName, String entityName, TestingPrivilegeType type) { TestingPrivilege testPrivilege = privilege(userName, entityName, type); for (TestingPrivilege denyPrivilege : denyPrivileges) { if (denyPrivilege.matches(testPrivilege)) { return true; } } return false; }
public static MaterializedResult toMaterializedResult(ConnectorSession session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public static MaterializedResult toMaterializedResult(Session session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }