simple epgsql bench

epgsql_bench.erl

-module(epgsql_bench).

-export([run/2]).

-define(HOST, "").
-define(PORT, "").
-define(USER, "").
-define(PW, "").
-define(DB, "").

run(N, QueueLen) ->
    Ref = epgsqla:connect(?HOST, ?USER, ?PW, [{database, ?DB}, {port, ?PORT}]),
    {C, Ref, connected} = receive A -> A end,
    {Time, _} = timer:tc(fun() ->
        loop(C, N, QueueLen)
    end),
    io:format("~p ~p ops/s~n", [Time, N * 1000000 / Time]),
    epgsqla:close(C),
    ok.

loop(C, N, QueueLen) ->
    send(C, QueueLen),
    loop(C, N, QueueLen, 0).

loop(_C, N, N, N) ->
    ok;
loop(C, N, Sent, Recv) ->
    RecvCount = recv(),
    SendCount = min(N-Sent, RecvCount),
    send(C, SendCount),
    loop(C, N, Sent + SendCount, Recv + RecvCount).

send(_C, 0) ->
    ok;
send(C, N) ->
    send(C),
    send(C, N-1).

send(C) ->
    Name = base64:encode(crypto:strong_rand_bytes(10)),
    epgsqla:parse(C, Name, <<"select 1">>, []).

recv() ->
    recv_stmt(0, 1000).

recv_stmt(Count, Timeout) ->
    receive
        {C, _, {ok, {statement, _, _, _} = Stmt}} ->
            epgsqla:equery(C, Stmt, []),
            recv_stmt(Count, 1000);
        _Resp ->
            recv_stmt(Count+1, 0)
    after Timeout ->
        Count
    end.

Revised version with prepared_query/3

-module(epgsql_bench).

-export([run/2]).

-define(HOST, "localhost").
-define(PORT, 5432).
-define(USER, "").
-define(PW, "").
-define(DB, "").

run(N, QueueLen) ->
    Ref = epgsqla:connect(?HOST, ?USER, ?PW, [{database, ?DB}, {port, ?PORT}]),
    {C, Ref, connected} = receive A -> A end,
    {Time, _} = timer:tc(fun() ->
        loop(C, N, QueueLen)
    end),
    io:format("~p ~p ops/s~n", [Time, N * 1000000 / Time]),
    epgsqla:close(C),
    ok.

loop(C, N, QueueLen) ->
    {ok, Stmt} = prepare(C),
    send(C, Stmt, QueueLen),
    loop(C, Stmt, N, QueueLen, 0).

loop(_C, _Stmt, N, N, N) ->
    ok;
loop(C, Stmt, N, Sent, Recv) ->
    RecvCount = recv(),
    SendCount = min(N-Sent, RecvCount),
    send(C, Stmt, SendCount),
    loop(C, Stmt, N, Sent + SendCount, Recv + RecvCount).

prepare(C) ->
    Name = base64:encode(crypto:strong_rand_bytes(10)),
    epgsqla:parse(C, Name, <<"select 1">>, []),
    receive
        {C, _, {ok, {statement, _, _, _} = Stmt}} ->
            {ok, Stmt}
    end.

send(_C, _Stmt, 0) ->
    ok;
send(C, Stmt, N) ->
    epgsqla:prepared_query(C, Stmt, []),
    send(C, Stmt, N-1).

recv() ->
    recv_stmt(0, 1000).

recv_stmt(Count, Timeout) ->
    receive
        _Resp ->
            recv_stmt(Count+1, 0)
    after Timeout ->
        Count
    end.