refactor: update direct use of context to use new version

hypertrace-graphql-entity-schema/src/main/java/org/hypertrace/graphql/entity/fetcher/EntityFetcher.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.entity.fetcher;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -29,7 +31,9 @@ static final class EntityFetcherImpl implements DataFetcher<CompletableFuture<En
     public CompletableFuture<EntityResultSet> get(DataFetchingEnvironment environment) {
       return this.requestBuilder
           .build(
-              environment.getContext(), environment.getArguments(), environment.getSelectionSet())
+              contextFromEnvironment(environment),
+              environment.getArguments(),
+              environment.getSelectionSet())
           .flatMap(this.entityDao::getEntities)
           .toCompletionStage()
           .toCompletableFuture();

hypertrace-graphql-explorer-schema/src/main/java/org/hypertrace/graphql/explorer/fetcher/ExplorerFetcher.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.explorer.fetcher;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -31,7 +33,9 @@ static final class ExplorerFetcherImpl
     public CompletableFuture<ExploreResultSet> get(DataFetchingEnvironment environment) {
       return this.requestBuilder
           .build(
-              environment.getContext(), environment.getArguments(), environment.getSelectionSet())
+              contextFromEnvironment(environment),
+              environment.getArguments(),
+              environment.getSelectionSet())
           .flatMap(this.explorerDao::explore)
           .toCompletionStage()
           .toCompletableFuture();

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/fetcher/LabelApplicationRuleFetcher.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.label.fetcher;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -31,7 +33,7 @@ static final class LabelApplicationRuleFetcherImpl
     public CompletableFuture<LabelApplicationRuleResultSet> get(
         DataFetchingEnvironment environment) {
       return this.labelApplicationRuleDao
-          .getLabelApplicationRules(this.requestBuilder.build(environment.getContext()))
+          .getLabelApplicationRules(this.requestBuilder.build(contextFromEnvironment(environment)))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/fetcher/LabelFetcher.java

@@ -1,10 +1,13 @@
 package org.hypertrace.graphql.label.fetcher;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
 import javax.inject.Inject;
 import org.hypertrace.core.graphql.common.fetcher.InjectableDataFetcher;
+import org.hypertrace.graphql.label.joiner.EntityAndRuleJoiner;
 import org.hypertrace.graphql.label.joiner.EntityAndRuleJoinerBuilder;
 import org.hypertrace.graphql.label.schema.LabelResultSet;
 
@@ -25,8 +28,8 @@ static final class LabelFetcherImpl implements DataFetcher<CompletableFuture<Lab
     @Override
     public CompletableFuture<LabelResultSet> get(DataFetchingEnvironment environment) {
       return this.requestBuilder
-          .build(environment.getContext(), environment.getSelectionSet())
-          .flatMap(request -> request.joinLabelsWithEntitiesAndRules())
+          .build(contextFromEnvironment(environment), environment.getSelectionSet())
+          .flatMap(EntityAndRuleJoiner::joinLabelsWithEntitiesAndRules)
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/mutator/LabelApplicationRuleCreateMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.label.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -33,7 +35,7 @@ public CompletableFuture<LabelApplicationRule> get(DataFetchingEnvironment envir
       return this.labelApplicationRuleDao
           .createLabelApplicationRule(
               this.requestBuilder.buildCreateLabelApplicationRuleRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/mutator/LabelApplicationRuleDeleteMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.label.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -31,7 +33,7 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) {
       return this.labelApplicationRuleDao
           .deleteLabelApplicationRule(
               this.requestBuilder.buildDeleteLabelApplicationRuleRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/mutator/LabelApplicationRuleUpdateMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.label.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -33,7 +35,7 @@ public CompletableFuture<LabelApplicationRule> get(DataFetchingEnvironment envir
       return this.labelApplicationRuleDao
           .updateLabelApplicationRule(
               this.requestBuilder.buildUpdateLabelApplicationRuleRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/mutator/LabelCreateMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.label.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -30,7 +32,7 @@ public CompletableFuture<Label> get(DataFetchingEnvironment environment) {
       return this.configDao
           .createLabel(
               this.requestBuilder.buildCreateRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-labels-schema-api/src/main/java/org/hypertrace/graphql/label/mutator/LabelUpdateMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.label.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -30,7 +32,7 @@ public CompletableFuture<Label> get(DataFetchingEnvironment environment) {
       return this.configDao
           .updateLabel(
               this.requestBuilder.buildUpdateRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-spaces-schema/src/main/java/org/hypertrace/graphql/spaces/fetcher/SpaceConfigRuleFetcher.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spaces.fetcher;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -26,7 +28,7 @@ static final class SpaceConfigRuleFetcherImpl
     @Override
     public CompletableFuture<SpaceConfigRuleResultSet> get(DataFetchingEnvironment environment) {
       return this.configDao
-          .getAllRules(environment.getContext())
+          .getAllRules(contextFromEnvironment(environment))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-spaces-schema/src/main/java/org/hypertrace/graphql/spaces/fetcher/SpacesFetcher.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spaces.fetcher;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -25,7 +27,7 @@ static final class SpacesFetcherImpl implements DataFetcher<CompletableFuture<Sp
     @Override
     public CompletableFuture<SpaceResultSet> get(DataFetchingEnvironment environment) {
       return this.configDao
-          .getAllSpaces(environment.getContext())
+          .getAllSpaces(contextFromEnvironment(environment))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-spaces-schema/src/main/java/org/hypertrace/graphql/spaces/mutator/SpaceConfigRuleCreationMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spaces.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -32,7 +34,7 @@ public CompletableFuture<SpaceConfigRule> get(DataFetchingEnvironment environmen
       return this.configDao
           .createRule(
               this.requestBuilder.buildCreationRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-spaces-schema/src/main/java/org/hypertrace/graphql/spaces/mutator/SpaceConfigRuleDeleteMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spaces.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -31,7 +33,7 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) {
       return this.configDao
           .deleteRule(
               this.requestBuilder.buildDeleteRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-spaces-schema/src/main/java/org/hypertrace/graphql/spaces/mutator/SpaceConfigRuleUpdateMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spaces.mutator;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -32,7 +34,7 @@ public CompletableFuture<SpaceConfigRule> get(DataFetchingEnvironment environmen
       return this.configDao
           .updateRule(
               this.requestBuilder.buildUpdateRequest(
-                  environment.getContext(), environment.getArguments()))
+                  contextFromEnvironment(environment), environment.getArguments()))
           .toCompletionStage()
           .toCompletableFuture();
     }

hypertrace-graphql-span-processing-schema/src/main/java/org/hypertrace/graphql/spanprocessing/fetcher/mutation/ExcludeSpanCreateRuleMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spanprocessing.fetcher.mutation;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -31,7 +33,7 @@ static final class ExcludeSpanCreateRuleMutatorImpl
     @Override
     public CompletableFuture<ExcludeSpanRule> get(DataFetchingEnvironment environment) {
       return this.requestBuilder
-          .build(environment.getContext(), environment.getArguments())
+          .build(contextFromEnvironment(environment), environment.getArguments())
           .flatMap(this.spanProcessingRuleDao::createExcludeSpanRule)
           .toCompletionStage()
           .toCompletableFuture();

hypertrace-graphql-span-processing-schema/src/main/java/org/hypertrace/graphql/spanprocessing/fetcher/mutation/ExcludeSpanDeleteRuleMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spanprocessing.fetcher.mutation;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -33,7 +35,7 @@ static final class ExcludeSpanDeleteRuleMutatorImpl
     public CompletableFuture<DeleteSpanProcessingRuleResponse> get(
         DataFetchingEnvironment environment) {
       return this.requestBuilder
-          .build(environment.getContext(), environment.getArguments())
+          .build(contextFromEnvironment(environment), environment.getArguments())
           .flatMap(this.spanProcessingRuleDao::deleteExcludeSpanRule)
           .toCompletionStage()
           .toCompletableFuture();

hypertrace-graphql-span-processing-schema/src/main/java/org/hypertrace/graphql/spanprocessing/fetcher/mutation/ExcludeSpanUpdateRuleMutator.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spanprocessing.fetcher.mutation;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -31,7 +33,7 @@ static final class ExcludeSpanUpdateRuleMutatorImpl
     @Override
     public CompletableFuture<ExcludeSpanRule> get(DataFetchingEnvironment environment) {
       return this.requestBuilder
-          .build(environment.getContext(), environment.getArguments())
+          .build(contextFromEnvironment(environment), environment.getArguments())
           .flatMap(this.spanProcessingRuleDao::updateExcludeSpanRule)
           .toCompletionStage()
           .toCompletableFuture();

hypertrace-graphql-span-processing-schema/src/main/java/org/hypertrace/graphql/spanprocessing/fetcher/query/ExcludeSpanRulesFetcher.java

@@ -1,5 +1,7 @@
 package org.hypertrace.graphql.spanprocessing.fetcher.query;
 
+import static org.hypertrace.core.graphql.context.GraphQlRequestContext.contextFromEnvironment;
+
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.concurrent.CompletableFuture;
@@ -30,7 +32,7 @@ static final class ExcludeSpanRulesFetcherImpl
     @Override
     public CompletableFuture<ExcludeSpanRuleResultSet> get(DataFetchingEnvironment environment) {
       return this.spanProcessingRuleDao
-          .getExcludeSpanRules(this.requestBuilder.build(environment.getContext()))
+          .getExcludeSpanRules(this.requestBuilder.build(contextFromEnvironment(environment)))
           .toCompletionStage()
           .toCompletableFuture();
     }