initial commit

Author: bachuv

client/src/main/java/com/microsoft/durabletask/DurableTaskGrpcClient.java

@@ -2,22 +2,30 @@
 // Licensed under the MIT License.
 package com.microsoft.durabletask;
 
-import com.google.protobuf.StringValue;
-import com.google.protobuf.Timestamp;
-import com.microsoft.durabletask.implementation.protobuf.OrchestratorService.*;
-import com.microsoft.durabletask.implementation.protobuf.TaskHubSidecarServiceGrpc;
-import com.microsoft.durabletask.implementation.protobuf.TaskHubSidecarServiceGrpc.*;
-
-import io.grpc.*;
-
-import javax.annotation.Nullable;
 import java.time.Duration;
 import java.time.Instant;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Optional;
+import java.util.UUID;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 import java.util.logging.Logger;
 
+import javax.annotation.Nullable;
+
+import com.google.protobuf.StringValue;
+import com.google.protobuf.Timestamp;
+import com.microsoft.durabletask.implementation.protobuf.OrchestratorService.*;
+import com.microsoft.durabletask.implementation.protobuf.TaskHubSidecarServiceGrpc;
+import com.microsoft.durabletask.implementation.protobuf.TaskHubSidecarServiceGrpc.TaskHubSidecarServiceBlockingStub;
+
+import io.grpc.Channel;
+import io.grpc.ManagedChannel;
+import io.grpc.ManagedChannelBuilder;
+import io.grpc.Status;
+import io.grpc.StatusRuntimeException;
+
 /**
  * Durable Task client implementation that uses gRPC to connect to a remote "sidecar" process.
  */
@@ -110,6 +118,16 @@ public String scheduleNewOrchestrationInstance(
             builder.setScheduledStartTimestamp(ts);
         }
 
+        String traceParent = options.getTraceParent();
+        String traceState = options.getTraceState();
+        if (traceParent != null) {
+            TraceContext traceContext = TraceContext.newBuilder()
+                .setTraceParent(traceParent)
+                .setTraceState(traceState != null ? StringValue.of(traceState) : StringValue.getDefaultInstance())
+                .build();
+            builder.setParentTraceContext(traceContext); // Set the TraceContext in the CreateInstanceRequest
+        }
+
         CreateInstanceRequest request = builder.build();
         CreateInstanceResponse response = this.sidecarClient.startInstance(request);
         return response.getInstanceId();

client/src/main/java/com/microsoft/durabletask/NewOrchestrationInstanceOptions.java

@@ -12,6 +12,8 @@ public final class NewOrchestrationInstanceOptions {
     private String instanceId;
     private Object input;
     private Instant startTime;
+    private String traceParent;
+    private String traceState;
 
     /**
      * Default constructor for the {@link NewOrchestrationInstanceOptions} class.
@@ -106,4 +108,20 @@ public Object getInput() {
     public Instant getStartTime() {
         return this.startTime;
     }
+
+    public String getTraceParent() {
+        return traceParent;
+    }
+
+    public void setTraceParent(String traceParent) {
+        this.traceParent = traceParent;
+    }
+
+    public String getTraceState() {
+        return traceState;
+    }
+
+    public void setTraceState(String traceState) {
+        this.traceState = traceState;
+    }
 }