batchPriority is union type

pmndrs · Jan 4, 2025 · d7ba0d5 · d7ba0d5
1 parent daaf056
commit d7ba0d5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/vanilla/store.ts b/src/vanilla/store.ts
@@ -167,14 +167,20 @@ const addDependency = <Value>(
 // Batch
 //
 
+type BatchPriority = 0 | 1 | 2
+
 type Batch = Set<() => void>[] & {
   /** Atom dependents map */
   D: Map<AnyAtom, Set<AnyAtom>>
 }
 
 const createBatch = (): Batch => Object.assign([], { D: new Map() })
 
-const addBatchFunc = (batch: Batch, priority: number, fn: () => void) => {
+const addBatchFunc = (
+  batch: Batch,
+  priority: BatchPriority,
+  fn: () => void,
+) => {
   ;(batch[priority] ||= new Set()).add(fn)
 }