Add test/mrdb_bench.erl, remove some debug logging

2022-04-07 18:48:24 +02:00
parent 2339232f59
commit 0830cecbf9
3 changed files with 85 additions and 37 deletions
@@ -0,0 +1,74 @@
+-module(mrdb_bench).
+
+-compile(export_all).
+
+init() ->
+    mnesia:delete_schema([node()]),
+    mnesia_rocksdb:create_schema([node()]),
+    mnesia:start(),
+    [mnesia:create_table(Name, [{Type, [node()]}, {record_name, r}])
+     || {Name, Type} <- tabs()],
+    ok.
+
+tabs() ->
+    [{rc, ram_copies},
+     {dc, disc_copies},
+     {do, disc_only_copies},
+     {rocks, rocksdb_copies},
+     {rdb, rocksdb_copies}].
+
+fill(N) ->
+    [{T, timer:tc(fun() -> fill_(T, N) end)} || {T,_} <- tabs()].
+
+fill_(_, 0) ->
+    ok;
+fill_(T, N) when N > 0 ->
+    write(T, {r, N, <<"1234567890">>}),
+    fill_(T, N-1).
+
+write(rdb, Obj) ->
+    mrdb:insert(rdb, Obj);
+write(T, Obj) ->
+    mnesia:dirty_write(T, Obj).
+
+
+fold() ->
+    [{T, timer:tc(fun() -> fold_(T) end)} || {T,_} <- tabs()].
+
+fold_(rdb) ->
+    mrdb:fold(rdb, fun(_, Acc) -> Acc end, ok);
+fold_(T) ->
+    mnesia:activity(
+      async_dirty,
+      fun() ->
+	      mnesia:foldl(fun(_, Acc) -> Acc end, ok, T)
+      end).
+				    
+tx(N) ->
+    [{T, timer:tc(fun() -> tx_(T, N) end)} || {T,_} <- tabs()].
+
+%% tx_(T, N) ->
+%% tx_(T, N, N, 10).
+
+tx_(_, 0) -> ok;
+tx_(T, N) when N > 0 ->
+    one_tx(T, N),
+    tx_(T, N-1).
+
+one_tx(rdb, N) ->
+    mrdb:activity(transaction, rocksdb_copies,
+		  fun() ->
+			  [{r, N, Str}] = mrdb:read(rdb, N),
+			  Str2 = <<Str/binary, "x">>,
+			  mrdb:insert(rdb, {r, N, Str2}),
+			  [{r, N, Str2}] = mrdb:read(rdb, N)
+		  end);
+one_tx(T, N) ->
+    mnesia:activity(transaction,
+		    fun() ->
+			    [{r, N, Str}] = mnesia:read(T, N),
+			    Str2 = <<Str/binary, "x">>,
+			    mnesia:write(T, {r, N, Str2}, write),
+			    [{r, N, Str2}] = mnesia:read(T, N)
+		    end).
+