DYN-8358 cleanup dependency graph from old nodes

src/Engine/ProtoAssociative/CodeGen.cs

@@ -2156,7 +2156,8 @@ public override int Emit(ProtoCore.AST.Node codeBlockNode, ProtoCore.Associative
                 hasReturnStatement = EmitCodeBlock(codeblock.Body, ref inferedType, ProtoCore.CompilerDefinitions.SubCompilePass.UnboundIdentifier, false);
                 if (compilePass == ProtoCore.CompilerDefinitions.CompilePass.GlobalScope && !hasReturnStatement)
                 {
-                    EmitReturnNull(); 
+                    var guidList = codeblock.Body.Select(x=> (x as BinaryExpressionNode)?.guid).Where(x=>x!=null).ToList();
+                    EmitReturnNull(guidList.Any() ? guidList.Last() : null); 
                 }
 
                 compilePass++;
@@ -5909,7 +5910,7 @@ private void EmitGroupExpressionNode(AssociativeNode node, ref ProtoCore.Type in
             }
         }
 
-        protected override void EmitReturnNull()
+        protected override void EmitReturnNull(Guid? guid = null)
         {
             int startpc = pc;
 
@@ -5928,6 +5929,10 @@ protected override void EmitReturnNull()
             retNode.updateBlock.startpc = startpc;
             retNode.updateBlock.endpc = pc - 1;
             retNode.isReturn = true;
+            if (guid != null)
+            {
+                retNode.guid = (Guid)guid;
+            }
 
             PushGraphNode(retNode);
         }
@@ -6130,4 +6135,4 @@ protected override void DfsTraverse(ProtoCore.AST.Node pNode, ref ProtoCore.Type
             int blockId = codeBlock.codeBlockId; 
         }
     }
-}
+}

src/Engine/ProtoCore/AssociativeGraph.cs

@@ -57,7 +57,12 @@ public static void BuildGraphNodeDependencies(List<AssociativeGraph.GraphNode> g
             {
                 return;
             }
-
+            for (int i = 0; i < graphNodesInScope.Count; ++i)
+            {
+                AssociativeGraph.GraphNode currentNode = graphNodesInScope[i];
+                currentNode.ParentNodes.Clear();
+                currentNode.ChildrenNodes.Clear();
+            }
             // Get the current graphnode to check against the list
             //  [a = 10]  -> this one
             //  c = 1
@@ -967,33 +972,32 @@ internal IEnumerable<Guid> ClearCycles(IEnumerable<GraphNode> graphNodes)
             Stack<GraphNode> stack = new Stack<GraphNode>();
             stack.Push(this);
 
-            var visited = graphNodes.ToDictionary(node => node, node => false);
+            var visited = new HashSet<int>();
 
-            var guids = new List<Guid>();
+            var guids = new HashSet<Guid>();
             while(stack.Any())
             {
                 var node = stack.Pop();
-                if (!visited[node])
+                if (!visited.Contains(node.UID))
                 {
                     guids.Add(node.guid);
                     if (node.isCyclic)
                     {
                         node.isCyclic = false;
                         node.isActive = true;
-
                     }
-                    visited[node] = true;
+                    visited.Add(node.UID);
                 }
 
                 foreach(var cNode in node.ChildrenNodes)
                 {
-                    if (!visited[cNode])
+                    if (!visited.Contains(cNode.UID))
                     {
                         stack.Push(cNode);
                     }
                 }
             }
-            return guids;
+            return guids.ToList();
         }
 
         public void PushDependent(GraphNode dependent)
@@ -1500,6 +1504,28 @@ private ulong GetGraphNodeKey(int classIndex, int procIndex)
             return (((ulong)ci) << 32) | pi;
         }
 
+        internal void Purge(List<AST.AssociativeAST.AssociativeNode> oldNodes)
+        {
+            HashSet<Guid> oldGuids = [];
+            foreach (AST.AssociativeAST.AssociativeNode astNode in oldNodes)
+            {
+                var x = (astNode as AST.AssociativeAST.BinaryExpressionNode)?.guid;
+                if (x != null)
+                {
+                    oldGuids.Add(x.Value);
+                }
+            }
+            if(oldGuids.Count == 0)
+            {
+                return;
+            }
+            List<GraphNode> nodeList;
+            if(graphNodeMap.TryGetValue(GetGraphNodeKey(Constants.kInvalidPC, Constants.kInvalidPC), out nodeList))
+            {
+                nodeList.RemoveAll(node => oldGuids.Contains(node.guid));
+            }
+        }
+
         public DependencyGraph(Core core)
         {
             this.core = core;

src/Engine/ProtoCore/CodeGen.cs

@@ -2363,7 +2363,7 @@ protected bool InsideFunction()
 
         // used to manully emit "return = null" instruction if a function or language block does not have a return statement
         // there is update code involved in associativen code gen, so it is not implemented here
-        protected abstract void EmitReturnNull();
+        protected abstract void EmitReturnNull(Guid? guid = null);
 
         protected abstract void DfsTraverse(Node node, ref ProtoCore.Type inferedType, bool isBooleanOp = false, ProtoCore.AssociativeGraph.GraphNode graphNode = null, 
             ProtoCore.CompilerDefinitions.SubCompilePass subPass = ProtoCore.CompilerDefinitions.SubCompilePass.None, ProtoCore.AST.Node parentNode = null);

src/Engine/ProtoImperative/CodeGen.cs

@@ -2543,7 +2543,7 @@ private void EmitSetExpressionUID(int exprId)
             //AppendInstruction(instr);
         }
 
-        protected override void EmitReturnNull()
+        protected override void EmitReturnNull(Guid? guid = null)
         {
             EmitPushNull();
             EmitReturnToRegister();

src/Engine/ProtoScript/Runners/LiveRunner.cs

@@ -1660,6 +1660,7 @@ private void CompileAndExecuteForDeltaExecution(List<AssociativeNode> astList)
             }
 
             ResetForDeltaExecution();
+            runnerCore.CodeBlockList[(int)Language.Associative].instrStream.dependencyGraph.Purge(dispatchASTList);
             CompileAndExecute(dispatchASTList);
             PostExecution();
         }
@@ -1721,6 +1722,8 @@ private void SynchronizeInternal(GraphSyncData syncData)
                 var guids = runtimeCore.ExecutedAstGuids.ToList();
                 executedAstGuids[syncData.SessionID] = guids;
                 runtimeCore.RemoveExecutedAstGuids();
+
+                runnerCore.CodeBlockList[(int)Language.Associative].instrStream.dependencyGraph.Purge(changeSetComputer.csData.DeletedBinaryExprASTNodes);
             }
         }