Clean up TextDocument line map after codegen

Nadeeshan96 · Oct 13, 2023 · 0bdc348 · 0bdc348
1 parent ec28a7c
commit 0bdc348
Show file tree

Hide file tree

Showing 3 changed files with 10 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
@@ -193,5 +193,8 @@ void shrink() {
             this.compilationUnit.topLevelNodes.clear();
         }
         this.content = null;
+        if (this.textDocument != null) {
+            this.textDocument.cleanUp();
+        }
     }
 }
diff --git a/compiler/ballerina-tools-api/src/main/java/io/ballerina/tools/text/StringTextDocument.java b/compiler/ballerina-tools-api/src/main/java/io/ballerina/tools/text/StringTextDocument.java
@@ -33,6 +33,11 @@ class StringTextDocument extends TextDocument {
         this.text = text;
     }
 
+    @Override
+    public void cleanUp() {
+        this.textLineMap = null;
+    }
+
     @Override
     public TextDocument apply(TextDocumentChange textDocumentChange) {
         int startOffset = 0;

diff --git a/compiler/ballerina-tools-api/src/main/java/io/ballerina/tools/text/TextDocument.java b/compiler/ballerina-tools-api/src/main/java/io/ballerina/tools/text/TextDocument.java
@@ -27,6 +27,8 @@
 public abstract class TextDocument {
     private LineMap lineMap;
 
+    public abstract void cleanUp();
+
     public abstract TextDocument apply(TextDocumentChange textDocumentChange);
 
     protected abstract LineMap populateTextLineMap();