From 8c87e99fbfab0ab26c73f42be40e8b1a3ae6a181 Mon Sep 17 00:00:00 2001
From: Linus Nordberg <linus@nordberg.se>
Date: Wed, 10 Sep 2014 14:01:02 +0200
Subject: Add size/1.

---
 src/ts.erl | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'src/ts.erl')

diff --git a/src/ts.erl b/src/ts.erl
index 8ea4895..345ed66 100644
--- a/src/ts.erl
+++ b/src/ts.erl
@@ -2,7 +2,7 @@
 -module(ts).
 -include_lib("eunit/include/eunit.hrl").
 -export_type([tree_store/0]).
--export([new/0, delete/1, store/3, retrieve/2, retrieve_hash/2]).
+-export([new/0, delete/1, size/1, store/3, retrieve/2, retrieve_hash/2]).
 
 %% -record(tree_store, {warm :: ets:tid(),
 %%                      frozen :: list()}).  % [ets:tid()]
@@ -15,7 +15,10 @@ new() ->
     #tree_store{table = ets:new(nil, [{read_concurrency, true}])}.
 
 delete(Store) ->
-    ets:delete(Store#tree_store.table).    
+    ets:delete(Store#tree_store.table).
+
+size(Store) ->
+    ets:info(Store#tree_store.table, size).
 
 -spec store(tree_store(), tuple(), binary()) -> tree_store().
 store(Store, IR, Hash) ->
-- 
cgit v1.1