lasp_bench

%% -------------------------------------------------------------------
%%
%% lasp_bench: Benchmarking Suite
%%
%% Copyright (c) 2009-2010 Basho Techonologies
%%
%% This file is provided to you under the Apache License,
%% Version 2.0 (the "License"); you may not use this file
%% except in compliance with the License.  You may obtain
%% a copy of the License at
%%
%%   http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing,
%% software distributed under the License is distributed on an
%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
%% KIND, either express or implied.  See the License for the
%% specific language governing permissions and limitations
%% under the License.
%%
%% -------------------------------------------------------------------
-module(lasp_bench_driver_dets).

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

-include("lasp_bench.hrl").

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

new(_Id) ->
    File = lasp_bench_config:get(dets_file, ?MODULE),
    {ok, _} = dets:open_file(?MODULE, [{file, File},
                                       {min_no_slots, 8192},
                                       {max_no_slots, 16777216}]),
    {ok, undefined}.

run(get, KeyGen, _ValueGen, State) ->
    Key = KeyGen(),
    case dets:lookup(?MODULE, Key) of
        [] ->
            {ok, State};
        [{Key, _}] ->
            {ok, State};
        {error, Reason} ->
            {error, Reason, State}
    end;
run(put, KeyGen, ValueGen, State) ->
    ok = dets:insert(?MODULE, {KeyGen(), ValueGen()}),
    {ok, State};
run(delete, KeyGen, _ValueGen, State) ->
    ok = dets:delete(?MODULE, KeyGen()),
    {ok, State}.