first version, based on mnesia_eleveldb

2018-02-06 09:43:57 +01:00
commit 6c9f5b565f
30 changed files with 4835 additions and 0 deletions
@@ -0,0 +1,84 @@
+%%----------------------------------------------------------------
+%% Copyright (c) 2013-2016 Klarna AB
+%%
+%% This file is provided to you under the Apache License,
+%% Version 2.0 (the "License"); you may not use this file
+%% except in compliance with the License.  You may obtain
+%% a copy of the License at
+%%
+%%   http://www.apache.org/licenses/LICENSE-2.0
+%%
+%% Unless required by applicable law or agreed to in writing,
+%% software distributed under the License is distributed on an
+%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+%% KIND, either express or implied.  See the License for the
+%% specific language governing permissions and limitations
+%% under the License.
+%%----------------------------------------------------------------
+
+-module(mnesia_rocksdb_size_info).
+
+-export([run/0]).
+
+-define(m(A, B), (fun(L) -> {L,A} = {L,B} end)(?LINE)).
+
+
+run() ->
+    initialize_mnesia(),
+    test_set(),
+    test_bag().
+
+initialize_mnesia() ->
+    mnesia:stop(),
+    mnesia:delete_schema([node()]),
+    mnesia:create_schema([node()], [{backend_types,
+                                     [{rocksdb_copies, mnesia_rocksdb}]}]),
+    mnesia:start(),
+    {atomic,ok} = mnesia:create_table(s, [{type, set},
+                                          {record_name, x},
+                                          {rocksdb_copies, [node()]}]),
+    {atomic,ok} = mnesia:create_table(b, [{type, bag},
+                                          {record_name, x},
+                                          {rocksdb_copies, [node()]}]),
+    ok.
+
+test_set() ->
+    ?m(0, mnesia:table_info(s, size)),
+    ?m(1, w(s, 1, a)),
+    ?m(1, w(s, 1, b)),
+    ?m(2, w(s, 2, c)),
+    ?m(3, w(s, 3, d)),
+    ?m(2, d(s, 3)),
+    mnesia:stop(),
+    mnesia:start(),
+    await(s),
+    ?m(2, mnesia:table_info(s, size)).
+
+test_bag() ->
+    ?m(0, mnesia:table_info(b, size)),
+    ?m(1, w(b, 1, a)),
+    ?m(2, w(b, 1, b)),
+    ?m(3, w(b, 2, a)),
+    ?m(4, w(b, 2, d)),
+    ?m(5, w(b, 2, c)),
+    ?m(4, do(b, 2, c)),
+    ?m(2, d(b, 2)),
+    mnesia:stop(),
+    mnesia:start(),
+    await(b),
+    ?m(2, mnesia:table_info(b, size)).
+
+w(T, K, V) ->
+    ok = mnesia:dirty_write(T, {x, K, V}),
+    mnesia:table_info(T, size).
+
+d(T, K) ->
+    mnesia:dirty_delete({T, K}),
+    mnesia:table_info(T, size).
+
+do(T, K, V) ->
+    mnesia:dirty_delete_object(T, {x, K, V}),
+    mnesia:table_info(T, size).
+
+await(T) ->
+    ?m(ok, mnesia:wait_for_tables([T], 10000)).