Add DurableTaskGrpcClientFactory

azurefunctions/src/main/java/com/microsoft/durabletask/azurefunctions/DurableClientContext.java

@@ -8,7 +8,7 @@
 
 import com.microsoft.azure.functions.HttpStatus;
 import com.microsoft.durabletask.DurableTaskClient;
-import com.microsoft.durabletask.DurableTaskGrpcClientBuilder;
+import com.microsoft.durabletask.DurableTaskGrpcClientFactory;
 import com.microsoft.durabletask.OrchestrationMetadata;
 import com.microsoft.durabletask.OrchestrationRuntimeStatus;
 
@@ -45,6 +45,10 @@ public String getTaskHubName() {
      * @return the Durable Task client object associated with the current function invocation.
      */
     public DurableTaskClient getClient() {
+        if (this.client != null) {
+            return this.client;
+        }
+
         if (this.rpcBaseUrl == null || this.rpcBaseUrl.length() == 0) {
             throw new IllegalStateException("The client context wasn't populated with an RPC base URL!");
         }
@@ -56,7 +60,7 @@ public DurableTaskClient getClient() {
             throw new IllegalStateException("The client context RPC base URL was invalid!", ex);
         }
 
-        this.client = new DurableTaskGrpcClientBuilder().port(rpcURL.getPort()).build();
+        this.client = DurableTaskGrpcClientFactory.getClient(rpcURL.getPort(), null);
         return this.client;
     }
 
@@ -78,9 +82,7 @@ public HttpResponseMessage waitForCompletionOrCreateCheckStatusResponse(
             HttpRequestMessage<?> request,
             String instanceId,
             Duration timeout) {
-        if (this.client == null) {
-            this.client = getClient();
-        }
+        this.client = getClient();
         OrchestrationMetadata orchestration;
         try {
             orchestration = this.client.waitForInstanceCompletion(instanceId, timeout, true);

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

@@ -61,6 +61,18 @@ public final class DurableTaskGrpcClient extends DurableTaskClient {
         this.sidecarClient = TaskHubSidecarServiceGrpc.newBlockingStub(sidecarGrpcChannel);
     }
 
+    DurableTaskGrpcClient(int port, String defaultVersion) {
+        this.dataConverter = new JacksonDataConverter();
+        this.defaultVersion = defaultVersion;
+
+        // Need to keep track of this channel so we can dispose it on close()
+        this.managedSidecarChannel = ManagedChannelBuilder
+                .forAddress("localhost", port)
+                .usePlaintext()
+                .build();
+        this.sidecarClient = TaskHubSidecarServiceGrpc.newBlockingStub(this.managedSidecarChannel);
+    }
+
     /**
      * Closes the internally managed gRPC channel, if one exists.
      * <p>

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

@@ -0,0 +1,15 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.microsoft.durabletask;
+
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+
+public final class DurableTaskGrpcClientFactory {
+    private static final ConcurrentMap<Integer, DurableTaskGrpcClient> portToClientMap = new ConcurrentHashMap<>();
+
+    public static DurableTaskClient getClient(int port, String defaultVersion) {
+        return portToClientMap.computeIfAbsent(port, DurableTaskGrpcClient::new);
+    }
+}