public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session) { return builder(notificationExecutor, yieldExecutor, session).build(); }
public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session) { return builder(notificationExecutor, yieldExecutor, session).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(); }
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(); }
}); TaskContext taskContext = TestingTaskContext.builder(notificationExecutor, yieldExecutor, session) .setMaxSpillSize(nodeSpillConfig.getMaxSpillPerNode()) .setQueryMaxSpillSize(nodeSpillConfig.getQueryMaxSpillPerNode())
}); TaskContext taskContext = TestingTaskContext.builder(notificationExecutor, yieldExecutor, session) .setMaxSpillSize(nodeSpillConfig.getMaxSpillPerNode()) .setQueryMaxSpillSize(nodeSpillConfig.getQueryMaxSpillPerNode())
private DriverContext createDriverContext() { return TestingTaskContext.builder(EXECUTOR, SCHEDULER, TEST_SESSION) .setMemoryPoolSize(MAX_MEMORY) .build() .addPipelineContext(0, true, true, false) .addDriverContext(); }
private DriverContext createDriverContext() { return TestingTaskContext.builder(EXECUTOR, SCHEDULER, TEST_SESSION) .setMemoryPoolSize(MAX_MEMORY) .build() .addPipelineContext(0, true, true, false) .addDriverContext(); }
private DriverContext createDriverContext(long memoryLimit) { return TestingTaskContext.builder(executor, scheduledExecutor, TEST_SESSION) .setMemoryPoolSize(succinctBytes(memoryLimit)) .build() .addPipelineContext(0, true, true, false) .addDriverContext(); }
private DriverContext createDriverContext(long memoryLimit) { return TestingTaskContext.builder(executor, scheduledExecutor, TEST_SESSION) .setMemoryPoolSize(succinctBytes(memoryLimit)) .build() .addPipelineContext(0, true, true, false) .addDriverContext(); }
@BeforeMethod public void setUp() { executor = newCachedThreadPool(daemonThreadsNamed("test-executor-%s")); scheduledExecutor = newScheduledThreadPool(2, daemonThreadsNamed("test-scheduledExecutor-%s")); TaskContext taskContext = TestingTaskContext.builder(executor, scheduledExecutor, TEST_SESSION) .setQueryMaxMemory(DataSize.valueOf("100MB")) .setMemoryPoolSize(DataSize.valueOf("10B")) .setQueryId(QUERY_ID) .build(); memoryPool = taskContext.getQueryContext().getMemoryPool(); driverContext = taskContext .addPipelineContext(0, true, true, false) .addDriverContext(); }
@BeforeMethod public void setUp() { executor = newCachedThreadPool(daemonThreadsNamed("test-executor-%s")); scheduledExecutor = newScheduledThreadPool(2, daemonThreadsNamed("test-scheduledExecutor-%s")); TaskContext taskContext = TestingTaskContext.builder(executor, scheduledExecutor, TEST_SESSION) .setQueryMaxMemory(DataSize.valueOf("100MB")) .setMemoryPoolSize(DataSize.valueOf("10B")) .setQueryId(QUERY_ID) .build(); memoryPool = taskContext.getQueryContext().getMemoryPool(); driverContext = taskContext .addPipelineContext(0, true, true, false) .addDriverContext(); }