update after saga context added

Signed-off-by: Sky Ao <[email protected]>
dapr · Jan 13, 2024 · e9d85a9 · e9d85a9
1 parent e786009
commit e9d85a9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.../order-processor-saga/src/main/java/io/dapr/quickstarts/saga/OrderProcessingWorkflow.java b/.../order-processor-saga/src/main/java/io/dapr/quickstarts/saga/OrderProcessingWorkflow.java
@@ -85,7 +85,7 @@ public WorkflowStub create() {
         ctx.complete(orderResult);
         return;
       }
-      ctx.registerCompensation(ProcessPaymentCompensationActivity.class.getName(), paymentRequest);
+      ctx.getSagaContext().registerCompensation(ProcessPaymentCompensationActivity.class.getName(), paymentRequest);
 
       // step5: Update the inventory (need compensation)
       inventoryResult = ctx.callActivity(UpdateInventoryActivity.class.getName(),
@@ -96,12 +96,12 @@ public WorkflowStub create() {
         ctx.callActivity(NotifyActivity.class.getName(), notification).await();
 
         // trigger saga compensation gracefully
-        ctx.compensate();
+        ctx.getSagaContext().compensate();
         orderResult.setCompensated(true);
         ctx.complete(orderResult);
         return;
       }
-      ctx.registerCompensation(UpdateInventoryCompensationActivity.class.getName(), inventoryRequest);
+      ctx.getSagaContext().registerCompensation(UpdateInventoryCompensationActivity.class.getName(), inventoryRequest);
 
       // step6: delevery (allways be failed to trigger compensation)
       ctx.callActivity(DeliveryActivity.class.getName()).await();