refactor(store/enhancers): use weakMemo

src/app/store/observers/actionObserver.ts

@@ -9,7 +9,7 @@ import {
 import { filter, map, type Observable, Subject } from 'rxjs'
 
 import { injectStoreExtension } from '../enhancers/injectStoreExtension'
-import { createWeakCache } from './weakCache'
+import { weakMemo } from './weakMemo'
 
 type ObserveAction = <Payload>(actionCreator: PayloadActionCreator<Payload>) => Observable<Payload>
 
@@ -36,11 +36,9 @@ export const createActionObserver = (): ActionObserver => {
     return result
   }
 
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  const cache = createWeakCache<PayloadActionCreator<any>>()
-
-  const onAction: ObserveAction = (actionCreator) =>
-    cache(actionCreator, () => action$.pipe(filter(matchType(actionCreator)), map(getPayload)))
+  const onAction: ObserveAction = weakMemo((actionCreator) =>
+    action$.pipe(filter(matchType(actionCreator)), map(getPayload)),
+  )
 
   const enhancer = injectStoreExtension(() => ({ onAction }))
 

src/app/store/observers/stateObserver.ts

@@ -4,7 +4,7 @@ import { BehaviorSubject, distinctUntilChanged, map, type Observable } from 'rxj
 import { invariant } from '@/common/utils'
 
 import { injectStoreExtension } from '../enhancers/injectStoreExtension'
-import { createWeakCache } from './weakCache'
+import { weakMemo } from './weakMemo'
 
 type ObserveState<State> = <Selected>(selector: Selector<State, Selected>) => Observable<Selected>
 
@@ -32,10 +32,9 @@ export const createStateObserver = <State>(): StateObserver<State> => {
     }
   }
 
-  const cache = createWeakCache<Selector<State, unknown>>()
-
-  const onState: ObserveState<State> = (selector) =>
-    cache(selector, () => distinctState$.pipe(map(selector), distinctUntilChanged()))
+  const onState: ObserveState<State> = weakMemo((selector) =>
+    distinctState$.pipe(map(selector), distinctUntilChanged()),
+  )
 
   const enhancer = injectStoreExtension(() => ({ onState }))
 

src/app/store/observers/weakMemo.ts

@@ -0,0 +1,9 @@
+export const weakMemo = <Arg extends WeakKey, Result>(fn: (arg: Arg) => Result): typeof fn => {
+  const cache = new WeakMap<Arg, Result>()
+  return (arg) => {
+    if (!cache.has(arg)) {
+      cache.set(arg, fn(arg))
+    }
+    return cache.get(arg)!
+  }
+}