bench.js

const path = require('node:path')
const fs = require('fs-extra')
const dayjs = require('dayjs')
const { set, get } = require('lodash')
const useBabel = process.env.BABEL
console.log('useBabel:', Boolean(useBabel))
class Bench {
  constructor(name) {
    this.name = name
  }

  start() {
    this.startTs = performance.now()
  }

  end() {
    this.endTs = performance.now()
  }

  timeSpan() {
    return this.endTs - this.startTs
  }

  get useBabel() {
    return Boolean(useBabel)
  }

  dump(key = this.useBabel ? 'babel' : 'ast-grep') {
    const ts = this.timeSpan()
    // if (ts < 20) {
    //   return
    // }
    const filename = dayjs().format('YYYY-MM-DD') + '.json'
    const targetDataFile = path.resolve(__dirname, '../benchmark/data', filename)
    if (fs.existsSync(targetDataFile)) {
      const json = fs.readJsonSync(targetDataFile)
      const arr = get(json, [this.name, key], [])
      arr.push(ts)
      set(json, [this.name, key], arr)
      fs.writeJSONSync(targetDataFile, json, { spaces: 2 })
    } else {
      fs.writeJSONSync(
        targetDataFile,
        {
          [this.name]: {
            [key]: [ts]
          }
        },
        { spaces: 2 }
      )
    }
  }
}

module.exports = function createBench(name) {
  return new Bench(name)
  // const bench = new Bench(name)
  // return {
  //   start() {
  //     bench.start()
  //   },
  //   end() {
  //     bench.end()
  //   },
  //   dump(key) {
  //     bench.dump(key)
  //   }
  // }
}