chore: expose server lifecycle into container environment

Author: aaron-steinfeld

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

@@ -12,6 +12,7 @@ dependencies {
   api("io.grpc:grpc-services")
   api("org.hypertrace.core.grpcutils:grpc-client-utils:0.7.5")
   api("com.typesafe:config:1.4.2")
+  api(project(":service-framework-spi"))
 
   annotationProcessor("org.projectlombok:lombok:1.18.24")
   compileOnly("org.projectlombok:lombok:1.18.24")

platform-grpc-service-framework/src/main/java/org/hypertrace/core/serviceframework/grpc/ConsolidatedGrpcPlatformServiceContainer.java

@@ -31,7 +31,7 @@ protected InProcessGrpcChannelRegistry buildChannelRegistry() {
   protected GrpcServiceContainerEnvironment buildContainerEnvironment(
       InProcessGrpcChannelRegistry channelRegistry, HealthStatusManager healthStatusManager) {
     return new ConsolidatedGrpcServiceContainerEnvironment(
-        channelRegistry, healthStatusManager, this.getInProcessServerName());
+        channelRegistry, healthStatusManager, this.getInProcessServerName(), this.getLifecycle());
   }
 
   @Override
@@ -51,7 +51,6 @@ protected Collection<GrpcPlatformServiceFactory> getServiceFactories() {
     return Collections.emptySet();
   }
 
-  @Deprecated
   protected int getServicePort() {
     return this.getAppConfig().getInt(DEFAULT_PORT_PATH);
   }

platform-grpc-service-framework/src/main/java/org/hypertrace/core/serviceframework/grpc/ConsolidatedGrpcServiceContainerEnvironment.java

@@ -7,6 +7,7 @@
 import lombok.Getter;
 import org.hypertrace.core.grpcutils.client.InProcessGrpcChannelRegistry;
 import org.hypertrace.core.serviceframework.config.ConfigClientFactory;
+import org.hypertrace.core.serviceframework.spi.PlatformServiceLifecycle;
 
 @AllArgsConstructor
 class ConsolidatedGrpcServiceContainerEnvironment implements GrpcServiceContainerEnvironment {
@@ -16,6 +17,8 @@ class ConsolidatedGrpcServiceContainerEnvironment implements GrpcServiceContaine
 
   @Getter private final String inProcessChannelName;
 
+  @Getter private final PlatformServiceLifecycle lifecycle;
+
   @Override
   public void reportServiceStatus(String serviceName, ServingStatus status) {
     this.healthStatusManager.setStatus(serviceName, status);

platform-grpc-service-framework/src/main/java/org/hypertrace/core/serviceframework/grpc/GrpcServiceContainerEnvironment.java

@@ -3,6 +3,7 @@
 import com.typesafe.config.Config;
 import io.grpc.health.v1.HealthCheckResponse.ServingStatus;
 import org.hypertrace.core.grpcutils.client.InProcessGrpcChannelRegistry;
+import org.hypertrace.core.serviceframework.spi.PlatformServiceLifecycle;
 
 public interface GrpcServiceContainerEnvironment {
   InProcessGrpcChannelRegistry getChannelRegistry();
@@ -12,4 +13,6 @@ public interface GrpcServiceContainerEnvironment {
   Config getConfig(String serviceName);
 
   String getInProcessChannelName();
+
+  PlatformServiceLifecycle getLifecycle();
 }

platform-grpc-service-framework/src/main/java/org/hypertrace/core/serviceframework/grpc/StandAloneGrpcPlatformServiceContainer.java

@@ -40,6 +40,10 @@ protected List<GrpcPlatformServerDefinition> getServerDefinitions() {
   protected GrpcServiceContainerEnvironment buildContainerEnvironment(
       InProcessGrpcChannelRegistry channelRegistry, HealthStatusManager healthStatusManager) {
     return new StandAloneGrpcServiceContainerEnvironment(
-        channelRegistry, healthStatusManager, this.configClient, this.getInProcessServerName());
+        channelRegistry,
+        healthStatusManager,
+        this.configClient,
+        this.getInProcessServerName(),
+        this.getLifecycle());
   }
 }

platform-grpc-service-framework/src/main/java/org/hypertrace/core/serviceframework/grpc/StandAloneGrpcServiceContainerEnvironment.java

@@ -7,6 +7,7 @@
 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 StandAloneGrpcServiceContainerEnvironment implements GrpcServiceContainerEnvironment {
@@ -18,6 +19,8 @@ class StandAloneGrpcServiceContainerEnvironment implements GrpcServiceContainerE
 
   @Getter private final String inProcessChannelName;
 
+  @Getter private final PlatformServiceLifecycle lifecycle;
+
   @Override
   public void reportServiceStatus(String serviceName, ServingStatus status) {
     this.healthStatusManager.setStatus(serviceName, status);