deno.land / x / jotai@v1.8.4 / src / utils / waitForAll.ts

نووسراو ببینە
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import { atom } from 'jotai'import type { Atom } from 'jotai'import { createMemoizeAtom } from './weakCache'
const memoizeAtom = createMemoizeAtom()const emptyArrayAtom = atom(() => [])
type ResolveAtom<T> = T extends Atom<infer V> ? V : Ttype AwaitedAtom<T> = Awaited<ResolveAtom<T>>
export function waitForAll<Atoms extends Atom<unknown>[]>( atoms: readonly [...Atoms]): Atom<{ [K in keyof Atoms]: AwaitedAtom<Atoms[K]>}>
export function waitForAll<Atoms extends Record<string, Atom<unknown>>>( atoms: Atoms): Atom<{ [K in keyof Atoms]: AwaitedAtom<Atoms[K]>}>
export function waitForAll< Atoms extends Record<string, Atom<unknown>> | readonly Atom<unknown>[]>(atoms: Atoms) { const createAtom = () => { const unwrappedAtoms = unwrapAtoms(atoms) const derivedAtom = atom((get) => { const promises: Promise<unknown>[] = [] const values = unwrappedAtoms.map((anAtom, index) => { try { return get(anAtom) } catch (e) { if (e instanceof Promise) { promises[index] = e } else { throw e } } }) if (promises.length) { throw Promise.all(promises) } return wrapResults(atoms, values) as { [K in keyof Atoms]: AwaitedAtom<Atoms[K]> } }) return derivedAtom }
if (Array.isArray(atoms)) { if (atoms.length) { return memoizeAtom(createAtom, atoms) } return emptyArrayAtom } return createAtom()}
const unwrapAtoms = < Atoms extends Record<string, Atom<unknown>> | readonly Atom<unknown>[]>( atoms: Atoms): Atom<unknown>[] => Array.isArray(atoms) ? atoms : Object.getOwnPropertyNames(atoms).map((key) => atoms[key as keyof Atoms])
const wrapResults = < Atoms extends Record<string, Atom<unknown>> | readonly Atom<unknown>[]>( atoms: Atoms, results: unknown[]): unknown[] | Record<string, unknown> => Array.isArray(atoms) ? results : Object.getOwnPropertyNames(atoms).reduce( (out, key, idx) => ({ ...out, [key]: results[idx] }), {} )
jotai

Version Info

Tagged at
2 years ago