refactor benchmarks a bit

plausible · Jan 29, 2024 · 7572662 · 7572662
1 parent 9d8e88a
commit 7572662
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/bench/insert.exs b/bench/insert.exs
@@ -37,18 +37,22 @@ Benchee.run(
     end,
     # "control stream" => fn rows -> rows |> Stream.chunk_every(60_000) |> Stream.run() end,
     "encode stream" => fn rows ->
+      encoding_types = Ch.RowBinary.encoding_types(types)
+
       rows
       |> Stream.chunk_every(60_000)
-      |> Stream.map(fn chunk -> RowBinary.encode_rows(chunk, types) end)
+      |> Stream.map(fn chunk -> RowBinary._encode_rows(chunk, encoding_types) end)
       |> Stream.run()
     end,
     "insert stream" => fn rows ->
       DBConnection.run(
         conn,
         fn conn ->
+          encoding_types = Ch.RowBinary.encoding_types(types)
+
           rows
           |> Stream.chunk_every(60_000)
-          |> Stream.map(fn chunk -> RowBinary.encode_rows(chunk, types) end)
+          |> Stream.map(fn chunk -> RowBinary._encode_rows(chunk, encoding_types) end)
           |> Enum.into(Ch.stream(conn, statement))
         end,
         timeout: :infinity

diff --git a/bench/stream.exs b/bench/stream.exs
@@ -17,7 +17,7 @@ Benchee.run(
         conn,
         fn conn ->
           conn
-          |> Ch.stream(statement, _params = [], format: "RowBinary")
+          |> Ch.stream(statement, _params = [], format: "RowBinary", timeout: :infinity)
           |> Stream.run()
         end,
         timeout: :infinity
@@ -28,7 +28,7 @@ Benchee.run(
         conn,
         fn conn ->
           conn
-          |> Ch.stream(statement, _params = [], format: "RowBinary")
+          |> Ch.stream(statement, _params = [], format: "RowBinary", timeout: :infinity)
           |> Stream.map(fn %Ch.Result{data: data} ->
             data
             |> IO.iodata_to_binary()