refactor pool

Author: Schultzer

benchmarks/bench.exs

@@ -7,21 +7,61 @@ defmodule SQL.Repo do
   use Ecto.Repo, otp_app: :sql, adapter: Ecto.Adapters.Postgres
   use SQL, adapter: SQL.Adapters.Postgres
   import Ecto.Query
-  def sql() do
+  def sql(type \\ :transaction)
+  def sql(:statement) do
+    Enum.to_list(~SQL"SELECT 1")
+  end
+  def sql(:empty) do
+    SQL.transaction do
+      :ok
+    end
+  end
+  def sql(:transaction) do
+    SQL.transaction do
+      Enum.to_list(~SQL"SELECT 1")
+    end
+  end
+  def sql(:savepoint) do
     SQL.transaction do
       SQL.transaction do
         Enum.to_list(~SQL"SELECT 1")
       end
     end
   end
+  def sql(:cursor) do
+    SQL.transaction do
+      Stream.run(~SQL"SELECT g, repeat(md5(g::text), 4) FROM generate_series(1, 5000000) AS g")
+    end
+  end
 
-  def ecto() do
+  def ecto(type \\ :transaction)
+  def ecto(:statement) do
+    SQL.Repo.all(select(from("users"), [1]))
+  end
+  def ecto(:empty) do
+    SQL.Repo.transaction(fn ->
+      :ok
+    end)
+  end
+  def ecto(:transaction) do
+    SQL.Repo.transaction(fn ->
+      SQL.Repo.all(select(from("users"), [1]))
+    end)
+  end
+  def ecto(:savepoint) do
     SQL.Repo.transaction(fn ->
       SQL.Repo.transaction(fn ->
         SQL.Repo.all(select(from("users"), [1]))
       end)
     end)
   end
+  def ecto(:cursor) do
+    SQL.Repo.transaction(fn ->
+      from(row in fragment("SELECT g, repeat(md5(g::text), 4) FROM generate_series(1, ?) AS g", 5000000), select: {fragment("?::int", row.g), fragment("?::text", row.repeat)})
+      |> SQL.Repo.stream()
+      |> Stream.run()
+    end)
+  end
 end
 Application.put_env(:sql, :ecto_repos, [SQL.Repo])
 Application.put_env(:sql, SQL.Repo, log: false, username: "postgres", password: "postgres", hostname: "localhost", database: "sql_test#{System.get_env("MIX_TEST_PARTITION")}", pool_size: :erlang.system_info(:schedulers_online), ssl: false)
@@ -49,20 +89,31 @@ Benchee.run(
   #   pcontext.module.to_iodata(tokens, pcontext)
   # end,
   # "parse/3" => fn _ -> SQL.parse("with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)") end,
-  "sql" => fn -> SQL.Repo.sql() end,
-  "ecto" => fn -> SQL.Repo.ecto() end,
+  # "sql" => fn -> SQL.Repo.sql(Enum.random([:statement, :transaction])) end,
+  # "ecto" => fn -> SQL.Repo.ecto(Enum.random([:statement, :transaction])) end,
+  "sql" => fn -> SQL.Repo.sql(:transaction) end,
+  "ecto" => fn -> SQL.Repo.ecto(:transaction) end,
+  # "sql" => fn -> SQL.Repo.sql(:transaction) end,
+  # "ecto" => fn -> SQL.Repo.ecto(:transaction) end,
+  # "sql" => fn -> SQL.Repo.sql(:savepoint) end,
+  # "ecto" => fn -> SQL.Repo.ecto(:savepoint) end,
+  # "sql" => fn -> SQL.Repo.sql(:cursor) end,
+  # "ecto" => fn -> SQL.Repo.ecto(:cursor) end,
+  # "sql" => fn -> SQL.Repo.sql(:empty) end,
+  # "ecto" => fn -> SQL.Repo.ecto(:empty) end,
   # "runtime to_string" => fn _ -> to_string(~SQL[with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)]) end,
   # "runtime to_sql" => fn _ -> SQL.to_sql(~SQL[with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)]) end,
   # "runtime inspect" => fn _ -> inspect(~SQL[with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)]) end,
   # "runtime ecto" => fn _ -> SQL.Repo.to_sql(:all, "temp" |> recursive_ctes(true) |> with_cte("temp", as: ^union_all(select("temp", [t], %{n: 0, fact: 1}), ^where(select("temp", [t], [t.n+1, t.n+1*t.fact]), [t], t.n < 9))) |> select([t], [t.n])) end,
   # "comptime ecto" => fn _ -> SQL.Repo.to_sql(:all, query) end
   },
-  parallel: 1,
+  parallel: 500,
+  warmup: 10,
   memory_time: 2,
   reduction_time: 2,
   unit_scaling: :smallest,
-  measure_function_call_overhead: true,
+  measure_function_call_overhead: true
   # profile_after: :eprof
   # profile_after: :cprof
-  profile_after: :fprof
+  # profile_after: :fprof
 )