Add unimplemented DiagnoseWorkflowExecution methods (#1015)

c-warren · web-flow · commit 8652ad743229 · 2025-08-01T14:51:34.000-07:00
diff --git a/src/main/idls b/src/main/idls
@@ -1 +1 @@
-Subproject commit 83d5cae7fc5176f73486ffe82144044711930073
+Subproject commit f5ac3a987dcaf996cf44c72064195faf76c381f3
diff --git a/src/main/java/com/uber/cadence/internal/compatibility/Thrift2ProtoAdapter.java b/src/main/java/com/uber/cadence/internal/compatibility/Thrift2ProtoAdapter.java
@@ -31,6 +31,8 @@
 import com.uber.cadence.DescribeTaskListResponse;
 import com.uber.cadence.DescribeWorkflowExecutionRequest;
 import com.uber.cadence.DescribeWorkflowExecutionResponse;
+import com.uber.cadence.DiagnoseWorkflowExecutionRequest;
+import com.uber.cadence.DiagnoseWorkflowExecutionResponse;
 import com.uber.cadence.DomainAlreadyExistsError;
 import com.uber.cadence.DomainNotActiveError;
 import com.uber.cadence.EntityNotExistsError;
@@ -156,6 +158,14 @@ public DescribeDomainResponse DescribeDomain(DescribeDomainRequest describeReque
     }
   }
 
+  @Override
+  public DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest)
+      throws DomainNotActiveError, ServiceBusyError, EntityNotExistsError,
+          ClientVersionNotSupportedError, TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public ListDomainsResponse ListDomains(ListDomainsRequest listRequest)
       throws BadRequestError, EntityNotExistsError, ServiceBusyError,
@@ -829,6 +839,13 @@ public void DescribeDomain(
     throw new UnsupportedOperationException("not implemented");
   }
 
+  @Override
+  public void DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest, AsyncMethodCallback resultHandler)
+      throws TException {
+    throw new UnsupportedOperationException("not implemented");
+  }
+
   @Override
   public void ListDomains(ListDomainsRequest listRequest, AsyncMethodCallback resultHandler)
       throws TException {
diff --git a/src/main/java/com/uber/cadence/internal/compatibility/proto/RequestMapper.java b/src/main/java/com/uber/cadence/internal/compatibility/proto/RequestMapper.java
@@ -29,6 +29,7 @@
 import static com.uber.cadence.internal.compatibility.proto.Helpers.newFieldMask;
 import static com.uber.cadence.internal.compatibility.proto.Helpers.nullToEmpty;
 import static com.uber.cadence.internal.compatibility.proto.Helpers.secondsToDuration;
+import static com.uber.cadence.internal.compatibility.proto.Helpers.unixNanoToTime;
 import static com.uber.cadence.internal.compatibility.proto.TypeMapper.badBinaries;
 import static com.uber.cadence.internal.compatibility.proto.TypeMapper.clusterReplicationConfigurationArray;
 import static com.uber.cadence.internal.compatibility.proto.TypeMapper.failure;
@@ -403,6 +404,7 @@ public static DescribeWorkflowExecutionRequest describeWorkflowExecutionRequest(
     return DescribeWorkflowExecutionRequest.newBuilder()
         .setDomain(t.getDomain())
         .setWorkflowExecution(workflowExecution(t.getExecution()))
+        .setQueryConsistencyLevel(queryConsistencyLevel(t.getQueryConsistencyLevel()))
         .build();
   }
 
@@ -418,7 +420,8 @@ public static GetWorkflowExecutionHistoryRequest getWorkflowExecutionHistoryRequ
             .setPageSize(t.getMaximumPageSize())
             .setWaitForNewEvent(t.isWaitForNewEvent())
             .setHistoryEventFilterType(eventFilterType(t.HistoryEventFilterType))
-            .setSkipArchival(t.isSkipArchival());
+            .setSkipArchival(t.isSkipArchival())
+            .setQueryConsistencyLevel(queryConsistencyLevel(t.getQueryConsistencyLevel()));
     if (t.getNextPageToken() != null) {
       builder.setNextPageToken(arrayToByteString(t.getNextPageToken()));
     }
@@ -445,6 +448,9 @@ public static SignalWithStartWorkflowExecutionRequest signalWithStartWorkflowExe
             .setSearchAttributes(searchAttributes(t.getSearchAttributes()))
             .setHeader(header(t.getHeader()))
             .setJitterStart(secondsToDuration(t.getJitterStartSeconds()));
+    if (t.isSetFirstRunAtTimestamp()) {
+      builder.setFirstRunAt(unixNanoToTime(t.getFirstRunAtTimestamp()));
+    }
     if (t.getRetryPolicy() != null) {
       builder.setRetryPolicy(retryPolicy(t.getRetryPolicy()));
     }
@@ -530,6 +536,9 @@ public static StartWorkflowExecutionRequest startWorkflowExecutionRequest(
             .setHeader(header(t.getHeader()))
             .setDelayStart(secondsToDuration(t.getDelayStartSeconds()))
             .setJitterStart(secondsToDuration(t.getJitterStartSeconds()));
+    if (t.isSetFirstRunAtTimestamp()) {
+      request.setFirstRunAt(unixNanoToTime(t.getFirstRunAtTimestamp()));
+    }
     if (t.getRetryPolicy() != null) {
       request.setRetryPolicy(retryPolicy(t.getRetryPolicy()));
     }
diff --git a/src/main/java/com/uber/cadence/internal/compatibility/thrift/HistoryMapper.java b/src/main/java/com/uber/cadence/internal/compatibility/thrift/HistoryMapper.java
@@ -947,6 +947,8 @@ static MarkerRecordedEventAttributes markerRecordedEventAttributes(
     res.setMemo(memo(t.getMemo()));
     res.setSearchAttributes(searchAttributes(t.getSearchAttributes()));
     res.setDelayStartSeconds(durationToSeconds(t.getDelayStart()));
+    res.setJitterStartSeconds(durationToSeconds(t.getJitterStart()));
+    res.setFirstRunAtTimestamp(timeToUnixNano(t.getFirstRunAt()));
     return res;
   }
 
diff --git a/src/main/java/com/uber/cadence/internal/compatibility/thrift/ResponseMapper.java b/src/main/java/com/uber/cadence/internal/compatibility/thrift/ResponseMapper.java
@@ -25,6 +25,7 @@
 import static com.uber.cadence.internal.compatibility.thrift.HistoryMapper.history;
 import static com.uber.cadence.internal.compatibility.thrift.TypeMapper.activityLocalDispatchInfoMap;
 import static com.uber.cadence.internal.compatibility.thrift.TypeMapper.activityType;
+import static com.uber.cadence.internal.compatibility.thrift.TypeMapper.autoConfigHint;
 import static com.uber.cadence.internal.compatibility.thrift.TypeMapper.badBinaries;
 import static com.uber.cadence.internal.compatibility.thrift.TypeMapper.clusterReplicationConfigurationArray;
 import static com.uber.cadence.internal.compatibility.thrift.TypeMapper.dataBlobArray;
@@ -239,6 +240,7 @@ public static PollForActivityTaskResponse pollForActivityTaskResponse(
     res.setWorkflowType(workflowType(t.getWorkflowType()));
     res.setWorkflowDomain(t.getWorkflowDomain());
     res.setHeader(header(t.getHeader()));
+    res.setAutoConfigHint(autoConfigHint(t.getAutoConfigHint()));
     return res;
   }
 
@@ -265,6 +267,7 @@ public static PollForDecisionTaskResponse pollForDecisionTaskResponse(
     res.setStartedTimestamp(timeToUnixNano(t.getStartedTime()));
     res.setQueries(workflowQueryMap(t.getQueriesMap()));
     res.setNextEventId(t.getNextEventId());
+    res.setAutoConfigHint(autoConfigHint(t.getAutoConfigHint()));
     return res;
   }
 
diff --git a/src/main/java/com/uber/cadence/internal/compatibility/thrift/TypeMapper.java b/src/main/java/com/uber/cadence/internal/compatibility/thrift/TypeMapper.java
@@ -31,6 +31,7 @@
 
 import com.uber.cadence.ActivityLocalDispatchInfo;
 import com.uber.cadence.ActivityType;
+import com.uber.cadence.AutoConfigHint;
 import com.uber.cadence.BadBinaries;
 import com.uber.cadence.BadBinaryInfo;
 import com.uber.cadence.ClusterReplicationConfiguration;
@@ -686,4 +687,14 @@ static Map<String, ActivityLocalDispatchInfo> activityLocalDispatchInfoMap(
     }
     return v;
   }
+
+  static AutoConfigHint autoConfigHint(com.uber.cadence.api.v1.AutoConfigHint t) {
+    if (t == null) {
+      return null;
+    }
+    AutoConfigHint autoConfigHint = new AutoConfigHint();
+    autoConfigHint.setEnableAutoConfig(t.getEnableAutoConfig());
+    autoConfigHint.setPollerWaitTimeInMs(t.getPollerWaitTimeInMs());
+    return autoConfigHint;
+  }
 }
diff --git a/src/main/java/com/uber/cadence/internal/sync/TestActivityEnvironmentInternal.java b/src/main/java/com/uber/cadence/internal/sync/TestActivityEnvironmentInternal.java
@@ -593,6 +593,13 @@ public void DescribeDomain(
       impl.DescribeDomain(describeRequest, resultHandler);
     }
 
+    @Override
+    public void DiagnoseWorkflowExecution(
+        DiagnoseWorkflowExecutionRequest diagnoseRequest, AsyncMethodCallback resultHandler)
+        throws TException {
+      impl.DiagnoseWorkflowExecution(diagnoseRequest, resultHandler);
+    }
+
     @Override
     public void ListDomains(ListDomainsRequest listRequest, AsyncMethodCallback resultHandler)
         throws TException {
@@ -921,6 +928,14 @@ public DescribeDomainResponse DescribeDomain(DescribeDomainRequest describeReque
       return impl.DescribeDomain(describeRequest);
     }
 
+    @Override
+    public DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(
+        DiagnoseWorkflowExecutionRequest diagnoseRequest)
+        throws DomainNotActiveError, ServiceBusyError, EntityNotExistsError,
+            ClientVersionNotSupportedError, TException {
+      return impl.DiagnoseWorkflowExecution(diagnoseRequest);
+    }
+
     @Override
     public ListDomainsResponse ListDomains(ListDomainsRequest listRequest)
         throws BadRequestError, InternalServiceError, EntityNotExistsError, ServiceBusyError,
diff --git a/src/main/java/com/uber/cadence/internal/sync/TestWorkflowEnvironmentInternal.java b/src/main/java/com/uber/cadence/internal/sync/TestWorkflowEnvironmentInternal.java
@@ -29,6 +29,8 @@
 import com.uber.cadence.DescribeTaskListResponse;
 import com.uber.cadence.DescribeWorkflowExecutionRequest;
 import com.uber.cadence.DescribeWorkflowExecutionResponse;
+import com.uber.cadence.DiagnoseWorkflowExecutionRequest;
+import com.uber.cadence.DiagnoseWorkflowExecutionResponse;
 import com.uber.cadence.DomainAlreadyExistsError;
 import com.uber.cadence.DomainNotActiveError;
 import com.uber.cadence.EntityNotExistsError;
@@ -475,6 +477,13 @@ public void DescribeDomain(
       impl.DescribeDomain(describeRequest, resultHandler);
     }
 
+    @Override
+    public void DiagnoseWorkflowExecution(
+        DiagnoseWorkflowExecutionRequest diagnoseRequest, AsyncMethodCallback resultHandler)
+        throws TException {
+      throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+    }
+
     @Override
     public void ListDomains(ListDomainsRequest listRequest, AsyncMethodCallback resultHandler)
         throws TException {
@@ -809,6 +818,14 @@ public DescribeDomainResponse DescribeDomain(DescribeDomainRequest describeReque
       return impl.DescribeDomain(describeRequest);
     }
 
+    @Override
+    public DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(
+        DiagnoseWorkflowExecutionRequest diagnoseRequest)
+        throws DomainNotActiveError, ServiceBusyError, EntityNotExistsError,
+            ClientVersionNotSupportedError, TException {
+      throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+    }
+
     @Override
     public ListDomainsResponse ListDomains(ListDomainsRequest listRequest)
         throws BadRequestError, InternalServiceError, EntityNotExistsError, ServiceBusyError,
diff --git a/src/main/java/com/uber/cadence/internal/testservice/TestWorkflowService.java b/src/main/java/com/uber/cadence/internal/testservice/TestWorkflowService.java
@@ -29,6 +29,8 @@
 import com.uber.cadence.DescribeTaskListResponse;
 import com.uber.cadence.DescribeWorkflowExecutionRequest;
 import com.uber.cadence.DescribeWorkflowExecutionResponse;
+import com.uber.cadence.DiagnoseWorkflowExecutionRequest;
+import com.uber.cadence.DiagnoseWorkflowExecutionResponse;
 import com.uber.cadence.DomainAlreadyExistsError;
 import com.uber.cadence.DomainNotActiveError;
 import com.uber.cadence.EntityNotExistsError;
@@ -206,6 +208,14 @@ public DescribeDomainResponse DescribeDomain(DescribeDomainRequest describeReque
     throw new UnsupportedOperationException("not implemented");
   }
 
+  @Override
+  public DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest)
+      throws DomainNotActiveError, ServiceBusyError, EntityNotExistsError,
+          ClientVersionNotSupportedError, TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public ListDomainsResponse ListDomains(ListDomainsRequest listRequest)
       throws BadRequestError, InternalServiceError, EntityNotExistsError, ServiceBusyError,
@@ -834,6 +844,13 @@ public void DescribeDomain(
     throw new UnsupportedOperationException("not implemented");
   }
 
+  @Override
+  public void DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest, AsyncMethodCallback resultHandler)
+      throws TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public void ListDomains(ListDomainsRequest listRequest, AsyncMethodCallback resultHandler)
       throws TException {
diff --git a/src/main/java/com/uber/cadence/serviceclient/IWorkflowServiceBase.java b/src/main/java/com/uber/cadence/serviceclient/IWorkflowServiceBase.java
@@ -43,6 +43,14 @@ public DescribeDomainResponse DescribeDomain(DescribeDomainRequest describeReque
     throw new UnsupportedOperationException("unimplemented");
   }
 
+  @Override
+  public DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest)
+      throws DomainNotActiveError, ServiceBusyError, EntityNotExistsError,
+          ClientVersionNotSupportedError, TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public ListDomainsResponse ListDomains(ListDomainsRequest listRequest)
       throws BadRequestError, EntityNotExistsError, ServiceBusyError,
@@ -377,6 +385,13 @@ public void DescribeDomain(
     throw new UnsupportedOperationException("unimplemented");
   }
 
+  @Override
+  public void DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest, AsyncMethodCallback resultHandler)
+      throws TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public void ListDomains(ListDomainsRequest listRequest, AsyncMethodCallback resultHandler)
       throws TException {
diff --git a/src/main/java/com/uber/cadence/serviceclient/WorkflowServiceTChannel.java b/src/main/java/com/uber/cadence/serviceclient/WorkflowServiceTChannel.java
@@ -443,6 +443,14 @@ private DescribeDomainResponse describeDomain(DescribeDomainRequest describeRequ
     }
   }
 
+  @Override
+  public DiagnoseWorkflowExecutionResponse DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest)
+      throws DomainNotActiveError, ServiceBusyError, EntityNotExistsError,
+          ClientVersionNotSupportedError, TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public ListDomainsResponse ListDomains(ListDomainsRequest listRequest)
       throws BadRequestError, InternalServiceError, EntityNotExistsError, ServiceBusyError,
@@ -2937,6 +2945,13 @@ public void DescribeDomain(
     throw new UnsupportedOperationException("not implemented");
   }
 
+  @Override
+  public void DiagnoseWorkflowExecution(
+      DiagnoseWorkflowExecutionRequest diagnoseRequest, AsyncMethodCallback resultHandler)
+      throws TException {
+    throw new UnsupportedOperationException("DiagnoseWorkflowExecution is not implemented");
+  }
+
   @Override
   public void ListDomains(ListDomainsRequest listRequest, AsyncMethodCallback resultHandler)
       throws TException {
diff --git a/src/test/java/com/uber/cadence/internal/compatibility/ProtoObjects.java b/src/test/java/com/uber/cadence/internal/compatibility/ProtoObjects.java
@@ -80,6 +80,8 @@ public final class ProtoObjects {
   public static final Header HEADER =
       Header.newBuilder().putFields("key", payload("value")).build();
   public static final Memo MEMO = Memo.newBuilder().putFields("memo", payload("memoValue")).build();
+  public static final AutoConfigHint AUTO_CONFIG_HINT =
+      AutoConfigHint.newBuilder().setEnableAutoConfig(true).setPollerWaitTimeInMs(100).build();
   public static final SearchAttributes SEARCH_ATTRIBUTES =
       SearchAttributes.newBuilder().putIndexedFields("search", payload("attributes")).build();
   public static final Map<String, String> DATA = ImmutableMap.of("dataKey", "dataValue");
@@ -629,6 +631,8 @@ public final class ProtoObjects {
               .setMemo(MEMO)
               .setSearchAttributes(SEARCH_ATTRIBUTES)
               .setDelayStart(seconds(4))
+              .setJitterStart(seconds(5))
+              .setFirstRunAt(timestampNanos(123456789))
               .build();
 
   public static final StartChildWorkflowExecutionFailedEventAttributes
@@ -984,6 +988,7 @@ public final class ProtoObjects {
       DescribeWorkflowExecutionRequest.newBuilder()
           .setDomain("domain")
           .setWorkflowExecution(WORKFLOW_EXECUTION)
+          .setQueryConsistencyLevel(QueryConsistencyLevel.QUERY_CONSISTENCY_LEVEL_STRONG)
           .build();
 
   public static final GetWorkflowExecutionHistoryRequest GET_WORKFLOW_EXECUTION_HISTORY_REQUEST =
@@ -995,6 +1000,7 @@ public final class ProtoObjects {
           .setHistoryEventFilterType(EventFilterType.EVENT_FILTER_TYPE_CLOSE_EVENT)
           .setSkipArchival(true)
           .setNextPageToken(utf8("nextPageToken"))
+          .setQueryConsistencyLevel(QueryConsistencyLevel.QUERY_CONSISTENCY_LEVEL_STRONG)
           .build();
 
   public static final StartWorkflowExecutionRequest START_WORKFLOW_EXECUTION =
@@ -1016,6 +1022,7 @@ public final class ProtoObjects {
           .setHeader(HEADER)
           .setDelayStart(seconds(3))
           .setJitterStart(seconds(0))
+          .setFirstRunAt(timestampNanos(123456789))
           .build();
 
   public static final SignalWithStartWorkflowExecutionRequest SIGNAL_WITH_START_WORKFLOW_EXECUTION =
@@ -1301,6 +1308,7 @@ public final class ProtoObjects {
           .setWorkflowType(WORKFLOW_TYPE)
           .setWorkflowDomain("domain")
           .setHeader(HEADER)
+          .setAutoConfigHint(AUTO_CONFIG_HINT)
           .build();
   public static final PollForDecisionTaskResponse POLL_FOR_DECISION_TASK_RESPONSE =
       PollForDecisionTaskResponse.newBuilder()
@@ -1319,6 +1327,7 @@ public final class ProtoObjects {
           .setStartedTime(timestampNanos(6))
           .putAllQueries(ImmutableMap.of("query", WORKFLOW_QUERY))
           .setNextEventId(7)
+          .setAutoConfigHint(AUTO_CONFIG_HINT)
           .build();
 
   public static final QueryWorkflowResponse QUERY_WORKFLOW_RESPONSE =
diff --git a/src/test/java/com/uber/cadence/internal/compatibility/Thrift2ProtoAdapterTest.java b/src/test/java/com/uber/cadence/internal/compatibility/Thrift2ProtoAdapterTest.java
@@ -786,9 +786,11 @@ public void testIsHealthy() throws Exception {
   @Test
   public void testAsyncUnsupported() {
     assertUnsupported(WorkflowService.Iface::RestartWorkflowExecution);
+    assertUnsupported(WorkflowService.Iface::DiagnoseWorkflowExecution);
     assertUnsupported(WorkflowService.Iface::GetTaskListsByDomain);
     assertAsyncUnsupported(WorkflowService.AsyncIface::RegisterDomain);
     assertAsyncUnsupported(WorkflowService.AsyncIface::DescribeDomain);
+    assertAsyncUnsupported(WorkflowService.AsyncIface::DiagnoseWorkflowExecution);
     assertAsyncUnsupported(WorkflowService.AsyncIface::ListDomains);
     assertAsyncUnsupported(WorkflowService.AsyncIface::UpdateDomain);
     assertAsyncUnsupported(WorkflowService.AsyncIface::DeprecateDomain);
diff --git a/src/test/java/com/uber/cadence/internal/compatibility/ThriftObjects.java b/src/test/java/com/uber/cadence/internal/compatibility/ThriftObjects.java
diff --git a/src/test/java/com/uber/cadence/internal/compatibility/thrift/HistoryMapperEventTest.java b/src/test/java/com/uber/cadence/internal/compatibility/thrift/HistoryMapperEventTest.java
diff --git a/src/test/java/com/uber/cadence/internal/testservice/TestWorkflowServiceDiagnoseTest.java b/src/test/java/com/uber/cadence/internal/testservice/TestWorkflowServiceDiagnoseTest.java
diff --git a/src/test/java/com/uber/cadence/serviceclient/IWorkflowServiceBaseTest.java b/src/test/java/com/uber/cadence/serviceclient/IWorkflowServiceBaseTest.java
diff --git a/src/test/java/com/uber/cadence/serviceclient/WorkflowServiceTChannelDiagnoseTest.java b/src/test/java/com/uber/cadence/serviceclient/WorkflowServiceTChannelDiagnoseTest.java

Original file line number	Diff line number	Diff line change
`@@ -947,6 +947,8 @@ static MarkerRecordedEventAttributes markerRecordedEventAttributes(`
`947`	`947`	`res.setMemo(memo(t.getMemo()));`
`948`	`948`	`res.setSearchAttributes(searchAttributes(t.getSearchAttributes()));`
`949`	`949`	`res.setDelayStartSeconds(durationToSeconds(t.getDelayStart()));`
	`950`	`+ res.setJitterStartSeconds(durationToSeconds(t.getJitterStart()));`
	`951`	`+ res.setFirstRunAtTimestamp(timeToUnixNano(t.getFirstRunAt()));`
`950`	`952`	`return res;`
`951`	`953`	`}`
`952`	`954`