Refactor pool and add transaction

Author: Schultzer

benchmarks/bench.exs

@@ -1,44 +1,39 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-FileCopyrightText: 2025 DBVisor
 
-import SQL
+# import SQL
 import Ecto.Query
 defmodule SQL.Repo do
   use Ecto.Repo, otp_app: :sql, adapter: Ecto.Adapters.Postgres
-  use SQL, adapter: SQL.Adapters.Postgres, repo: __MODULE__
-
+  use SQL, adapter: SQL.Adapters.Postgres
+  import Ecto.Query
   def sql() do
-    case SQL.Pool.checkout(%{id: 1}, :default) do
-      {idx, _socket} ->
-        Process.sleep(5)
-        SQL.Pool.checkin(idx, :default)
-      :none ->
-        Process.sleep(5)
-    end
+    # SQL.transaction do
+    #   SQL.transaction do
+        Enum.to_list(~SQL"SELECT 1")
+    #   end
+    # end
   end
 
   def ecto() do
-    try do
-      checkout(fn ->
-        Process.sleep(5)
-      end)
-    rescue
-      _ -> :error
-        Process.sleep(5)
-    end
+    # SQL.Repo.transaction(fn ->
+    #   SQL.Repo.transaction(fn ->
+        SQL.Repo.all(select(from("users"), [1]))
+    #   end)
+    # 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: 10)
+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: 10, ssl: false)
 SQL.Repo.__adapter__().storage_up(SQL.Repo.config())
 SQL.Repo.start_link()
-SQL.Pool.start_link(%{name: :default, size: 10, protocol: :tcp})
-query = "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])
-sql = ~SQL[with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)]
-result = Tuple.to_list(SQL.Lexer.lex("with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)"))
-tokens = Enum.at(result, -1)
-context = Map.merge(Enum.at(result, 1), %{sql_lock: nil, module: SQL.Adapters.ANSI})
-{:ok, pcontext, ptokens} = SQL.Parser.parse(tokens, context)
+SQL.Pool.start_link(%{username: "postgres", password: "postgres", hostname: "localhost", database: "sql_test#{System.get_env("MIX_TEST_PARTITION")}", adapter: SQL.Adapters.Postgres, ssl: false})
+# query = "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])
+# sql = ~SQL[with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)]
+# result = Tuple.to_list(SQL.Lexer.lex("with recursive temp (n, fact) as (select 0, 1 union all select n+1, (n+1)*fact from temp where n < 9)"))
+# tokens = Enum.at(result, -1)
+# context = Map.merge(Enum.at(result, 1), %{sql_lock: nil, module: SQL.Adapters.ANSI})
+# {:ok, pcontext, ptokens} = SQL.Parser.parse(tokens, context)
 
 Benchee.run(
   %{
@@ -55,19 +50,20 @@ 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() end,
+  "ecto" => fn -> SQL.Repo.ecto() 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: 10, time: 1,
-  inputs: %{"1..100_000" => Enum.to_list(1..100_000)},
+  parallel: 50,
   memory_time: 2,
   reduction_time: 2,
   unit_scaling: :smallest,
   measure_function_call_overhead: true,
-  profile_after: :eprof
+  # profile_after: :eprof
+  # profile_after: :cprof
+  profile_after: :fprof
 )