chore: expose hybrid service framework (#110)

* chore: expose hybrid service framework

* style: format

* chore: add hybrid http factory

Author: aaron-steinfeld

platform-hybrid-service-framework/build.gradle.kts

@@ -0,0 +1,15 @@
+plugins {
+  `java-library`
+  jacoco
+  id("org.hypertrace.publish-plugin")
+  id("org.hypertrace.jacoco-report-plugin")
+}
+
+dependencies {
+  api(project(":platform-grpc-service-framework"))
+  api(project(":platform-http-service-framework"))
+  api(project(":platform-service-framework"))
+
+  annotationProcessor("org.projectlombok:lombok:1.18.24")
+  compileOnly("org.projectlombok:lombok:1.18.24")
+}

platform-hybrid-service-framework/src/main/java/org/hypertrace/core/serviceframework/hybrid/HybridHttpHandlerFactory.java

@@ -0,0 +1,10 @@
+package org.hypertrace.core.serviceframework.hybrid;
+
+import java.util.List;
+import org.hypertrace.core.serviceframework.http.HttpHandlerDefinition;
+
+@FunctionalInterface
+public interface HybridHttpHandlerFactory {
+
+  List<HttpHandlerDefinition> buildHandlers(HybridServiceContainerEnvironment containerEnvironment);
+}

platform-hybrid-service-framework/src/main/java/org/hypertrace/core/serviceframework/hybrid/HybridPlatformService.java

@@ -0,0 +1,79 @@
+package org.hypertrace.core.serviceframework.hybrid;
+
+import com.google.common.collect.Streams;
+import io.grpc.protobuf.services.HealthStatusManager;
+import java.util.Collection;
+import java.util.List;
+import java.util.stream.Collectors;
+import lombok.extern.slf4j.Slf4j;
+import org.hypertrace.core.grpcutils.client.InProcessGrpcChannelRegistry;
+import org.hypertrace.core.serviceframework.config.ConfigClient;
+import org.hypertrace.core.serviceframework.grpc.GrpcPlatformServerDefinition;
+import org.hypertrace.core.serviceframework.grpc.StandAloneGrpcPlatformServiceContainer;
+import org.hypertrace.core.serviceframework.http.HttpContainer;
+import org.hypertrace.core.serviceframework.http.HttpHandlerDefinition;
+import org.hypertrace.core.serviceframework.http.HttpHandlerFactory;
+import org.hypertrace.core.serviceframework.http.jetty.JettyHttpServerBuilder;
+
+@Slf4j
+public abstract class HybridPlatformService extends StandAloneGrpcPlatformServiceContainer {
+
+  private HttpContainer httpContainer;
+
+  public HybridPlatformService(ConfigClient configClient) {
+    super(configClient);
+  }
+
+  @Override
+  protected void doStart() {
+    this.httpContainer.start();
+    super.doStart();
+  }
+
+  @Override
+  protected void doStop() {
+    this.httpContainer.stop();
+    super.doStop();
+  }
+
+  protected abstract List<GrpcPlatformServerDefinition> getServerDefinitions();
+
+  protected List<HttpHandlerFactory> getHttpHandlerFactories() {
+    return List.of();
+  }
+
+  protected List<HybridHttpHandlerFactory> getHybridHttpHandlerFactories() {
+    return List.of();
+  }
+
+  @Override
+  protected HybridServiceContainerEnvironment buildContainerEnvironment(
+      InProcessGrpcChannelRegistry channelRegistry, HealthStatusManager healthStatusManager) {
+    HybridServiceContainerEnvironment containerEnvironment =
+        new StandAloneHybridServiceContainerEnvironment(
+            channelRegistry,
+            healthStatusManager,
+            this.configClient,
+            this.getInProcessServerName(),
+            this.getLifecycle());
+    this.httpContainer = this.buildHttpContainer(containerEnvironment);
+    return containerEnvironment;
+  }
+
+  private HttpContainer buildHttpContainer(HybridServiceContainerEnvironment environment) {
+    return new JettyHttpServerBuilder()
+        .addHandlers(this.buildHandlerDefinitions(environment))
+        .build();
+  }
+
+  private List<HttpHandlerDefinition> buildHandlerDefinitions(
+      HybridServiceContainerEnvironment environment) {
+    return Streams.concat(
+            this.getHttpHandlerFactories().stream()
+                .map(handlerFactory -> handlerFactory.buildHandlers(environment)),
+            this.getHybridHttpHandlerFactories().stream()
+                .map(handlerFactory -> handlerFactory.buildHandlers(environment)))
+        .flatMap(Collection::stream)
+        .collect(Collectors.toUnmodifiableList());
+  }
+}

platform-hybrid-service-framework/src/main/java/org/hypertrace/core/serviceframework/hybrid/HybridServiceContainerEnvironment.java

@@ -0,0 +1,7 @@
+package org.hypertrace.core.serviceframework.hybrid;
+
+import org.hypertrace.core.serviceframework.grpc.GrpcServiceContainerEnvironment;
+import org.hypertrace.core.serviceframework.http.HttpContainerEnvironment;
+
+public interface HybridServiceContainerEnvironment
+    extends GrpcServiceContainerEnvironment, HttpContainerEnvironment {}

platform-hybrid-service-framework/src/main/java/org/hypertrace/core/serviceframework/hybrid/StandAloneHybridServiceContainerEnvironment.java

@@ -0,0 +1,33 @@
+package org.hypertrace.core.serviceframework.hybrid;
+
+import com.typesafe.config.Config;
+import io.grpc.health.v1.HealthCheckResponse.ServingStatus;
+import io.grpc.protobuf.services.HealthStatusManager;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import org.hypertrace.core.grpcutils.client.InProcessGrpcChannelRegistry;
+import org.hypertrace.core.serviceframework.config.ConfigClient;
+import org.hypertrace.core.serviceframework.spi.PlatformServiceLifecycle;
+
+@AllArgsConstructor
+class StandAloneHybridServiceContainerEnvironment implements HybridServiceContainerEnvironment {
+
+  @Getter private final InProcessGrpcChannelRegistry channelRegistry;
+  private final HealthStatusManager healthStatusManager;
+
+  private final ConfigClient configClient;
+
+  @Getter private final String inProcessChannelName;
+
+  @Getter private final PlatformServiceLifecycle lifecycle;
+
+  @Override
+  public void reportServiceStatus(String serviceName, ServingStatus status) {
+    this.healthStatusManager.setStatus(serviceName, status);
+  }
+
+  @Override
+  public Config getConfig(String serviceName) {
+    return this.configClient.getConfig(serviceName, null, null, null);
+  }
+}

settings.gradle.kts

@@ -14,6 +14,7 @@ plugins {
 
 include(":platform-grpc-service-framework")
 include(":platform-http-service-framework")
+include(":platform-hybrid-service-framework")
 include(":platform-service-framework")
 include(":platform-metrics")
 include(":docstore-metrics")