greg/hanoidb
1
0
Fork 0
Code Issues 10 Pull requests 3 Projects Releases Packages Wiki Activity

Fix problem in merge hibernation

Browse source 
The merge state includes an bloom reference,
which needed to be properly serialized.
This commit is contained in:
Kresten Krab Thorup 2012-04-29 00:55:26 +02:00
parent f0833de3fc
commit 77a81499f9
2 changed files with 20 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/hanoi_merger.erl
View file

@ -75,8 +75,8 @@ hibernate_scan(Keep) ->
receive
{step, From, HowMany} ->
error_logger:info_msg("waking up ~p~n", [self()]),
{BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count} = erlang:binary_to_term( zlib:gunzip( Keep ) ),
scan(BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count, {HowMany, From})
{BT1, BT2, OutBin, IsLastLevel, AKVs, BKVs, Count} = erlang:binary_to_term( zlib:gunzip( Keep ) ),
scan(BT1, BT2, hanoi_writer:deserialize(OutBin), IsLastLevel, AKVs, BKVs, Count, {HowMany, From})
end.
scan(BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count, {0, FromPID}) ->
@ -91,9 +91,14 @@ scan(BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count, {0, FromPID}) ->
{step, From, HowMany} ->
scan(BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count, {HowMany, From})
after 10000 ->
Args = {BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count},
Keep = zlib:gzip ( erlang:term_to_binary( Args ) ),
hibernate_scan(Keep)
case ?LOCAL_WRITER of
true ->
Args = {BT1, BT2, hanoi_writer:serialize(Out), IsLastLevel, AKVs, BKVs, Count},
Keep = zlib:gzip ( erlang:term_to_binary( Args ) ),
hibernate_scan(Keep);
false ->
scan(BT1, BT2, Out, IsLastLevel, AKVs, BKVs, Count, {0, none})
end
end;
scan(BT1, BT2, Out, IsLastLevel, [], BKVs, Count, Step) ->

11
src/hanoi_writer.erl
View file

@ -37,7 +37,7 @@
%% gen_server callbacks
-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
terminate/2, code_change/3]).
terminate/2, code_change/3, serialize/1, deserialize/1]).
-export([open/1, open/2, add/3,close/1]).
@ -126,6 +126,15 @@ code_change(_OldVsn, State, _Extra) ->
%%%%% INTERNAL FUNCTIONS
serialize(#state{ bloom=Bloom }=State) ->
erlang:term_to_binary( { State, ebloom:serialize(Bloom) } ).
deserialize(Binary) ->
{ State, BinBloom } = erlang:binary_to_term( Binary ),
{ok, Bloom } = ebloom:deserialize(BinBloom),
State#state{ bloom = Bloom }.
% @doc flush pending nodes and write trailer
flush_nodes(#state{ nodes=[], last_node_pos=LastNodePos, last_node_size=_LastNodeSize, bloom=Ref }=State) ->