Add identity to WorkflowOptions (#2080)

Provides the ability to override identity specified in a WorkflowClient

Author: Duzhinsky

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

@@ -594,9 +594,15 @@ private static SingleWorkerOptions.Builder toSingleWorkerOptions(
     } else if (clientOptions.getBinaryChecksum() != null) {
       buildId = clientOptions.getBinaryChecksum();
     }
+
+    String identity = clientOptions.getIdentity();
+    if (options.getIdentity() != null) {
+      identity = options.getIdentity();
+    }
+
     return SingleWorkerOptions.newBuilder()
         .setDataConverter(clientOptions.getDataConverter())
-        .setIdentity(clientOptions.getIdentity())
+        .setIdentity(identity)
         .setBuildId(buildId)
         .setUseBuildIdForVersioning(options.isUsingBuildIdForVersioning())
         .setEnableLoggingInReplay(factoryOptions.isEnableLoggingInReplay())

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

@@ -86,6 +86,7 @@ public static final class Builder {
     private SlotSupplier<WorkflowSlotInfo> workflowSlotSupplier;
     private SlotSupplier<ActivitySlotInfo> activitySlotSupplier;
     private SlotSupplier<LocalActivitySlotInfo> localActivitySlotSupplier;
+    private String identity;
 
     private Builder() {}
 
@@ -416,6 +417,12 @@ public void setLocalActivitySlotSupplier(
       this.localActivitySlotSupplier = localActivitySlotSupplier;
     }
 
+    /** Override identity of the worker primary specified in a WorkflowClient options. */
+    public Builder setIdentity(String identity) {
+      this.identity = identity;
+      return this;
+    }
+
     public WorkerOptions build() {
       return new WorkerOptions(
           maxWorkerActivitiesPerSecond,
@@ -436,7 +443,8 @@ public WorkerOptions build() {
           disableEagerExecution,
           useBuildIdForVersioning,
           buildId,
-          stickyTaskQueueDrainTimeout);
+          stickyTaskQueueDrainTimeout,
+          identity);
     }
 
     public WorkerOptions validateAndBuildWithDefaults() {
@@ -525,7 +533,8 @@ public WorkerOptions validateAndBuildWithDefaults() {
           buildId,
           stickyTaskQueueDrainTimeout == null
               ? DEFAULT_STICKY_TASK_QUEUE_DRAIN_TIMEOUT
-              : stickyTaskQueueDrainTimeout);
+              : stickyTaskQueueDrainTimeout,
+          identity);
     }
   }
 
@@ -548,6 +557,7 @@ public WorkerOptions validateAndBuildWithDefaults() {
   private final boolean useBuildIdForVersioning;
   private final String buildId;
   private final Duration stickyTaskQueueDrainTimeout;
+  private final String identity;
 
   private WorkerOptions(
       double maxWorkerActivitiesPerSecond,
@@ -568,7 +578,8 @@ private WorkerOptions(
       boolean disableEagerExecution,
       boolean useBuildIdForVersioning,
       String buildId,
-      Duration stickyTaskQueueDrainTimeout) {
+      Duration stickyTaskQueueDrainTimeout,
+      String identity) {
     this.maxWorkerActivitiesPerSecond = maxWorkerActivitiesPerSecond;
     this.maxConcurrentActivityExecutionSize = maxConcurrentActivityExecutionSize;
     this.maxConcurrentWorkflowTaskExecutionSize = maxConcurrentWorkflowTaskExecutionSize;
@@ -588,6 +599,7 @@ private WorkerOptions(
     this.useBuildIdForVersioning = useBuildIdForVersioning;
     this.buildId = buildId;
     this.stickyTaskQueueDrainTimeout = stickyTaskQueueDrainTimeout;
+    this.identity = identity;
   }
 
   public double getMaxWorkerActivitiesPerSecond() {
@@ -683,6 +695,11 @@ public SlotSupplier<LocalActivitySlotInfo> getLocalActivitySlotSupplier() {
     return localActivitySlotSupplier;
   }
 
+  @Nullable
+  public String getIdentity() {
+    return identity;
+  }
+
   @Override
   public boolean equals(Object o) {
     if (this == o) return true;
@@ -706,7 +723,8 @@ && compare(maxTaskQueueActivitiesPerSecond, that.maxTaskQueueActivitiesPerSecond
         && Objects.equals(defaultHeartbeatThrottleInterval, that.defaultHeartbeatThrottleInterval)
         && Objects.equals(stickyQueueScheduleToStartTimeout, that.stickyQueueScheduleToStartTimeout)
         && Objects.equals(buildId, that.buildId)
-        && Objects.equals(stickyTaskQueueDrainTimeout, that.stickyTaskQueueDrainTimeout);
+        && Objects.equals(stickyTaskQueueDrainTimeout, that.stickyTaskQueueDrainTimeout)
+        && Objects.equals(identity, that.identity);
   }
 
   @Override
@@ -730,7 +748,8 @@ public int hashCode() {
         disableEagerExecution,
         useBuildIdForVersioning,
         buildId,
-        stickyTaskQueueDrainTimeout);
+        stickyTaskQueueDrainTimeout,
+        identity);
   }
 
   @Override
@@ -775,6 +794,8 @@ public String toString() {
         + '\''
         + ", stickyTaskQueueDrainTimeout="
         + stickyTaskQueueDrainTimeout
+        + ", identity="
+        + identity
         + '}';
   }
 }