Add guard against null pointer

Signed-off-by: Juan Manuel Leflet Estrada <jleflete@redhat.com>
konveyor · Jul 16, 2024 · 9967fe8 · 9967fe8
1 parent 74033e8
commit 9967fe8
Showing 1 changed file with 17 additions and 10 deletions.
diff --git a/...undle.core/src/main/java/io/konveyor/tackle/core/internal/symbol/WithAnnotationQuery.java b/...undle.core/src/main/java/io/konveyor/tackle/core/internal/symbol/WithAnnotationQuery.java
@@ -2,16 +2,19 @@
 
 import io.konveyor.tackle.core.internal.query.AnnotationQuery;
 import org.eclipse.jdt.core.IAnnotation;
+import org.eclipse.jdt.core.ICompilationUnit;
 import org.eclipse.jdt.core.IImportDeclaration;
 import org.eclipse.jdt.core.IMemberValuePair;
 import org.eclipse.jdt.core.JavaModelException;
 import org.eclipse.jdt.core.search.SearchMatch;
 import org.eclipse.jdt.internal.core.Annotation;
+import org.eclipse.jdt.internal.core.CompilationUnit;
 import org.eclipse.jdt.internal.core.SourceRefElement;
 
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.Set;
 import java.util.regex.Pattern;
 
@@ -106,16 +109,20 @@ private String getFQN(IAnnotation annotation) {
             return name;
         } else {
             // If not, the annotation must have been imported. Look in the imports:
-            try {
-                return Arrays.stream(((Annotation) annotation).getCompilationUnit().getImports())
-                        .filter(i -> i.getElementName().endsWith(name))
-                        .findFirst()
-                        .map(IImportDeclaration::getElementName)
-                        .orElse("");
-            } catch (JavaModelException e) {
-                e.printStackTrace();
-                return "";
-            }
+            return tryToGetImports(annotation).stream()
+                    .filter(i -> i.getElementName().endsWith(name))
+                    .findFirst()
+                    .map(IImportDeclaration::getElementName)
+                    .orElse("");
+        }
+    }
+
+    private List<IImportDeclaration> tryToGetImports(IAnnotation annotation) {
+        try {
+            return Optional.ofNullable(((Annotation) annotation).getCompilationUnit().getImports()).map(Arrays::asList).orElse(List.of());
+        } catch (Throwable e) {
+            e.printStackTrace();
+            return List.of();
         }
     }
 }