[FLINK-21658] Adopts JobGraphBuilder to create job graph

akalash · Mar 8, 2021 · 7c8207b · 7c8207b
1 parent d0ec358
commit 7c8207b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/org/apache/flink/scheduler/benchmark/SchedulerBenchmarkUtils.java b/src/main/java/org/apache/flink/scheduler/benchmark/SchedulerBenchmarkUtils.java
@@ -30,6 +30,7 @@
 import org.apache.flink.runtime.executiongraph.ExecutionVertex;
 import org.apache.flink.runtime.executiongraph.TaskExecutionStateTransition;
 import org.apache.flink.runtime.jobgraph.JobGraph;
+import org.apache.flink.runtime.jobgraph.JobGraphBuilder;
 import org.apache.flink.runtime.jobgraph.JobVertex;
 import org.apache.flink.runtime.jobgraph.JobVertexID;
 import org.apache.flink.runtime.jobmaster.LogicalSlot;
@@ -80,7 +81,9 @@ public static JobGraph createJobGraph(
 			List<JobVertex> jobVertices,
 			JobConfiguration jobConfiguration) throws IOException {
 
-		final JobGraph jobGraph = new JobGraph(jobVertices.toArray(new JobVertex[0]));
+		final JobGraph jobGraph = JobGraphBuilder.newStreamingJobGraphBuilder()
+				.addJobVertices(jobVertices)
+				.build();
 
 		jobGraph.setJobType(jobConfiguration.getJobType());