Skip to content

Commit 597dbc4

Browse files
fix: improve method body retrieval logic in InstructionPsiChangeListener
1 parent 75b449f commit 597dbc4

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

javasm-intellij-plugin/src/main/java/tokyo/peya/javasm/intellij/editor/linenumber/InstructionPsiChangeListener.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,12 @@ private static void invalidateElement(@NotNull PsiElement element)
4747
if (!(element.getContainingFile() instanceof JALFile))
4848
return;
4949

50-
MethodBodyNode methodBody = PsiTreeUtil.getParentOfType(element, MethodBodyNode.class);
50+
MethodBodyNode methodBody;
51+
if (element instanceof MethodBodyNode)
52+
methodBody = (MethodBodyNode) element;
53+
else
54+
methodBody = PsiTreeUtil.getParentOfType(element, MethodBodyNode.class);
55+
5156
if (methodBody == null)
5257
return; // メソッドボディでない場合は無視
5358

0 commit comments

Comments
 (0)