fix(entities): support for API filters for filtered data specs (#232)

* fix(entities): support for API filters for filtered data specs

* nit: remove extra line

Author: skjindal93

hypertrace-graphql-entity-schema/build.gradle.kts

@@ -24,6 +24,7 @@ dependencies {
   implementation(commonLibs.hypertrace.grpcutils.client)
   implementation(projects.hypertraceGraphqlLabelsSchemaApi)
 
+  implementation(localLibs.core.request.transformation)
   implementation(localLibs.core.context)
   implementation(localLibs.core.grpc)
   implementation(localLibs.core.schema.utils)

hypertrace-graphql-entity-schema/src/main/java/org/hypertrace/graphql/entity/dao/GatewayServiceEntityDao.java

@@ -13,6 +13,7 @@
 import lombok.extern.slf4j.Slf4j;
 import org.hypertrace.core.graphql.common.request.AttributeRequest;
 import org.hypertrace.core.graphql.context.GraphQlRequestContext;
+import org.hypertrace.core.graphql.request.transformation.RequestTransformer;
 import org.hypertrace.core.graphql.rx.BoundedIoScheduler;
 import org.hypertrace.core.graphql.spi.config.GraphQlServiceConfig;
 import org.hypertrace.core.graphql.utils.grpc.GrpcChannelRegistry;
@@ -44,6 +45,7 @@ class GatewayServiceEntityDao implements EntityDao {
   private final GraphQlServiceConfig serviceConfig;
   private final Scheduler boundedIoScheduler;
   private final LabelJoinerBuilder labelJoinerBuilder;
+  private final RequestTransformer requestTransformer;
 
   @Inject
   GatewayServiceEntityDao(
@@ -55,13 +57,15 @@ class GatewayServiceEntityDao implements EntityDao {
       GatewayServiceEntityConverter entityConverter,
       BaselineDao baselineDao,
       LabelJoinerBuilder labelJoinerBuilder,
+      RequestTransformer requestTransformer,
       @BoundedIoScheduler Scheduler boundedIoScheduler) {
     this.grpcContextBuilder = grpcContextBuilder;
     this.requestBuilder = requestBuilder;
     this.entityConverter = entityConverter;
     this.baselineDao = baselineDao;
     this.labelJoinerBuilder = labelJoinerBuilder;
     this.serviceConfig = serviceConfig;
+    this.requestTransformer = requestTransformer;
     this.boundedIoScheduler = boundedIoScheduler;
 
     final GrpcChannelConfig grpcChannelConfig =
@@ -81,8 +85,9 @@ class GatewayServiceEntityDao implements EntityDao {
   @Override
   public Single<EntityResultSet> getEntities(EntityRequest request) {
     GraphQlRequestContext context = request.resultSetRequest().context();
-    return this.requestBuilder
-        .buildRequest(request)
+    return this.requestTransformer
+        .transform(request)
+        .flatMap(this.requestBuilder::buildRequest)
         .subscribeOn(this.boundedIoScheduler)
         .flatMap(serverRequest -> this.fetchAndMapEntities(context, request, serverRequest));
   }