fix(lens): match ctx instead of cause in select

artalar · Jun 22, 2024 · 3f1fd8e · 3f1fd8e
1 parent e0b9cf8
commit 3f1fd8e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/lens/src/select.ts b/packages/lens/src/select.ts
@@ -1,7 +1,7 @@
 import {
   Atom,
-  AtomCache,
   AtomProto,
+  Ctx,
   CtxSpy,
   __count,
   atom,
@@ -16,16 +16,16 @@ const mapAtom = atom(
 )
 mapAtom.__reatom.initState = () => new WeakMap()
 
-const touchedMap = new WeakMap<AtomCache, Set<FunctionSource>>()
+const touchedMap = new WeakMap<Ctx, Set<FunctionSource>>()
 
 export const select = <T>(
   ctx: CtxSpy,
   cb: (ctx: CtxSpy) => T,
   equal: (oldState: T, newState: T) => boolean = () => false,
 ): T => {
-  let touched = touchedMap.get(ctx.cause)
+  let touched = touchedMap.get(ctx)
   if (!touched) {
-    touchedMap.set(ctx.cause, (touched = new Set()))
+    touchedMap.set(ctx, (touched = new Set()))
   }
 
   const map = ctx.get(mapAtom)