1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
#!/usr/bin/env npx ts-node
import { add, complete, cycle, save, suite } from 'benny'import { atom } from '../src/core/atom'import type { PrimitiveAtom } from '../src/core/atom'import {  READ_ATOM,  SUBSCRIBE_ATOM,  WRITE_ATOM,  createStore,} from '../src/core/store'
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[READ_ATOM](a)    const unsub = store[SUBSCRIBE_ATOM](a, () => {      store[READ_ATOM](a)    })    cleanupFns.add(unsub)  }  if (!targetAtom) {    throw new Error()  }  return [store, targetAtom] as const}
const main = async () => {  for (const n of [2, 3, 4, 5, 6]) {    await suite(      `subscribe-write-${n}`,      add(`atoms=${10 ** n}`, () => {        cleanup()        const [store, targetAtom] = createStateWithAtoms(10 ** n)        return () => store[WRITE_ATOM](targetAtom, (c) => c + 1)      }),      cycle(),      complete(),      save({        folder: __dirname,        file: `subscribe-write-${n}`,        format: 'json',      }),      save({        folder: __dirname,        file: `subscribe-write-${n}`,        format: 'chart.html',      })    )  }}
main()