[smart-subscriptions] handle subscribe method being async

hayes · Oct 23, 2024 · 71a722a · 71a722a
1 parent 69024a6
commit 71a722a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.changeset/slow-kings-stare.md b/.changeset/slow-kings-stare.md
@@ -0,0 +1,5 @@
+---
+"@pothos/plugin-smart-subscriptions": patch
+---
+
+Handle async subsdcribe methods
diff --git a/packages/plugin-smart-subscriptions/src/resolve-with-cache.ts b/packages/plugin-smart-subscriptions/src/resolve-with-cache.ts
@@ -36,7 +36,11 @@ export default function resolveWithCache<Types extends SchemaTypes>(
   function cacheResult(result: unknown) {
     const cacheNode = cache.add(info, key, canRefetch, result);
 
-    subscribe?.(cacheNode.managerForField(), parent, args, context, info);
+    const sub = subscribe?.(cacheNode.managerForField(), parent, args, context, info);
+
+    if (isThenable(sub)) {
+      return sub.then(() => result);
+    }
 
     return result;
   }