Received Digit fixup

docwho2 · Dec 2, 2023 · b4ab4b5 · b4ab4b5
1 parent e618f11
commit b4ab4b5
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/ChimeSMAFlow/src/main/java/cloud/cleo/chimesma/actions/ReceiveDigitsAction.java b/ChimeSMAFlow/src/main/java/cloud/cleo/chimesma/actions/ReceiveDigitsAction.java
@@ -6,6 +6,7 @@
 
 import cloud.cleo.chimesma.model.*;
 import static cloud.cleo.chimesma.model.SMARequest.SMAEventType.DIGITS_RECEIVED;
+import java.util.Map;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 import lombok.experimental.SuperBuilder;
@@ -34,9 +35,17 @@ protected StringBuilder getDebugSummary() {
                 .append(" [").append(getInputDigitsRegex()).append(']');
     }
 
+    /**
+     * Override this because this is special case where ID needs to be
+     * set at render time, not after success like LexBot for example.
+     *
+     * @return
+     */
     @Override
-    protected void onActionSuccessful() {
-        setTransactionAttribute(RECEIVE_DIGITS_ID, getId().toString());
+    public Map<String, Object> getTransactionAttributes() {
+        final var attrs = super.getTransactionAttributes();
+        attrs.put(RECEIVE_DIGITS_ID, getId().toString());
+        return attrs;
     }
 
     @Override