public TaskId getTaskId() { return taskStateMachine.getTaskId(); }
public TaskId getTaskId() { return taskStateMachine.getTaskId(); }
public TaskId getTaskId() { return taskStateMachine.getTaskId(); }
public TaskId getTaskId() { return taskStateMachine.getTaskId(); }
@Override public TaskId getTaskId() { return taskStateMachine.getTaskId(); }
@Override public TaskId getTaskId() { return taskStateMachine.getTaskId(); }
public TaskContext addTaskContext(TaskStateMachine taskStateMachine, Session session, boolean perOperatorCpuTimerEnabled, boolean cpuTimerEnabled, OptionalInt totalPartitions) { TaskContext taskContext = TaskContext.createTaskContext( this, taskStateMachine, gcMonitor, notificationExecutor, yieldExecutor, session, queryMemoryContext.newMemoryTrackingContext(), perOperatorCpuTimerEnabled, cpuTimerEnabled, totalPartitions); taskContexts.put(taskStateMachine.getTaskId(), taskContext); return taskContext; }
@Override public TaskContext addTaskContext(TaskStateMachine taskStateMachine, Session session, boolean perOperatorCpuTimerEnabled, boolean cpuTimerEnabled, OptionalInt totalPartitions) { TaskContext taskContext = TaskContext.createTaskContext( this, taskStateMachine, gcMonitor, notificationExecutor, yieldExecutor, session, queryMemoryContext.newMemoryTrackingContext(), perOperatorCpuTimerEnabled, cpuTimerEnabled, totalPartitions); taskContexts.put(taskStateMachine.getTaskId(), taskContext); return taskContext; }
@Override public TaskContext addTaskContext(TaskStateMachine taskStateMachine, Session session, boolean perOperatorCpuTimerEnabled, boolean cpuTimerEnabled, OptionalInt totalPartitions) { TaskContext taskContext = TaskContext.createTaskContext( this, taskStateMachine, gcMonitor, notificationExecutor, yieldExecutor, session, queryMemoryContext.newMemoryTrackingContext(), perOperatorCpuTimerEnabled, cpuTimerEnabled, totalPartitions); taskContexts.put(taskStateMachine.getTaskId(), taskContext); return taskContext; }
this.taskId = taskStateMachine.getTaskId(); this.taskContext = requireNonNull(taskContext, "taskContext is null"); this.outputBuffer = requireNonNull(outputBuffer, "outputBuffer is null");
try (SetThreadName ignored = new SetThreadName("Task-%s", taskStateMachine.getTaskId())) { try { localExecutionPlan = planner.plan(
try (SetThreadName ignored = new SetThreadName("Task-%s", taskStateMachine.getTaskId())) { try { localExecutionPlan = planner.plan(
@Override public TaskStatus getTaskStatus() { TaskStats stats = taskContext.getTaskStats(); return new TaskStatus(taskStateMachine.getTaskId(), TASK_INSTANCE_ID, nextTaskInfoVersion.get(), taskStateMachine.getState(), location, nodeId, ImmutableSet.of(), ImmutableList.of(), stats.getQueuedPartitionedDrivers(), stats.getRunningPartitionedDrivers(), false, stats.getPhysicalWrittenDataSize(), stats.getUserMemoryReservation(), stats.getSystemMemoryReservation(), 0, new Duration(0, MILLISECONDS)); }
@Override public TaskStatus getTaskStatus() { TaskStats stats = taskContext.getTaskStats(); return new TaskStatus(taskStateMachine.getTaskId(), TASK_INSTANCE_ID, nextTaskInfoVersion.get(), taskStateMachine.getState(), location, nodeId, ImmutableSet.of(), ImmutableList.of(), stats.getQueuedPartitionedDrivers(), stats.getRunningPartitionedDrivers(), false, stats.getPhysicalWrittenDataSize(), stats.getUserMemoryReservation(), stats.getSystemMemoryReservation(), 0, new Duration(0, MILLISECONDS)); }
private static TaskHandle createTaskHandle( TaskStateMachine taskStateMachine, TaskContext taskContext, OutputBuffer outputBuffer, LocalExecutionPlan localExecutionPlan, TaskExecutor taskExecutor) { TaskHandle taskHandle = taskExecutor.addTask( taskStateMachine.getTaskId(), outputBuffer::getUtilization, getInitialSplitsPerNode(taskContext.getSession()), getSplitConcurrencyAdjustmentInterval(taskContext.getSession()), getMaxDriversPerTask(taskContext.getSession())); taskStateMachine.addStateChangeListener(state -> { if (state.isDone()) { taskExecutor.removeTask(taskHandle); for (DriverFactory factory : localExecutionPlan.getDriverFactories()) { factory.noMoreDrivers(); } } }); return taskHandle; }
private static TaskHandle createTaskHandle( TaskStateMachine taskStateMachine, TaskContext taskContext, OutputBuffer outputBuffer, LocalExecutionPlan localExecutionPlan, TaskExecutor taskExecutor) { TaskHandle taskHandle = taskExecutor.addTask( taskStateMachine.getTaskId(), outputBuffer::getUtilization, getInitialSplitsPerNode(taskContext.getSession()), getSplitConcurrencyAdjustmentInterval(taskContext.getSession()), getMaxDriversPerTask(taskContext.getSession())); taskStateMachine.addStateChangeListener(state -> { if (state.isDone()) { taskExecutor.removeTask(taskHandle); for (DriverFactory factory : localExecutionPlan.getDriverFactories()) { factory.noMoreDrivers(); } } }); return taskHandle; }
taskStateMachine.getTaskId(), TASK_INSTANCE_ID, nextTaskInfoVersion.getAndIncrement(),
taskStateMachine.getTaskId(), TASK_INSTANCE_ID, nextTaskInfoVersion.getAndIncrement(),
return new TaskStatus(taskStateMachine.getTaskId(), taskInstanceId, versionNumber,
return new TaskStatus(taskStateMachine.getTaskId(), taskInstanceId, versionNumber,