Skip to content

Commit ca4dea2

Browse files
authored
fixes issue \#222 (#223)
1 parent 06ca33b commit ca4dea2

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

java/src/main/java/org/digma/intellij/plugin/idea/psi/java/JavaSpanDiscoveryUtils.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -310,9 +310,14 @@ public static Query<PsiReference> filterNonRelevantReferencesForSpanDiscovery(@N
310310
@NotNull
311311
public static Query<PsiMethod> filterNonRelevantMethodsForSpanDiscovery(@NotNull Query<PsiMethod> psiMethods) {
312312
return psiMethods.filtering(psiMethod -> {
313-
var aClass = PsiTreeUtil.getParentOfType(psiMethod, PsiClass.class);
314-
return aClass == null ||
315-
(!aClass.isAnnotationType() && !aClass.isEnum() && !aClass.isRecord());
313+
var file = PsiTreeUtil.getParentOfType(psiMethod,PsiFile.class);
314+
//only java files are relevant
315+
if (file instanceof PsiJavaFile) {
316+
var aClass = PsiTreeUtil.getParentOfType(psiMethod, PsiClass.class);
317+
return aClass == null ||
318+
(!aClass.isAnnotationType() && !aClass.isEnum() && !aClass.isRecord());
319+
}
320+
return false;
316321
});
317322
}
318323
}

0 commit comments

Comments
 (0)