Fix document lookup in compiler plugin

ballerina-platform · May 6, 2024 · 18df40b · 18df40b
1 parent 1069021
commit 18df40b
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/...gin/src/main/java/io/ballerina/lib/data/xmldata/compiler/XmldataRecordFieldValidator.java b/...gin/src/main/java/io/ballerina/lib/data/xmldata/compiler/XmldataRecordFieldValidator.java
@@ -47,6 +47,7 @@
 import io.ballerina.compiler.syntax.tree.VariableDeclarationNode;
 import io.ballerina.lib.data.xmldata.compiler.objects.QualifiedName;
 import io.ballerina.projects.Document;
+import io.ballerina.projects.Module;
 import io.ballerina.projects.plugins.AnalysisTask;
 import io.ballerina.projects.plugins.SyntaxNodeAnalysisContext;
 import io.ballerina.tools.diagnostics.Diagnostic;
@@ -76,7 +77,16 @@ public class XmldataRecordFieldValidator implements AnalysisTask<SyntaxNodeAnaly
     @Override
     public void perform(SyntaxNodeAnalysisContext ctx) {
         semanticModel = ctx.semanticModel();
-        srcFile = ctx.currentPackage().getDefaultModule().document(ctx.documentId());
+        for (Module module : ctx.currentPackage().modules()) {
+            if (module.moduleId().equals(ctx.documentId().moduleId())) {
+                srcFile = module.document(ctx.documentId());
+                break;
+            }
+        }
+        if (srcFile == null) {
+            return;
+        }
+
         List<Diagnostic> diagnostics = semanticModel.diagnostics();
         boolean erroneousCompilation = diagnostics.stream()
                 .anyMatch(d -> d.diagnosticInfo().severity().equals(DiagnosticSeverity.ERROR));