chore(client-registry): entity data caching client (#259)

Author: skjindal93

hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/clients/ClientRegistry.java

@@ -5,6 +5,7 @@
 import org.hypertrace.core.datamodel.Event;
 import org.hypertrace.core.datamodel.StructuredTrace;
 import org.hypertrace.entity.data.service.client.EdsCacheClient;
+import org.hypertrace.entity.data.service.rxclient.EntityDataClient;
 import org.hypertrace.trace.accessor.entities.TraceEntityAccessor;
 import org.hypertrace.trace.reader.attributes.TraceAttributeReader;
 import org.hypertrace.traceenricher.enrichment.enrichers.cache.EntityCache;
@@ -23,6 +24,8 @@ public interface ClientRegistry {
 
   EdsCacheClient getEdsCacheClient();
 
+  EntityDataClient getEntityDataClient();
+
   EntityCache getEntityCache();
 
   CachingAttributeClient getCachingAttributeClient();

hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/clients/DefaultClientRegistry.java

@@ -34,6 +34,7 @@ public class DefaultClientRegistry implements ClientRegistry {
   private final ManagedChannel configServiceChannel;
   private final ManagedChannel entityServiceChannel;
   private final EdsCacheClient edsCacheClient;
+  private final EntityDataClient entityDataClient;
   private final CachingAttributeClient cachingAttributeClient;
   private final EntityCache entityCache;
   private final TraceEntityAccessor entityAccessor;
@@ -63,11 +64,12 @@ public DefaultClientRegistry(Config config) {
         new EdsCacheClient(
             new EntityDataServiceClient(this.entityServiceChannel),
             EntityServiceClientConfig.from(config).getCacheConfig());
+    this.entityDataClient = EntityDataClient.builder(this.entityServiceChannel).build();
     this.entityCache = new EntityCache(this.edsCacheClient);
     this.entityAccessor =
         new TraceEntityAccessorBuilder(
                 EntityTypeClient.builder(this.entityServiceChannel).build(),
-                EntityDataClient.builder(this.entityServiceChannel).build(),
+                this.entityDataClient,
                 this.cachingAttributeClient)
             .withEntityWriteThrottleDuration(
                 config.hasPath(TRACE_ENTITY_WRITE_THROTTLE_DURATION)
@@ -106,6 +108,11 @@ public EdsCacheClient getEdsCacheClient() {
     return this.edsCacheClient;
   }
 
+  @Override
+  public EntityDataClient getEntityDataClient() {
+    return this.entityDataClient;
+  }
+
   @Override
   public EntityCache getEntityCache() {
     return this.entityCache;