Monitor document store (#272)

Author: suresh-prakash

entity-service-factory/src/main/java/org/hypertrace/entity/service/EntityServiceFactory.java

@@ -17,6 +17,7 @@
 import org.hypertrace.entity.attribute.translator.EntityAttributeMapping;
 import org.hypertrace.entity.data.service.EntityDataServiceImpl;
 import org.hypertrace.entity.metric.EntityCounterMetricSender;
+import org.hypertrace.entity.metric.EntityMetricsReporter;
 import org.hypertrace.entity.query.service.EntityQueryServiceImpl;
 import org.hypertrace.entity.service.change.event.api.EntityChangeEventGenerator;
 import org.hypertrace.entity.service.change.event.impl.EntityChangeEventGeneratorFactory;
@@ -38,6 +39,8 @@ public List<GrpcPlatformService> buildServices(
     EntityServiceDataStoreConfig entityServiceDatastoreConfig =
         new EntityServiceDataStoreConfig(config);
     datastore = DatastoreProvider.getDatastore(entityServiceDatastoreConfig.getDataStoreConfig());
+
+    new EntityMetricsReporter(datastore, grpcServiceContainerEnvironment.getLifecycle()).monitor();
     grpcServiceContainerEnvironment.getLifecycle().shutdownComplete().thenRun(datastore::close);
 
     EntityAttributeMapping entityAttributeMapping =

entity-service-impl/build.gradle.kts

@@ -17,6 +17,7 @@ dependencies {
   implementation("org.hypertrace.core.grpcutils:grpc-client-utils:0.12.6")
   implementation("org.hypertrace.core.attribute.service:caching-attribute-service-client:0.14.15")
   implementation("org.hypertrace.core.serviceframework:platform-metrics:0.1.62")
+  implementation("org.hypertrace.core.serviceframework:docstore-metrics:0.1.62")
   implementation("com.google.inject:guice:5.0.1")
 
   implementation(project(":entity-type-service-rx-client"))

entity-service-impl/src/main/java/org/hypertrace/entity/metric/EntityMetricsReporter.java

@@ -0,0 +1,18 @@
+package org.hypertrace.entity.metric;
+
+import org.hypertrace.core.documentstore.Datastore;
+import org.hypertrace.core.serviceframework.docstore.metrics.DocStoreMetricsRegistry;
+import org.hypertrace.core.serviceframework.spi.PlatformServiceLifecycle;
+
+public class EntityMetricsReporter {
+  private final DocStoreMetricsRegistry metricsRegistry;
+
+  public EntityMetricsReporter(
+      final Datastore datastore, final PlatformServiceLifecycle lifecycle) {
+    metricsRegistry = new DocStoreMetricsRegistry(datastore).withPlatformLifecycle(lifecycle);
+  }
+
+  public void monitor() {
+    metricsRegistry.monitor();
+  }
+}