Add support for workflow init in Springboot

Author: Quinn-With-Two-Ns

temporal-sdk/src/main/java/io/temporal/internal/sync/POJOWorkflowImplementationFactory.java

@@ -91,7 +91,7 @@ public final class POJOWorkflowImplementationFactory implements ReplayWorkflowFa
       workflowDefinitions = Collections.synchronizedMap(new HashMap<>());
 
   /** Factories providing instances of workflow classes. */
-  private final Map<Class<?>, Functions.Func<?>> workflowInstanceFactories =
+  private final Map<Class<?>, Functions.Func1<EncodedValues, ?>> workflowInstanceFactories =
       Collections.synchronizedMap(new HashMap<>());
 
   /** If present then it is called for any unknown workflow type. */
@@ -146,7 +146,9 @@ public void registerWorkflowImplementationTypes(
    */
   @SuppressWarnings("unchecked")
   public <R> void addWorkflowImplementationFactory(
-      WorkflowImplementationOptions options, Class<R> clazz, Functions.Func<R> factory) {
+      WorkflowImplementationOptions options,
+      Class<R> clazz,
+      Functions.Func1<EncodedValues, R> factory) {
     if (DynamicWorkflow.class.isAssignableFrom(clazz)) {
       if (dynamicWorkflowImplementationFactory != null) {
         throw new TypeAlreadyRegisteredException(
@@ -433,16 +435,17 @@ public WorkflowOutput execute(WorkflowInput input) {
       }
 
       protected void newInstance(Optional<Payloads> input) {
-        Func<?> factory = workflowInstanceFactories.get(workflowImplementationClass);
+        Functions.Func1<EncodedValues, ?> factory =
+            workflowInstanceFactories.get(workflowImplementationClass);
         if (factory != null) {
-          workflow = factory.apply();
+          workflow = factory.apply(new EncodedValues(input, dataConverterWithWorkflowContext));
         } else {
           // Historically any exception thrown from the constructor was wrapped into Error causing a
           // workflow task failure.
           // This is not consistent with throwing exception from the workflow method which can
           // causes a workflow failure depending on the exception type.
           // To preserve backwards compatibility we only change behaviour if a constructor is
-          // annotated with WorkflowInit.
+          // annotated with @WorkflowInit.
           if (ctor != null) {
             try {
               workflow =

temporal-sdk/src/main/java/io/temporal/internal/worker/SyncWorkflowWorker.java

@@ -26,6 +26,7 @@
 import io.temporal.api.taskqueue.v1.TaskQueue;
 import io.temporal.client.WorkflowClient;
 import io.temporal.common.converter.DataConverter;
+import io.temporal.common.converter.EncodedValues;
 import io.temporal.internal.activity.ActivityExecutionContextFactory;
 import io.temporal.internal.activity.ActivityTaskHandlerImpl;
 import io.temporal.internal.activity.LocalActivityExecutionContextFactoryImpl;
@@ -38,6 +39,7 @@
 import io.temporal.worker.tuning.SlotSupplier;
 import io.temporal.worker.tuning.WorkflowSlotInfo;
 import io.temporal.workflow.Functions.Func;
+import io.temporal.workflow.Functions.Func1;
 import java.lang.reflect.Type;
 import java.time.Duration;
 import java.util.Objects;
@@ -165,6 +167,11 @@ public void registerWorkflowImplementationTypes(
 
   public <R> void registerWorkflowImplementationFactory(
       WorkflowImplementationOptions options, Class<R> clazz, Func<R> factory) {
+    this.factory.addWorkflowImplementationFactory(options, clazz, unused -> factory.apply());
+  }
+
+  public <R> void registerWorkflowImplementationFactory(
+      WorkflowImplementationOptions options, Class<R> clazz, Func1<EncodedValues, R> factory) {
     this.factory.addWorkflowImplementationFactory(options, clazz, factory);
   }
 

temporal-sdk/src/main/java/io/temporal/worker/Worker.java

@@ -31,12 +31,14 @@
 import io.temporal.common.WorkflowExecutionHistory;
 import io.temporal.common.context.ContextPropagator;
 import io.temporal.common.converter.DataConverter;
+import io.temporal.common.converter.EncodedValues;
 import io.temporal.failure.TemporalFailure;
 import io.temporal.internal.sync.WorkflowInternal;
 import io.temporal.internal.sync.WorkflowThreadExecutor;
 import io.temporal.internal.worker.*;
 import io.temporal.serviceclient.MetricsTag;
 import io.temporal.worker.tuning.*;
+import io.temporal.workflow.Functions;
 import io.temporal.workflow.Functions.Func;
 import io.temporal.workflow.WorkflowMethod;
 import java.time.Duration;
@@ -325,6 +327,14 @@ public <R> void registerWorkflowImplementationFactory(
     workflowWorker.registerWorkflowImplementationFactory(options, workflowInterface, factory);
   }
 
+  @VisibleForTesting
+  public <R> void registerWorkflowImplementationFactory(
+      Class<R> workflowInterface,
+      Functions.Func1<EncodedValues, R> factory,
+      WorkflowImplementationOptions options) {
+    workflowWorker.registerWorkflowImplementationFactory(options, workflowInterface, factory);
+  }
+
   /**
    * Configures a factory to use when an instance of a workflow implementation is created.
    *

temporal-spring-boot-autoconfigure/build.gradle

@@ -34,6 +34,12 @@ dependencies {
     testImplementation group: 'ch.qos.logback', name: 'logback-classic', version: "${logbackVersion}"
 
     testImplementation "org.springframework.boot:spring-boot-starter-test"
+
+    testImplementation('org.slf4j:slf4j-api') {
+        version {
+            strictly "${slf4jVersion}"
+        }
+    }
 }
 
 tasks.test {

temporal-spring-boot-autoconfigure/src/main/java/io/temporal/spring/boot/autoconfigure/template/WorkersTemplate.java

@@ -37,6 +37,8 @@
 import io.temporal.spring.boot.autoconfigure.properties.NamespaceProperties;
 import io.temporal.spring.boot.autoconfigure.properties.WorkerProperties;
 import io.temporal.worker.*;
+import java.lang.reflect.Constructor;
+import java.lang.reflect.InvocationTargetException;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.HashMap;
@@ -510,7 +512,6 @@ private void configureWorkflowImplementationAutoDiscovery(
 
   @SuppressWarnings("unchecked")
   private <T> void configureWorkflowImplementation(Worker worker, Class<?> clazz) {
-
     POJOWorkflowImplMetadata workflowMetadata =
         POJOWorkflowImplMetadata.newInstanceForWorkflowFactory(clazz);
     List<POJOWorkflowMethodMetadata> workflowMethods = workflowMetadata.getWorkflowMethods();
@@ -524,10 +525,15 @@ private <T> void configureWorkflowImplementation(Worker worker, Class<?> clazz)
     WorkflowImplementationOptions workflowImplementationOptions =
         new WorkflowImplementationOptionsTemplate(workflowImplementationCustomizer)
             .createWorkflowImplementationOptions();
-
     WorkerDeploymentOptions deploymentOptions = worker.getWorkerOptions().getDeploymentOptions();
-
-    for (POJOWorkflowMethodMetadata workflowMethod : workflowMetadata.getWorkflowMethods()) {
+    if (workflowMetadata.getWorkflowInit() != null) {
+      if (workflowMethods.size() > 1) {
+        throw new BeanDefinitionValidationException(
+            "Workflow implementation class "
+                + clazz
+                + " has more then one workflow method and a constructor annotated with @WorkflowInit.");
+      }
+      POJOWorkflowMethodMetadata workflowMethod = workflowMetadata.getWorkflowMethods().get(0);
       if (deploymentOptions != null && deploymentOptions.isUsingVersioning()) {
         POJOWorkflowImplementationFactory.validateVersioningBehavior(
             clazz,
@@ -538,10 +544,43 @@ private <T> void configureWorkflowImplementation(Worker worker, Class<?> clazz)
 
       worker.registerWorkflowImplementationFactory(
           (Class<T>) workflowMethod.getWorkflowInterface(),
-          () -> (T) beanFactory.createBean(clazz),
+          (encodedValues) -> {
+            try {
+              Constructor<?> ctor = workflowMetadata.getWorkflowInit();
+              Object[] parameters = new Object[ctor.getParameterCount()];
+              for (int i = 0; i < ctor.getParameterCount(); i++) {
+                parameters[i] =
+                    encodedValues.get(
+                        i, ctor.getParameterTypes()[i], ctor.getGenericParameterTypes()[i]);
+              }
+              T workflowInstance = (T) workflowMetadata.getWorkflowInit().newInstance(parameters);
+              beanFactory.autowireBean(workflowInstance);
+              return workflowInstance;
+            } catch (InstantiationException
+                | IllegalAccessException
+                | InvocationTargetException e) {
+              throw new RuntimeException(e);
+            }
+          },
           workflowImplementationOptions);
       addRegisteredWorkflowImpl(
           worker, workflowMethod.getWorkflowInterface().getName(), workflowMetadata);
+    } else {
+      for (POJOWorkflowMethodMetadata workflowMethod : workflowMetadata.getWorkflowMethods()) {
+        if (deploymentOptions != null && deploymentOptions.isUsingVersioning()) {
+          POJOWorkflowImplementationFactory.validateVersioningBehavior(
+              clazz,
+              workflowMethod,
+              deploymentOptions.getDefaultVersioningBehavior(),
+              deploymentOptions.isUsingVersioning());
+        }
+        worker.registerWorkflowImplementationFactory(
+            (Class<T>) workflowMethod.getWorkflowInterface(),
+            () -> (T) beanFactory.createBean(clazz),
+            workflowImplementationOptions);
+        addRegisteredWorkflowImpl(
+            worker, workflowMethod.getWorkflowInterface().getName(), workflowMetadata);
+      }
     }
   }
 

temporal-spring-boot-autoconfigure/src/test/java/io/temporal/spring/boot/autoconfigure/bytaskqueue/TestWorkflowImpl.java

@@ -26,29 +26,44 @@
 import io.temporal.workflow.NexusOperationOptions;
 import io.temporal.workflow.NexusServiceOptions;
 import io.temporal.workflow.Workflow;
+import io.temporal.workflow.WorkflowInit;
 import java.time.Duration;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ConfigurableApplicationContext;
 
 @WorkflowImpl(taskQueues = {"${default-queue.name:UnitTest}"})
 public class TestWorkflowImpl implements TestWorkflow {
+  private final TestNexusService nexusService;
+  private final TestActivity activity;
+
+  @Autowired private ConfigurableApplicationContext applicationContext;
+
+  @WorkflowInit
+  public TestWorkflowImpl(String input) {
+    nexusService =
+        Workflow.newNexusServiceStub(
+            TestNexusService.class,
+            NexusServiceOptions.newBuilder()
+                .setEndpoint("AutoDiscoveryByTaskQueueEndpoint")
+                .setOperationOptions(
+                    NexusOperationOptions.newBuilder()
+                        .setScheduleToCloseTimeout(Duration.ofSeconds(10))
+                        .build())
+                .build());
+
+    activity =
+        Workflow.newActivityStub(
+            TestActivity.class,
+            ActivityOptions.newBuilder()
+                .setStartToCloseTimeout(Duration.ofSeconds(1))
+                .validateAndBuildWithDefaults());
+  }
+
   @Override
   public String execute(String input) {
     if (input.equals("nexus")) {
-      Workflow.newNexusServiceStub(
-              TestNexusService.class,
-              NexusServiceOptions.newBuilder()
-                  .setEndpoint("AutoDiscoveryByTaskQueueEndpoint")
-                  .setOperationOptions(
-                      NexusOperationOptions.newBuilder()
-                          .setScheduleToCloseTimeout(Duration.ofSeconds(10))
-                          .build())
-                  .build())
-          .operation(input);
+      nexusService.operation(input);
     }
-    return Workflow.newActivityStub(
-            TestActivity.class,
-            ActivityOptions.newBuilder()
-                .setStartToCloseTimeout(Duration.ofSeconds(1))
-                .validateAndBuildWithDefaults())
-        .execute("done");
+    return activity.execute("done");
   }
 }