123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
#!/usr/bin/env npx tsx
import { add, complete, cycle, save, suite } from 'benny'import { atom } from '../src/vanilla/atom.ts'import type { PrimitiveAtom } from '../src/vanilla/atom.ts'import { createStore } from '../src/vanilla/store.ts'
const cleanupFns = new Set<() => void>()const cleanup = () => {  cleanupFns.forEach((fn) => fn())  cleanupFns.clear()}
const createStateWithAtoms = (n: number) => {  let targetAtom: PrimitiveAtom<number> | undefined  const store = createStore()  for (let i = 0; i < n; ++i) {    const a = atom(i)    if (!targetAtom) {      targetAtom = a    }    store.get(a)    const unsub = store.sub(a, () => {      store.get(a)    })    cleanupFns.add(unsub)  }  if (!targetAtom) {    throw new Error()  }  return [store, targetAtom] as const}
const main = async () => {  await suite(    'subscribe-write',    ...[2, 3, 4, 5, 6].map((n) =>      add(`atoms=${10 ** n}`, () => {        cleanup()        const [store, targetAtom] = createStateWithAtoms(10 ** n)        return () => store.set(targetAtom, (c) => c + 1)      }),    ),    cycle(),    complete(),    save({      folder: __dirname,      file: 'subscribe-write',      format: 'json',    }),    save({      folder: __dirname,      file: 'subscribe-write',      format: 'chart.html',    }),  )}
main()