diff --git a/state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/ConstantExtractor.java b/state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/ConstantExtractor.java index cde5ea5..a23eb72 100644 --- a/state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/ConstantExtractor.java +++ b/state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/ConstantExtractor.java @@ -30,6 +30,14 @@ public class ConstantExtractor { } public void extractConstantsFromExpression(Expression expr, List constants) { + if (constantResolver != null) { + String resolved = constantResolver.resolve(expr, context); + if (resolved != null) { + constants.add(resolved); + return; + } + } + if (expr instanceof QualifiedName qn) { constants.add(qn.toString()); } else if (expr instanceof StringLiteral sl) {