From cc037396bf405ec942352ea834e137ea1b24e6af Mon Sep 17 00:00:00 2001
From: Magnus Ahltorp <map@kth.se>
Date: Fri, 13 May 2016 17:20:08 +0200
Subject: Added 'keyexists' command to permdb

---
 src/permdb.erl | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/permdb.erl b/src/permdb.erl
index 16ba28a..461b8b3 100644
--- a/src/permdb.erl
+++ b/src/permdb.erl
@@ -6,7 +6,7 @@
 -behaviour(gen_server).
 
 -export([start_link/2, stop/1, init_module/0]).
--export([getvalue/2, addvalue/3, commit/1, commit/2]).
+-export([getvalue/2, addvalue/3, commit/1, commit/2, keyexists/2]).
 
 %% gen_server callbacks.
 -export([init/1, handle_call/3, terminate/2, handle_cast/2, handle_info/2,
@@ -23,9 +23,15 @@ addvalue_port_command(Port, Key, Value) ->
 commit_port_command(Port) ->
     Port ! {self(), {command, <<2:8>>}}.
 
+keyexists_port_command(Port, Key) ->
+    Port ! {self(), {command, <<3:8, Key/binary>>}}.
+
 getvalue(Name, Key) ->
     gen_server:call(Name, {getvalue, Key}, 600000).
 
+keyexists(Name, Key) ->
+    gen_server:call(Name, {keyexists, Key}, 600000).
+
 addvalue(Name, Key, Value) ->
     gen_server:call(Name, {addvalue, Key, Value}).
 
@@ -74,12 +80,19 @@ handle_info({Port, {data, Data}}, State) when is_port(Port) ->
                                addvalue ->
                                    case Data of
                                        <<>> ->
-                                           util:exit_with_error(putvalue, unknown, "Error in putvalue");
+                                           util:exit_with_error(addvalue, unknown, "Error in addvalue");
                                        _ ->
                                            ok
                                    end;
                                commit ->
-                                   Data
+                                   Data;
+                               keyexists ->
+                                   case Data of
+                                       <<0>> ->
+                                           false;
+                                       <<1>> ->
+                                           true
+                                   end
                            end),
     {noreply, State#state{requests = Requests}};
 handle_info(_Info, State) ->
@@ -115,4 +128,9 @@ handle_call({addvalue, Key, Value}, From, State) ->
 handle_call({commit}, From, State) ->
     lager:debug("commit ~p ~p", [State#state.name, State#state.requestcounter]),
     commit_port_command(State#state.port),
-    {noreply, add_request(State, From, commit)}.
+    {noreply, add_request(State, From, commit)};
+
+handle_call({keyexists, Key}, From, State) ->
+    lager:debug("keyexists ~p ~p: ~p", [State#state.name, State#state.requestcounter, Key]),
+    keyexists_port_command(State#state.port, Key),
+    {noreply, add_request(State, From, keyexists)}.
-- 
cgit v1.1