[GOBBLIN-855] persist dag after addspec

Closes apache#2711 from arjun4084346/persistdags
jhsenjaliya · Mar 24, 2020 · 4ae7e94 · 4ae7e94
1 parent 93dedca
commit 4ae7e94
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...in-service/src/main/java/org/apache/gobblin/service/modules/orchestration/DagManager.java b/...in-service/src/main/java/org/apache/gobblin/service/modules/orchestration/DagManager.java
@@ -596,6 +596,8 @@ private void submitJob(DagNode<JobExecutionPlan> dagNode) {
         // blocks (by calling Future#get()) until the submission is completed.
         Future addSpecFuture = producer.addSpec(jobSpec);
         dagNode.getValue().setJobFuture(Optional.of(addSpecFuture));
+        //Persist the dag
+        this.dagStateStore.writeCheckpoint(this.dags.get(DagManagerUtils.generateDagId(dagNode)));
 
         if (this.metricContext != null) {
           getRunningJobsCounter(dagNode).inc();