squiggle/packages/squiggle-lang/benchmark/Benchmark_Map.res

module StringMap: Benchmark_Helpers.BenchmarkTopic = {
  let size = 1000
  let iterations = 10_000

  let kv = Belt.Array.range(1, size)->Belt.Array.map(v => ("key" ++ v->Belt.Int.toString, v))

  let beltMap = () => {
    Belt.Range.forEach(1, iterations, _ => {
      let m = Belt.Map.String.empty
      let _ = Belt.Array.reduce(kv, m, (acc, (k, v)) => acc->Belt.Map.String.set(k, v))
    })
  }

  let beltMutableMap = () => {
    Belt.Range.forEach(1, iterations, _ => {
      let m = Belt.MutableMap.String.make()
      let _ = Belt.Array.reduce(kv, m, (acc, (k, v)) => {
        acc->Belt.MutableMap.String.set(k, v)
        acc
      })
    })
  }

  let beltHashMap = () => {
    Belt.Range.forEach(1, iterations, _ => {
      let m = Belt.HashMap.String.make(~hintSize=100)
      let _ = Belt.Array.reduce(kv, m, (acc, (k, v)) => {
        acc->Belt.HashMap.String.set(k, v)
        acc
      })
    })
  }

  let jsDict = () => {
    Belt.Range.forEach(1, iterations, _ => {
      let m = Js.Dict.empty()
      let _ = Belt.Array.reduce(kv, m, (acc, (k, v)) => {
        acc->Js.Dict.set(k, v)
        acc
      })
    })
  }

  let jsMap = () => {
    Belt.Range.forEach(1, iterations, _ => {
      let m = Js_map.make()
      let _ = Belt.Array.reduce(kv, m, (acc, (k, v)) => acc->Js_map.set(k, v))
    })
  }

  let runAll = () => {
    Js.log(
      `Filling a map with ("key{i}" => "i") key-value pairs, size ${size->Js.Int.toString} (${iterations->Js.Int.toString} iterations)`,
    )
    Benchmark_Helpers.measure("Belt.Map.String", beltMap)
    Benchmark_Helpers.measure("Belt.MutableMap.String", beltMutableMap)
    Benchmark_Helpers.measure("Belt.HashMap.String", beltHashMap)
    Benchmark_Helpers.measure("Js.Dict", jsDict)
    Benchmark_Helpers.measure("Js.Map", jsMap)
  }
}

let runAll = StringMap.runAll()