fix: should serialize subscription fn key

vercel · Jul 13, 2023 · 5f6a735 · 5f6a735
1 parent 2efed28
commit 5f6a735
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/_internal/src/utils/serialize.ts b/_internal/src/utils/serialize.ts
@@ -1,9 +1,9 @@
 import { stableHash } from './hash'
 import { isFunction } from './shared'
 
-import type { Key } from '../types'
+import type { Key, Arguments } from '../types'
 
-export const serialize = (key: Key): [string, Key] => {
+export const serialize = (key: Key): [string, Arguments] => {
   if (isFunction(key)) {
     try {
       key = key()

diff --git a/subscription/src/index.ts b/subscription/src/index.ts
@@ -25,8 +25,8 @@ export const subscription = (<Data = any, Error = any>(useSWRNext: SWRHook) =>
     subscribe: SWRSubscription<any, Data, Error>,
     config: SWRConfiguration & typeof SWRConfig.defaultValue
   ): SWRSubscriptionResponse<Data, Error> => {
-    const [key] = serialize(_key)
-    const originKey = _key
+    const [key, args] = serialize(_key)
+    const originKey = args
 
     // Prefix the key to avoid conflicts with other SWR resources.
     const subscriptionKey = key ? SUBSCRIPTION_PREFIX + key : undefined

diff --git a/test/use-swr-subscription.test.tsx b/test/use-swr-subscription.test.tsx
@@ -82,7 +82,7 @@ describe('useSWRSubscription', () => {
     }
 
     function Page() {
-      const { data, error } = useSWRSubscription([swrKey], subscribe, {
+      const { data, error } = useSWRSubscription(() => [swrKey], subscribe, {
         fallbackData: 'fallback'
       })
       return (