query to verify

Author: mnichols

core/src/main/java/io/temporal/samples/updatewithstart/UpdateWithStartClient.java

@@ -23,6 +23,7 @@
 import io.temporal.api.enums.v1.WorkflowIdReusePolicy;
 import io.temporal.client.*;
 import io.temporal.serviceclient.WorkflowServiceStubs;
+import java.util.UUID;
 
 public class UpdateWithStartClient {
   private static final String TASK_QUEUE = "UpdateWithStartTQ";
@@ -46,11 +47,13 @@ private static void runWorkflowWithUpdateWithStart(
       WorkflowClient client, WorkflowOptions options) {
 
     var args = new StartWorkflowRequest();
+    args.setValue(UUID.randomUUID().toString());
 
     UpdateWithStartWorkflow workflow =
         client.newWorkflowStub(UpdateWithStartWorkflow.class, options);
 
     try {
+      // // First I tried this to updateWithStart
       //      var result =
       //          WorkflowClient.executeUpdateWithStart(
       //              workflow::putApplication,
@@ -63,17 +66,25 @@ private static void runWorkflowWithUpdateWithStart(
               workflow::putApplication,
               args,
               UpdateOptions.<UpdateWithStartWorkflowState>newBuilder()
-                  .setWaitForStage(WorkflowUpdateStage.ACCEPTED)
+                  .setWaitForStage(WorkflowUpdateStage.COMPLETED)
                   .build(),
               new WithStartWorkflowOperation<>(workflow::execute, args));
       var result = handle.getResult();
 
       System.out.println(
           "Workflow UwS with value: "
-              + result.getArgs().getValue()
+              + result.getInitArgs().getValue()
               + ", with updates count:"
               + result.getUpdates().size());
 
+      System.out.println(
+          "Workflow QUERY with initArgs: "
+              + workflow.getState().getInitArgs().getValue()
+              + "\n with updates: "
+              + workflow.getState().getUpdates().size()
+              + "\nwith execute args "
+              + workflow.getState().getExecuteArgs().getValue());
+
     } catch (WorkflowExecutionAlreadyStarted e) {
       System.err.println("WorkflowAlreadyStarted" + e);
     } catch (WorkflowServiceException e) {

core/src/main/java/io/temporal/samples/updatewithstart/UpdateWithStartWorkflow.java

@@ -19,6 +19,7 @@
 
 package io.temporal.samples.updatewithstart;
 
+import io.temporal.workflow.QueryMethod;
 import io.temporal.workflow.UpdateMethod;
 import io.temporal.workflow.WorkflowInterface;
 import io.temporal.workflow.WorkflowMethod;
@@ -30,4 +31,7 @@ public interface UpdateWithStartWorkflow {
 
   @UpdateMethod
   UpdateWithStartWorkflowState putApplication(StartWorkflowRequest args);
+
+  @QueryMethod
+  UpdateWithStartWorkflowState getState();
 }

core/src/main/java/io/temporal/samples/updatewithstart/UpdateWithStartWorkflowImpl.java

@@ -19,28 +19,43 @@
 
 package io.temporal.samples.updatewithstart;
 
+import io.temporal.workflow.Workflow;
 import io.temporal.workflow.WorkflowInit;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 public class UpdateWithStartWorkflowImpl implements UpdateWithStartWorkflow {
-  //  private static final Logger log = LoggerFactory.getLogger(UpdateWithStartWorkflowImpl.class);
+  private static final Logger log = LoggerFactory.getLogger(UpdateWithStartWorkflowImpl.class);
   private UpdateWithStartWorkflowState state;
 
   @WorkflowInit
   public UpdateWithStartWorkflowImpl(StartWorkflowRequest args) {
-    this.state = new UpdateWithStartWorkflowState(args);
+    this.state = new UpdateWithStartWorkflowState();
+    this.state.setInitArgs(args);
+    System.out.println("WorkflowInit args = " + args);
   }
 
   @Override
   public void execute(StartWorkflowRequest args) {
+    log.info("Workflow started {}", args);
+
+    this.state.setExecuteArgs(args);
     System.out.println("execute called");
-    //    log.info("Workflow started {}", args);
+
+    Workflow.await(() -> this.state.getUpdates().size() == 2);
   }
 
   @Override
   public UpdateWithStartWorkflowState putApplication(StartWorkflowRequest args) {
-    System.out.println("putApplication called");
+
     this.state.getUpdates().add(args);
+    System.out.println("put application called " + this.state.getUpdates().size());
+    return this.state;
+  }
 
+  @Override
+  public UpdateWithStartWorkflowState getState() {
+    System.out.println("getState called " + this.state.getInitArgs().getValue());
     return this.state;
   }
 }

core/src/main/java/io/temporal/samples/updatewithstart/UpdateWithStartWorkflowState.java

@@ -4,21 +4,18 @@
 import java.util.List;
 
 public class UpdateWithStartWorkflowState {
-  private StartWorkflowRequest args;
+  private StartWorkflowRequest initArgs;
+  private StartWorkflowRequest executeArgs;
   private List<StartWorkflowRequest> updates = new ArrayList<>();
 
   public UpdateWithStartWorkflowState() {}
 
-  public UpdateWithStartWorkflowState(StartWorkflowRequest args) {
-    this.args = args;
+  public StartWorkflowRequest getInitArgs() {
+    return initArgs;
   }
 
-  public StartWorkflowRequest getArgs() {
-    return args;
-  }
-
-  public void setArgs(StartWorkflowRequest args) {
-    this.args = args;
+  public void setInitArgs(StartWorkflowRequest initArgs) {
+    this.initArgs = initArgs;
   }
 
   public List<StartWorkflowRequest> getUpdates() {
@@ -28,4 +25,12 @@ public List<StartWorkflowRequest> getUpdates() {
   public void setUpdates(List<StartWorkflowRequest> updates) {
     this.updates = updates;
   }
+
+  public StartWorkflowRequest getExecuteArgs() {
+    return executeArgs;
+  }
+
+  public void setExecuteArgs(StartWorkflowRequest executeArgs) {
+    this.executeArgs = executeArgs;
+  }
 }