lmdb-erl/tools/basho_bench_driver_lmdb.erl

-module(basho_bench_driver_lmdb).

-record(state, {
	  handle
	 }).

-export([new/1,
         run/4]).

-include_lib("basho_bench/include/basho_bench.hrl").


%% ====================================================================
%% API
%% ====================================================================

new(1) ->
    %% Make sure lmdb is available
    case code:which(lmdb) of
        non_existing ->
            ?FAIL_MSG("~s requires lmdb to be available on code path.\n",
                      [?MODULE]);
        _ ->
            ok
    end,
    %{ok, _} = lmdb_sup:start_link(),
    setup(1);
new(Id) ->
    setup(Id).

setup(_Id) ->
    %% Get the target directory
    Dir = basho_bench_config:get(lmdb_dir, "/tmp"),
    %Config = basho_bench_config:get(lmdb, []),

    %% Start Lightning MDB
    case lmdb:open(Dir, 32212254720, 16#10000 bor 16#40000 bor 16#80000) of
	{ok, H} ->
	    {ok, #state{handle=H}};
	{error, Reason} ->
	    ?FAIL_MSG("Failed to establish a Lightning MDB connection, lmdb backend unable to start: ~p\n", [Reason]),
	    {error, Reason}
    end.

run(get, KeyGen, _ValueGen, #state{handle=Handle}=State) ->
    case lmdb:get(Handle, KeyGen()) of
        {ok, _Value} ->
            {ok, State};
        not_found ->
            {ok, State};
        {error, Reason} ->
            {error, Reason}
    end;
run(put, KeyGen, ValueGen, #state{handle=Handle}=State) ->
    Key = KeyGen(),
    Val = ValueGen(),
    case lmdb:upd(Handle, Key, Val) of
        ok ->
            {ok, State};
        {error, Reason} ->
            {error, Reason}
    end;
run(delete, KeyGen, _ValueGen, #state{handle=Handle}=State) ->
    case lmdb:del(Handle, KeyGen()) of
        ok ->
            {ok, State};
        not_found ->
            {ok, State};
        {error, Reason} ->
            {error, Reason}
    end.

%% config_value(Key, Config, Default) ->
%%     case proplists:get_value(Key, Config) of
%%         undefined ->
%%             Default;
%%         Value ->
%%             Value
%%     end.