types: allow passing function as Data for `useSWRSubscriptionOption…

…s` (#2551) feat: allow passing function as `Data` for `useSWRSubscriptionOptions`
vercel · Apr 10, 2023 · 1f71783 · 1f71783
1 parent 540a449
commit 1f71783
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/subscription/index.ts b/subscription/index.ts
@@ -52,7 +52,10 @@ export const subscription = (<Data = any, Error = any>(useSWRNext: SWRHook) =>
       const [, set] = createCacheHelper<Data>(cache, subscriptionKey)
       const refCount = subscriptions.get(subscriptionKey) || 0
 
-      const next = (error?: Error | null, data?: Data) => {
+      const next: SWRSubscriptionOptions<Data, Error>['next'] = (
+        error,
+        data
+      ) => {
         if (error !== null && typeof error !== 'undefined') {
           set({ error })
         } else {

diff --git a/subscription/types.ts b/subscription/types.ts
@@ -1,7 +1,7 @@
-import type { Key, SWRConfiguration } from 'swr'
+import type { Key, SWRConfiguration, MutatorCallback } from 'swr'
 
 export type SWRSubscriptionOptions<Data = any, Error = any> = {
-  next: (err?: Error | null, data?: Data) => void
+  next: <T = Data>(err?: Error | null, data?: Data | MutatorCallback<T>) => void
 }
 
 export type SWRSubscription<