Inherit StdCallbackContext for CallbackContext (#57)

aws-cloudformation · May 27, 2024 · 9d1965b · 9d1965b
1 parent 818c911
commit 9d1965b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/...rc/main/java/com/amazonaws/stepfunctions/cloudformation/statemachine/CallbackContext.java b/...rc/main/java/com/amazonaws/stepfunctions/cloudformation/statemachine/CallbackContext.java
@@ -4,12 +4,13 @@
 import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
+import software.amazon.cloudformation.proxy.StdCallbackContext;
 
 @Data
 @NoArgsConstructor
 @AllArgsConstructor
 @Builder
-public class CallbackContext {
+public class CallbackContext extends StdCallbackContext {
  @Builder.Default
  private boolean deletionStarted = false;
 }
diff --git a/...in/java/com/amazonaws/stepfunctions/cloudformation/statemachinealias/CallbackContext.java b/...in/java/com/amazonaws/stepfunctions/cloudformation/statemachinealias/CallbackContext.java
@@ -3,6 +3,7 @@
 import lombok.AllArgsConstructor;
 import lombok.NoArgsConstructor;
 import lombok.Builder;
+
 import java.time.Instant;
 
 import software.amazon.cloudformation.proxy.StdCallbackContext;

diff --git a/.../java/com/amazonaws/stepfunctions/cloudformation/statemachineversion/CallbackContext.java b/.../java/com/amazonaws/stepfunctions/cloudformation/statemachineversion/CallbackContext.java
@@ -4,12 +4,13 @@
 import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
+import software.amazon.cloudformation.proxy.StdCallbackContext;
 
 @Data
 @NoArgsConstructor
 @AllArgsConstructor
 @Builder
-public class CallbackContext {
- @Builder.Default
- private boolean deletionStarted = false;
+public class CallbackContext extends StdCallbackContext {
+ @Builder.Default
+ private boolean deletionStarted = false;
 }