feat(entity-joiner): support for fixed time range argument

Author: skjindal93

hypertrace-graphql-entity-schema/src/main/java/org/hypertrace/graphql/entity/joiner/DefaultEntityJoinerBuilder.java

@@ -72,7 +72,6 @@ class DefaultEntityJoinerBuilder implements EntityJoinerBuilder {
   private final ArgumentDeserializer argumentDeserializer;
   private final ResultSetRequestBuilder resultSetRequestBuilder;
   private final FilterRequestBuilder filterRequestBuilder;
-  private final AttributeRequestBuilder attributeRequestBuilder;
   private final Scheduler boundedIoScheduler;
   private final EntityLabelRequestBuilder entityLabelRequestBuilder;
 
@@ -83,7 +82,6 @@ class DefaultEntityJoinerBuilder implements EntityJoinerBuilder {
       ArgumentDeserializer argumentDeserializer,
       ResultSetRequestBuilder resultSetRequestBuilder,
       FilterRequestBuilder filterRequestBuilder,
-      AttributeRequestBuilder attributeRequestBuilder,
       @BoundedIoScheduler Scheduler boundedIoScheduler,
       EntityLabelRequestBuilder entityLabelRequestBuilder) {
 
@@ -92,7 +90,6 @@ class DefaultEntityJoinerBuilder implements EntityJoinerBuilder {
     this.argumentDeserializer = argumentDeserializer;
     this.resultSetRequestBuilder = resultSetRequestBuilder;
     this.filterRequestBuilder = filterRequestBuilder;
-    this.attributeRequestBuilder = attributeRequestBuilder;
     this.boundedIoScheduler = boundedIoScheduler;
     this.entityLabelRequestBuilder = entityLabelRequestBuilder;
   }
@@ -104,7 +101,20 @@ public Single<EntityJoiner> build(
       List<String> pathToEntityJoinable) {
     return Single.just(
         new DefaultEntityJoiner(
-            context, this.groupEntityFieldsByType(selectionSet, pathToEntityJoinable)));
+            context,
+            new InstantTimeRange(),
+            this.groupEntityFieldsByType(selectionSet, pathToEntityJoinable)));
+  }
+
+  @Override
+  public Single<EntityJoiner> build(
+      GraphQlRequestContext context,
+      TimeRangeArgument timeRange,
+      DataFetchingFieldSelectionSet selectionSet,
+      List<String> pathToEntityJoinable) {
+    return Single.just(
+        new DefaultEntityJoiner(
+            context, timeRange, this.groupEntityFieldsByType(selectionSet, pathToEntityJoinable)));
   }
 
   private String getEntityType(SelectedField entityField) {
@@ -135,6 +145,7 @@ private Multimap<String, SelectedField> groupEntityFieldsByType(
   @AllArgsConstructor
   private class DefaultEntityJoiner implements EntityJoiner {
     private final GraphQlRequestContext context;
+    private final TimeRangeArgument timeRange;
     private final Multimap<String, SelectedField> entityFieldsByType;
 
     @Override
@@ -264,7 +275,7 @@ private Single<EntityRequest> buildEntityRequest(
               entityType,
               entityIdsToFilterSize,
               ZERO_OFFSET,
-              new InstantTimeRange(),
+              timeRange,
               List.<AttributeAssociation<AggregatableOrderArgument>>of(), // Order does not matter
               filterArguments,
               this.entityFieldsByType.get(entityType).stream(),

hypertrace-graphql-entity-schema/src/main/java/org/hypertrace/graphql/entity/joiner/EntityJoinerBuilder.java

@@ -3,6 +3,7 @@
 import graphql.schema.DataFetchingFieldSelectionSet;
 import io.reactivex.rxjava3.core.Single;
 import java.util.List;
+import org.hypertrace.core.graphql.common.schema.arguments.TimeRangeArgument;
 import org.hypertrace.core.graphql.context.GraphQlRequestContext;
 
 public interface EntityJoinerBuilder {
@@ -11,4 +12,10 @@ Single<EntityJoiner> build(
       GraphQlRequestContext context,
       DataFetchingFieldSelectionSet selectionSet,
       List<String> pathToEntityJoinable);
+
+  Single<EntityJoiner> build(
+      GraphQlRequestContext context,
+      TimeRangeArgument timeRange,
+      DataFetchingFieldSelectionSet selectionSet,
+      List<String> pathToEntityJoinable);
 }

hypertrace-graphql-entity-schema/src/test/java/org/hypertrace/graphql/entity/joiner/DefaultEntityJoinerBuilderTest.java

@@ -66,7 +66,6 @@ class DefaultEntityJoinerBuilderTest {
   @Mock ArgumentDeserializer mockDeserializer;
   @Mock ResultSetRequestBuilder mockResultSetRequestBuilder;
   @Mock FilterRequestBuilder mockFilterRequestBuilder;
-  @Mock AttributeRequestBuilder attributeRequestBuilder;
   @Mock GraphQlRequestContext mockRequestContext;
   @Mock DataFetchingFieldSelectionSet mockSelectionSet;
   @Mock AttributeAssociation<FilterArgument> mockFilter;
@@ -86,7 +85,6 @@ void beforeEach() {
             mockDeserializer,
             mockResultSetRequestBuilder,
             mockFilterRequestBuilder,
-            attributeRequestBuilder,
             testScheduler,
             mockEntityLabelRequestBuilder);
   }