tiny/spring-state-machine-renderer
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

gemini back 3

Browse Source
This commit is contained in:
Kamil Patryk Kozakowski
2026-06-27 05:42:15 +02:00
parent bd55a66985
commit c35c75efa3
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/AbstractCallGraphEngine.java
View File

@@ -2138,10 +2138,20 @@ import java.util.*;
List<Expression> results = new ArrayList<>(); List<Expression> results = new ArrayList<>();
if (expr instanceof SimpleName sn) { if (expr instanceof SimpleName sn) {
String varName = sn.getIdentifier(); String varName = sn.getIdentifier();
if (!visitedVariables.add(varName)) { String methodKey = "";
MethodDeclaration enclosingMethod = findEnclosingMethod(expr);
if (enclosingMethod != null) {
TypeDeclaration enclosingType = findEnclosingType(enclosingMethod);
if (enclosingType != null) {
methodKey = context.getFqn(enclosingType) + "." + enclosingMethod.getName().getIdentifier() + "#" + varName;
}
}
if (methodKey.isEmpty()) {
methodKey = varName;
}
if (!visitedVariables.add(methodKey)) {
return results; // Break infinite loop return results; // Break infinite loop
} }
MethodDeclaration enclosingMethod = findEnclosingMethod(expr);
if (enclosingMethod != null) { if (enclosingMethod != null) {
enclosingMethod.accept(new ASTVisitor() { enclosingMethod.accept(new ASTVisitor() {
@Override @Override
@@ -2160,7 +2170,7 @@ import java.util.*;
} }
}); });
if (!results.isEmpty()) { if (!results.isEmpty()) {
visitedVariables.remove(varName); visitedVariables.remove(methodKey);
return results; return results;
} }
@@ -2200,14 +2210,14 @@ import java.util.*;
} }
}); });
if (!callerExprs.isEmpty()) { if (!callerExprs.isEmpty()) {
visitedVariables.remove(varName); visitedVariables.remove(methodKey);
return callerExprs; return callerExprs;
} }
} }
} }
} }
} }
visitedVariables.remove(varName); visitedVariables.remove(methodKey);
} }
results.add(expr); results.add(expr);
return results; return results;