Handle error from db:find/1.

1 files changed, 9 insertions, 10 deletions

diff --git a/src/plop.erl b/src/plop.erl
index d7d25ba..ba74ead 100644
--- a/src/plop.erl
+++ b/src/plop.erl
@@ -146,28 +146,27 @@ do_add(TimestampedEntry = #timestamped_entry{entry = PlopEntry},
         #plop{index = I, mtl = #mtl{entry = E}, spt = SPT} ->
             io:format("Found entry: index=~p, data=~p~n",
                       [I, E#timestamped_entry.entry#plop_entry.data]),
-            Record = Record#plop{           % DB consistency checking.
+            %% Database consistency checking. FIXME: Remove.
+            Record = Record#plop{
                        hash = DB_hash,
                        mtl = #mtl{entry =
                                       #timestamped_entry{
                                          timestamp = E#timestamped_entry.timestamp,
-                                         entry = PlopEntry}
-                                 }},
+                                         entry = PlopEntry}}},
             {Tree, SPT};              % State not changed, cached SPT.
-        _ ->
+        [] ->
             NewSPT = spt(LogID, Privkey, TimestampedEntry),
             MTL = #mtl{entry = TimestampedEntry},
             io:format("Creating new entry: index=~p,data=~p~n",
-                      [ht:size(Tree) + 1, PlopEntry#plop_entry.data]),
-            %% io:format("Creating new entry: index=~p~ndb hash: ~p~nMTL: ~p~nSPT: ~p~n",
-            %%           [ht:size(Tree) + 1, DB_hash, MTL, NewSPT]),
-            DB_data = #plop{index = ht:size(Tree) + 1,
+                      [ht:size(Tree), PlopEntry#plop_entry.data]),
+            DB_data = #plop{index = ht:size(Tree),
                             hash = DB_hash,
                             mtl = MTL,
                             spt = NewSPT},
-            db:add(DB_data),
+            ok = db:add(DB_data),
             {ht:append(Tree, serialise(MTL)),   % New tree.
-             NewSPT}                            % New SPT.
+             NewSPT};                           % New SPT.
+        Err -> {error, Err}
     end.
     
 %% @doc Signed Plop Timestamp, conformant to an SCT in RFC6962 3.2 and