Add deletion support to the driver.

eriksoe · Jan 7, 2012 · c80d7d1 · c80d7d1
1 parent e102084
commit c80d7d1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/fractal_btree_drv.erl b/test/fractal_btree_drv.erl
@@ -7,6 +7,7 @@
 -export([start_link/0]).
 
 -export([
+         delete_exist/2,
          lookup_exist/2,
          open/1,
          put/3,
@@ -32,6 +33,9 @@ call(X) ->
 lookup_exist(N, K) ->
     call({lookup_exist, N, K}).
 
+delete_exist(N, K) ->
+    call({delete_exist, N, K}).
+
 open(N) ->
     call({open, N}).
 
@@ -61,6 +65,10 @@ handle_call({put, N, K, V}, _, #state { btrees = D} = State) ->
         Other ->
             {reply, {error, Other}, State}
     end;
+handle_call({delete_exist, N, K}, _, #state { btrees = D} = State) ->
+    Tree = dict:fetch(N, D),
+    Reply = fractal_btree:delete(Tree, K),
+    {reply, Reply, State};
 handle_call({lookup_exist, N, K}, _, #state { btrees = D} = State) ->
     Tree = dict:fetch(N, D),
     Reply = fractal_btree:lookup(Tree, K),