use of org.apache.flink.runtime.jobmaster.LogicalSlot in project flink by apache.
the class SchedulerBenchmarkUtils method deployAllTasks.
public static void deployAllTasks(ExecutionGraph executionGraph, TestingLogicalSlotBuilder slotBuilder) throws JobException, ExecutionException, InterruptedException {
for (ExecutionVertex vertex : executionGraph.getAllExecutionVertices()) {
LogicalSlot slot = slotBuilder.createTestingLogicalSlot();
Execution execution = vertex.getCurrentExecutionAttempt();
execution.transitionState(ExecutionState.SCHEDULED);
execution.registerProducedPartitions(slot.getTaskManagerLocation(), true).get();
assignResourceAndDeploy(vertex, slot);
}
}
use of org.apache.flink.runtime.jobmaster.LogicalSlot in project flink by apache.
the class SchedulerBenchmarkUtils method deployTasks.
public static void deployTasks(ExecutionGraph executionGraph, JobVertexID jobVertexID, TestingLogicalSlotBuilder slotBuilder, boolean sendScheduleOrUpdateConsumersMessage) throws JobException, ExecutionException, InterruptedException {
for (ExecutionVertex vertex : executionGraph.getJobVertex(jobVertexID).getTaskVertices()) {
LogicalSlot slot = slotBuilder.createTestingLogicalSlot();
Execution execution = vertex.getCurrentExecutionAttempt();
execution.transitionState(ExecutionState.SCHEDULED);
execution.registerProducedPartitions(slot.getTaskManagerLocation(), sendScheduleOrUpdateConsumersMessage).get();
assignResourceAndDeploy(vertex, slot);
}
}
Aggregations