Drop proc bodies after find-usages finishes

SpaceManiac · SpaceManiac · commit bd255d1ca7ec · 2019-11-02T18:05:44.000-07:00
diff --git a/src/dreammaker/objtree.rs b/src/dreammaker/objtree.rs
@@ -1008,6 +1008,17 @@ impl ObjectTree {
 
         self.register_proc(context, location, parent, proc_name, declaration, parameters, code)
     }
+
+    /// Drop all code ASTs to attempt to reduce memory usage.
+    pub fn drop_code(&mut self) {
+        for node in self.graph.node_weights_mut() {
+            for (_, typroc) in node.procs.iter_mut() {
+                for proc in typroc.value.iter_mut() {
+                    proc.code = Code::Disabled;
+                }
+            }
+        }
+    }
 }
 
 #[inline]
diff --git a/src/langserver/main.rs b/src/langserver/main.rs
@@ -400,6 +400,8 @@ impl<'a, W: io::ResponseWrite> Engine<'a, W> {
             );
         }
 
+        self.objtree.drop_code();
+
         Ok(())
     }
 

Original file line number	Diff line number	Diff line change
`@@ -400,6 +400,8 @@ impl<'a, W: io::ResponseWrite> Engine<'a, W> {`
`400`	`400`	`);`
`401`	`401`	`}`
`402`	`402`
	`403`	`+ self.objtree.drop_code();`
	`404`	`+`
`403`	`405`	`Ok(())`
`404`	`406`	`}`
`405`	`407`