Expose OrchestratorServiceStubs

temporal-serviceclient/src/main/java/io/temporal/internal/testservice/GRPCServerHelper.java

@@ -0,0 +1,43 @@
+/*
+ *  Copyright (C) 2020 Temporal Technologies, Inc. All Rights Reserved.
+ *
+ *  Copyright 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ *  Modifications copyright (C) 2017 Uber Technologies, Inc.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License"). You may not
+ *  use this file except in compliance with the License. A copy of the License is
+ *  located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ *  or in the "license" file accompanying this file. This file is distributed on
+ *  an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ *  express or implied. See the License for the specific language governing
+ *  permissions and limitations under the License.
+ */
+
+package io.temporal.internal.testservice;
+
+import io.grpc.BindableService;
+import io.grpc.ServerBuilder;
+import io.grpc.ServerServiceDefinition;
+import io.grpc.health.v1.HealthCheckResponse;
+import io.grpc.protobuf.services.HealthStatusManager;
+import java.util.Collection;
+
+// TODO move to temporal-testing or temporal-test-server modules after WorkflowServiceStubs cleanup
+class GRPCServerHelper {
+  static void registerServicesAndHealthChecks(
+      Collection<BindableService> services, ServerBuilder<?> toServerBuilder) {
+    HealthStatusManager healthStatusManager = new HealthStatusManager();
+    for (BindableService service : services) {
+      ServerServiceDefinition serverServiceDefinition = service.bindService();
+      toServerBuilder.addService(serverServiceDefinition);
+      healthStatusManager.setStatus(
+          service.bindService().getServiceDescriptor().getName(),
+          HealthCheckResponse.ServingStatus.SERVING);
+    }
+    toServerBuilder.addService(healthStatusManager.getHealthService());
+  }
+}

temporal-serviceclient/src/main/java/io/temporal/internal/testservice/InProcessGRPCServer.java

@@ -0,0 +1,124 @@
+/*
+ *  Copyright (C) 2020 Temporal Technologies, Inc. All Rights Reserved.
+ *
+ *  Copyright 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ *  Modifications copyright (C) 2017 Uber Technologies, Inc.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License"). You may not
+ *  use this file except in compliance with the License. A copy of the License is
+ *  located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ *  or in the "license" file accompanying this file. This file is distributed on
+ *  an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ *  express or implied. See the License for the specific language governing
+ *  permissions and limitations under the License.
+ */
+
+package io.temporal.internal.testservice;
+
+import io.grpc.BindableService;
+import io.grpc.ManagedChannel;
+import io.grpc.Server;
+import io.grpc.inprocess.InProcessChannelBuilder;
+import io.grpc.inprocess.InProcessServerBuilder;
+import io.temporal.api.workflowservice.v1.WorkflowServiceGrpc;
+import java.io.IOException;
+import java.util.Collection;
+import java.util.concurrent.TimeUnit;
+import javax.annotation.Nullable;
+
+/**
+ * Generates an in-process service and the channel for it. Useful for testing, usually with mock and
+ * spy services.
+ */
+// TODO move to temporal-testing or temporal-test-server modules after WorkflowServiceStubs cleanup
+public class InProcessGRPCServer {
+  private final Server server;
+  @Nullable private final ManagedChannel channel;
+
+  /**
+   * Register the passed services in a new in-memory gRPC service with health service providing
+   * SERVING status for all of them. Also provides a channel to access the created server, see
+   * {@link #getChannel()}
+   *
+   * @param services implementations of the gRPC services. For example, one of them can be an
+   *     instance extending {@link WorkflowServiceGrpc.WorkflowServiceImplBase}
+   */
+  public InProcessGRPCServer(Collection<BindableService> services) {
+    this(services, true);
+  }
+
+  /**
+   * Register the passed services in a new in-memory gRPC service with health service providing
+   * SERVING status for all of them.
+   *
+   * @param services implementations of the gRPC services. For example, one of them can be an
+   *     instance extending {@link WorkflowServiceGrpc.WorkflowServiceImplBase}
+   * @param createChannel defines if a channel to access the created server should be created. If
+   *     true, the created channel will be accessible using {@link #getChannel()}
+   */
+  public InProcessGRPCServer(Collection<BindableService> services, boolean createChannel) {
+    String serverName = InProcessServerBuilder.generateName();
+    try {
+      InProcessServerBuilder inProcessServerBuilder =
+          InProcessServerBuilder.forName(serverName).directExecutor();
+      GRPCServerHelper.registerServicesAndHealthChecks(services, inProcessServerBuilder);
+      server = inProcessServerBuilder.build().start();
+    } catch (IOException unexpected) {
+      throw new RuntimeException(unexpected);
+    }
+    channel =
+        createChannel ? InProcessChannelBuilder.forName(serverName).directExecutor().build() : null;
+  }
+
+  public void shutdown() {
+    if (channel != null) {
+      channel.shutdown();
+    }
+    server.shutdown();
+  }
+
+  public void shutdownNow() {
+    if (channel != null) {
+      channel.shutdownNow();
+    }
+    server.shutdownNow();
+  }
+
+  public boolean isShutdown() {
+    return (channel == null || channel.isShutdown()) && server.isShutdown();
+  }
+
+  public boolean isTerminated() {
+    return (channel == null || channel.isTerminated()) && server.isTerminated();
+  }
+
+  public boolean awaitTermination(long timeout, TimeUnit unit) {
+    long start = System.currentTimeMillis();
+    long deadline = start + unit.toMillis(timeout);
+    long left = deadline - System.currentTimeMillis();
+    try {
+      if (channel != null && !channel.awaitTermination(left, TimeUnit.MILLISECONDS)) {
+        return false;
+      }
+
+      left = deadline - System.currentTimeMillis();
+      return server.awaitTermination(left, TimeUnit.MILLISECONDS);
+    } catch (InterruptedException e) {
+      Thread.currentThread().interrupt();
+      return false;
+    }
+  }
+
+  public Server getServer() {
+    return server;
+  }
+
+  @Nullable
+  public ManagedChannel getChannel() {
+    return channel;
+  }
+}