Clean up ModuleContext srcDocContextMap

Nadeeshan96 · Oct 12, 2023 · e4018fd · e4018fd
1 parent b0cd6ae
commit e4018fd
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/compiler/ballerina-lang/src/main/java/io/ballerina/projects/DocumentContext.java b/compiler/ballerina-lang/src/main/java/io/ballerina/projects/DocumentContext.java
@@ -61,6 +61,10 @@ class DocumentContext {
     private final String content;
     private boolean disableSyntaxTree = false;
 
+    public void cleanUp() {
+        textDocument = null;
+    }
+
     private DocumentContext(DocumentId documentId, String name, String content, boolean disableSyntaxTree) {
         this.documentId = documentId;
         this.name = name;

diff --git a/compiler/ballerina-lang/src/main/java/io/ballerina/projects/JBallerinaBackend.java b/compiler/ballerina-lang/src/main/java/io/ballerina/projects/JBallerinaBackend.java
@@ -176,6 +176,7 @@ private void performCodeGen() {
                 moduleDiagnostics.add(
                         new PackageDiagnostic(diagnostic, moduleContext.descriptor(), moduleContext.project()));
             }
+            moduleContext.cleanUp();
         }
         // add compilation diagnostics
         diagnostics.addAll(moduleDiagnostics);

diff --git a/compiler/ballerina-lang/src/main/java/io/ballerina/projects/ModuleContext.java b/compiler/ballerina-lang/src/main/java/io/ballerina/projects/ModuleContext.java
@@ -91,6 +91,12 @@ class ModuleContext {
     private Set<ModuleLoadRequest> allModuleLoadRequests = null;
     private Set<ModuleLoadRequest> allTestModuleLoadRequests = null;
 
+    public void cleanUp() {
+        for (DocumentContext value : srcDocContextMap.values()) {
+            value.cleanUp();
+        }
+    }
+
     ModuleContext(Project project,
                   ModuleId moduleId,
                   ModuleDescriptor moduleDescriptor,