Current section

Files

Jump to
db_connection lib db_connection task.ex
Raw

lib/db_connection/task.ex

defmodule DBConnection.Task do
@moduledoc false
def start_link() do
Task.Supervisor.start_link([name: __MODULE__])
end
def run_child(mod, fun, state, opts) do
ref = make_ref()
arg = [mod, fun, ref, self(), state, opts]
{:ok, pid} = Task.Supervisor.start_child(__MODULE__, __MODULE__, :init, arg)
mon = Process.monitor(pid)
send(pid, {:go, ref, mon})
mon
end
def init(mod, fun, ref, conn, state, opts) do
Process.link(conn)
receive do
{:go, ^ref, mon} ->
Process.unlink(conn)
pool = {:via, __MODULE__, {{conn, mon}, mod, state}}
_ = DBConnection.run(pool, make(fun), [pool: __MODULE__] ++ opts)
exit(:normal)
end
end
def checkout({:via, __MODULE__, {info, mod, state}}, _) do
{:ok, info, mod, state}
end
defdelegate checkin(info, state, opts), to: DBConnection.Connection
defdelegate disconnect(info, err, state, opts), to: DBConnection.Connection
defdelegate stop(info, reason, state, opts), to: DBConnection.Connection
defp make(fun) when is_function(fun, 1) do
fun
end
defp make(mfargs) do
fn(conn) ->
{mod, fun, args} = mfargs
apply(mod, fun, [conn | args])
end
end
end