feat(effects): add isInit method

artalar · Oct 20, 2023 · 4a8aeb1 · 4a8aeb1 · vercel · Oct 20, 2023
1 parent 8ac256c
commit 4a8aeb1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/packages/effects/src/index.ts b/packages/effects/src/index.ts
@@ -1,4 +1,5 @@
 import {
+  atom,
   Atom,
   AtomCache,
   AtomProto,
@@ -272,3 +273,12 @@ export const withAbortableSchedule = <T extends Ctx>(ctx: T): T => {
     },
   })
 }
+
+const initSetAtom = atom(
+  (ctx, state = new WeakSet<AtomProto>()) => state,
+  'initSetAtom',
+)
+export const isInit = (ctx: Ctx) => {
+  const set = ctx.get(initSetAtom)
+  return set.has(ctx.cause.proto) ? false : !!set.add(ctx.cause.proto)
+}