Fix termination bug caused by in-degree zero mirrors

atrostan · Dec 3, 2021 · aefaa7b · aefaa7b
1 parent b46a2a0
commit aefaa7b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/scala/com/cluster/graph/entity/MirrorEntity.scala b/src/main/scala/com/cluster/graph/entity/MirrorEntity.scala
@@ -64,6 +64,12 @@ class MirrorEntity(
  case ApplyResult(stepNum, oldVal, newVal) => {
  ctxLog("Received apply value from Main " + newVal)
  localScatter(stepNum, oldVal, newVal, sharding)
+ // If no incoming edges, then need to send a message to Main for the next superstep
+ if(this.partitionInDegree == 0) {
+ val cmd = MirrorTotal(stepNum + 1, None)
+ val mainRef = sharding.entityRefFor(VertexEntity.TypeKey, main.toString())
+ mainRef ! cmd
+ }
  Behaviors.same
  }