context builder

hypertrace · Mar 27, 2024 · e191c2f · e191c2f
1 parent f0bbaa4
commit e191c2f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...ute-store/src/main/java/org/hypertrace/core/graphql/attributes/CachingAttributeStore.java b/...ute-store/src/main/java/org/hypertrace/core/graphql/attributes/CachingAttributeStore.java
@@ -89,9 +89,9 @@ public Single<AttributeModel> getForeignIdAttribute(
   @Override
   public Single<AttributeModel> getAttributeById(
       GraphQlRequestContext context, String attributeId) {
-    return GrpcRxExecutionContext.forContext(this.grpcContextBuilder.build(context))
-        .wrapSingle(() -> this.cachingAttributeClient.get(attributeId))
-        .toMaybe()
+    return grpcContextBuilder
+        .build(context)
+        .call(() -> cachingAttributeClient.get(attributeId))
         .mapOptional(this.translator::translate)
         .switchIfEmpty(Single.error(this.buildErrorForMissingAttributeId(attributeId)));
   }