Merge pull request #97 from lorefnon/fix-subscription-types

Fix subscription type
stepci · Feb 25, 2024 · bc1d582 · bc1d582
2 parents 9de43dd + d08bf1f
commit bc1d582
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/examples/subscriptions.ts b/examples/subscriptions.ts
@@ -7,7 +7,8 @@ const queryType = g.type('Query', {
 })
 
 const subscriptionType = g.type('Subscription', {
-  counter: g.int()
+  counter: g.int(),
+  winner: g.string()
 })
 
 const resolvers: InferResolvers<{ Subscription: typeof subscriptionType }, {}> = {
@@ -19,6 +20,14 @@ const resolvers: InferResolvers<{ Subscription: typeof subscriptionType }, {}> =
           yield { counter: i }
         }
       }
+    },
+    winner: {
+      subscribe: async function* (parent, args, context, info) {
+        for (let i = 100; i >= 0; i--) {
+          await new Promise((resolve) => setTimeout(resolve, 1000))
+          yield { winner: `User ${i}` }
+        }
+      }
     }
   }
 }

diff --git a/src/index.ts b/src/index.ts
@@ -134,7 +134,9 @@ export type InferUnionNames<T> = T extends AnyUnion ? ObjectToUnion<T['_inner']>
 export type InferResolvers<T extends AnyTypes, X extends InferResolverConfig> = {
   [K in keyof T]: K extends 'Subscription' ? {
     [G in keyof T[K]['_shape']]?: {
-      subscribe: (parent: {}, args: InferArg<T[K]['_shape'][G]>, context: X['context'], info: GraphQLResolveInfo) => MaybePromise<AsyncIterator<{ [G in keyof T[K]['_shape']]: Infer<T[K]['_shape'][G], { omitResolver: AnyOmitResolver }> }>>
+      subscribe: (parent: {}, args: InferArg<T[K]['_shape'][G]>, context: X['context'], info: GraphQLResolveInfo) => MaybePromise<AsyncIterator<{
+        [key in G]: Infer<T[K]['_shape'][G], { omitResolver: AnyOmitResolver }> 
+      }>>
       resolve?: (value: Infer<T[K]['_shape'][G]>, args: InferArg<T[K]['_shape'][G]>, context: X['context'], info: GraphQLResolveInfo) => MaybePromise<Infer<T[K]['_shape'][G], { omitResolver: AnyOmitResolver }>>
     }
   } : {