spline #831 fix persistent model builder for schema agnostic operatio…

…n graphs
AbsaOSS · Apr 7, 2021 · bf1cd99 · bf1cd99
1 parent 2849c37
commit bf1cd99
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/.../scala/za/co/absa/spline/producer/service/model/ExecutionPlanPersistentModelBuilder.scala b/.../scala/za/co/absa/spline/producer/service/model/ExecutionPlanPersistentModelBuilder.scala
@@ -171,11 +171,13 @@ class ExecutionPlanPersistentModelBuilder private(
         }
       }
 
-      this._pmEmits :+= EdgeDef.Emits.edge(
-        opKey,
-        keyCreator.asSchemaKey(schemaInfoByOpId(op.id).oid),
-        epPKey
-      )
+      for (schemaInfo <- schemaInfoByOpId.get(op.id)) {
+        this._pmEmits :+= EdgeDef.Emits.edge(
+          opKey,
+          keyCreator.asSchemaKey(schemaInfo.oid),
+          epPKey
+        )
+      }
 
       this._pmFollows ++= op.childIds.zipWithIndex map {
         case (childId, i) =>