2017-01-09 14:52:26 +00:00
|
|
|
%% -------- PENCILLER MANIFEST ---------
|
|
|
|
%%
|
|
|
|
%% The manifest is an ordered set of files for each level to be used to find
|
|
|
|
%% which file is relevant for a given key or range lookup at a given level.
|
|
|
|
%%
|
2017-01-17 21:09:42 +00:00
|
|
|
%% This implementation is incomplete, in that it just uses a plain list at
|
|
|
|
%% each level. This is fine for short-lived volume tests, but as the deeper
|
|
|
|
%% levels are used there will be an exponential penalty.
|
|
|
|
%%
|
|
|
|
%% The originial intention was to swap out this implementation for a
|
|
|
|
%% multi-version ETS table - but that became complex. So one of two changes
|
|
|
|
%% are pending:
|
|
|
|
%% - Use a single version ES cache for lower levels (and not allow snapshots to
|
|
|
|
%% access the cache)
|
|
|
|
%% - Use a skiplist like enhanced list at lower levels.
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-17 14:11:50 +00:00
|
|
|
-module(leveled_pmanifest).
|
2017-01-09 14:52:26 +00:00
|
|
|
|
|
|
|
-include("include/leveled.hrl").
|
|
|
|
|
|
|
|
-export([
|
|
|
|
new_manifest/0,
|
|
|
|
open_manifest/1,
|
2017-01-13 18:23:57 +00:00
|
|
|
copy_manifest/1,
|
|
|
|
load_manifest/3,
|
2017-01-17 10:12:15 +00:00
|
|
|
close_manifest/2,
|
2017-01-13 18:23:57 +00:00
|
|
|
save_manifest/2,
|
|
|
|
get_manifest_sqn/1,
|
|
|
|
key_lookup/3,
|
|
|
|
range_lookup/4,
|
2017-01-17 10:12:15 +00:00
|
|
|
merge_lookup/4,
|
2017-01-09 14:52:26 +00:00
|
|
|
insert_manifest_entry/4,
|
|
|
|
remove_manifest_entry/4,
|
2017-01-23 11:02:54 +00:00
|
|
|
replace_manifest_entry/5,
|
2017-01-16 23:27:42 +00:00
|
|
|
switch_manifest_entry/4,
|
2017-01-13 18:23:57 +00:00
|
|
|
mergefile_selector/2,
|
|
|
|
add_snapshot/3,
|
2017-01-09 14:52:26 +00:00
|
|
|
release_snapshot/2,
|
2017-01-13 18:23:57 +00:00
|
|
|
ready_to_delete/2,
|
|
|
|
check_for_work/2,
|
|
|
|
is_basement/2,
|
2017-01-16 23:27:42 +00:00
|
|
|
levelzero_present/1
|
2017-01-09 14:52:26 +00:00
|
|
|
]).
|
|
|
|
|
2017-01-14 16:36:05 +00:00
|
|
|
-export([
|
|
|
|
filepath/2
|
|
|
|
]).
|
|
|
|
|
2017-01-09 14:52:26 +00:00
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
|
|
|
|
-define(MANIFEST_FILEX, "man").
|
|
|
|
-define(MANIFEST_FP, "ledger_manifest").
|
2017-01-13 18:23:57 +00:00
|
|
|
-define(MAX_LEVELS, 8).
|
2017-01-22 23:36:16 +00:00
|
|
|
-define(TREE_TYPE, idxt).
|
|
|
|
-define(TREE_WIDTH, 8).
|
2017-01-13 18:23:57 +00:00
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
-record(manifest, {levels,
|
|
|
|
% an array of lists or trees representing the manifest
|
2017-01-13 18:23:57 +00:00
|
|
|
manifest_sqn = 0 :: integer(),
|
|
|
|
% The current manifest SQN
|
|
|
|
snapshots :: list(),
|
|
|
|
% A list of snaphots (i.e. clones)
|
2017-01-16 23:27:42 +00:00
|
|
|
min_snapshot_sqn = 0 :: integer(),
|
|
|
|
% The smallest snapshot manifest SQN in the snapshot
|
|
|
|
% list
|
2017-01-17 19:00:30 +00:00
|
|
|
pending_deletes, % OTP16 does not like defining type
|
2017-01-17 16:54:54 +00:00
|
|
|
% a dictionary mapping keys (filenames) to SQN when the
|
|
|
|
% deletion was made, and the original Manifest Entry
|
2017-01-15 12:41:16 +00:00
|
|
|
basement :: integer()
|
|
|
|
% Currently the lowest level (the largest number)
|
2017-01-13 18:23:57 +00:00
|
|
|
}).
|
2017-01-09 14:52:26 +00:00
|
|
|
|
|
|
|
%%%============================================================================
|
|
|
|
%%% API
|
|
|
|
%%%============================================================================
|
|
|
|
|
|
|
|
new_manifest() ->
|
2017-01-22 23:36:16 +00:00
|
|
|
LevelArray0 = array:new([{size, ?MAX_LEVELS + 1}, {default, []}]),
|
|
|
|
SetLowerLevelFun =
|
|
|
|
fun(IDX, Acc) ->
|
|
|
|
array:set(IDX, leveled_tree:empty(?TREE_TYPE), Acc)
|
|
|
|
end,
|
|
|
|
LevelArray1 = lists:foldl(SetLowerLevelFun,
|
|
|
|
LevelArray0,
|
|
|
|
lists:seq(2, ?MAX_LEVELS)),
|
2017-01-16 23:27:42 +00:00
|
|
|
#manifest{
|
2017-01-22 23:36:16 +00:00
|
|
|
levels = LevelArray1,
|
2017-01-16 23:27:42 +00:00
|
|
|
manifest_sqn = 0,
|
|
|
|
snapshots = [],
|
|
|
|
pending_deletes = dict:new(),
|
|
|
|
basement = 0
|
|
|
|
}.
|
2017-01-09 14:52:26 +00:00
|
|
|
|
|
|
|
open_manifest(RootPath) ->
|
|
|
|
% Open the manifest in the file path which has the highest SQN, and will
|
|
|
|
% open without error
|
|
|
|
ManifestPath = filepath(RootPath, manifest),
|
|
|
|
{ok, Filenames} = file:list_dir(ManifestPath),
|
|
|
|
CurrRegex = "nonzero_(?<MSN>[0-9]+)\\." ++ ?MANIFEST_FILEX,
|
|
|
|
ExtractSQNFun =
|
|
|
|
fun(FN, Acc) ->
|
|
|
|
case re:run(FN, CurrRegex, [{capture, ['MSN'], list}]) of
|
|
|
|
nomatch ->
|
|
|
|
Acc;
|
|
|
|
{match, [Int]} when is_list(Int) ->
|
|
|
|
Acc ++ [list_to_integer(Int)]
|
|
|
|
end
|
|
|
|
end,
|
|
|
|
ValidManSQNs = lists:reverse(lists:sort(lists:foldl(ExtractSQNFun,
|
|
|
|
[],
|
|
|
|
Filenames))),
|
2017-01-17 10:12:15 +00:00
|
|
|
open_manifestfile(RootPath, ValidManSQNs).
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
copy_manifest(Manifest) ->
|
|
|
|
% Copy the manifest ensuring anything only the master process should care
|
|
|
|
% about is switched to undefined
|
2017-01-16 23:27:42 +00:00
|
|
|
Manifest#manifest{snapshots = undefined, pending_deletes = undefined}.
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
load_manifest(Manifest, PidFun, SQNFun) ->
|
2017-01-16 23:27:42 +00:00
|
|
|
UpdateLevelFun =
|
|
|
|
fun(LevelIdx, {AccMaxSQN, AccMan}) ->
|
|
|
|
L0 = array:get(LevelIdx, AccMan#manifest.levels),
|
|
|
|
{L1, SQN1} = load_level(LevelIdx, L0, PidFun, SQNFun),
|
2017-01-17 10:12:15 +00:00
|
|
|
UpdLevels = array:set(LevelIdx, L1, AccMan#manifest.levels),
|
|
|
|
{max(AccMaxSQN, SQN1), AccMan#manifest{levels = UpdLevels}}
|
2017-01-09 15:19:11 +00:00
|
|
|
end,
|
2017-01-16 23:27:42 +00:00
|
|
|
lists:foldl(UpdateLevelFun, {0, Manifest},
|
|
|
|
lists:seq(0, Manifest#manifest.basement)).
|
2017-01-17 10:12:15 +00:00
|
|
|
|
|
|
|
close_manifest(Manifest, CloseEntryFun) ->
|
|
|
|
CloseLevelFun =
|
|
|
|
fun(LevelIdx) ->
|
|
|
|
Level = array:get(LevelIdx, Manifest#manifest.levels),
|
|
|
|
close_level(LevelIdx, Level, CloseEntryFun)
|
|
|
|
end,
|
2017-01-17 16:54:54 +00:00
|
|
|
lists:foreach(CloseLevelFun, lists:seq(0, Manifest#manifest.basement)),
|
|
|
|
|
|
|
|
ClosePDFun =
|
|
|
|
fun({_FN, {_SQN, ME}}) ->
|
|
|
|
CloseEntryFun(ME)
|
|
|
|
end,
|
|
|
|
lists:foreach(ClosePDFun, dict:to_list(Manifest#manifest.pending_deletes)).
|
2017-01-17 10:12:15 +00:00
|
|
|
|
2017-01-13 18:23:57 +00:00
|
|
|
save_manifest(Manifest, RootPath) ->
|
|
|
|
FP = filepath(RootPath, Manifest#manifest.manifest_sqn, current_manifest),
|
2017-01-17 17:15:04 +00:00
|
|
|
ManBin = term_to_binary(Manifest#manifest{snapshots = [],
|
|
|
|
pending_deletes = dict:new(),
|
|
|
|
min_snapshot_sqn = 0}),
|
2017-01-16 23:27:42 +00:00
|
|
|
CRC = erlang:crc32(ManBin),
|
|
|
|
ok = file:write_file(FP, <<CRC:32/integer, ManBin/binary>>).
|
|
|
|
|
2017-01-23 11:02:54 +00:00
|
|
|
|
|
|
|
replace_manifest_entry(Manifest, ManSQN, LevelIdx, Removals, []) ->
|
|
|
|
remove_manifest_entry(Manifest, ManSQN, LevelIdx, Removals);
|
|
|
|
replace_manifest_entry(Manifest, ManSQN, LevelIdx, Removals, Additions) ->
|
|
|
|
Levels = Manifest#manifest.levels,
|
|
|
|
Level = array:get(LevelIdx, Levels),
|
|
|
|
UpdLevel = replace_entry(LevelIdx, Level, Removals, Additions),
|
|
|
|
leveled_log:log("PC019", ["insert", LevelIdx, UpdLevel]),
|
|
|
|
PendingDeletes = update_pendingdeletes(ManSQN,
|
|
|
|
Removals,
|
|
|
|
Manifest#manifest.pending_deletes),
|
|
|
|
UpdLevels = array:set(LevelIdx, UpdLevel, Levels),
|
|
|
|
case is_empty(LevelIdx, UpdLevel) of
|
|
|
|
true ->
|
|
|
|
Manifest#manifest{levels = UpdLevels,
|
|
|
|
basement = get_basement(UpdLevels),
|
|
|
|
manifest_sqn = ManSQN,
|
|
|
|
pending_deletes = PendingDeletes};
|
|
|
|
false ->
|
|
|
|
Basement = max(LevelIdx, Manifest#manifest.basement),
|
|
|
|
Manifest#manifest{levels = UpdLevels,
|
|
|
|
basement = Basement,
|
|
|
|
manifest_sqn = ManSQN,
|
|
|
|
pending_deletes = PendingDeletes}
|
|
|
|
end.
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
insert_manifest_entry(Manifest, ManSQN, LevelIdx, Entry) ->
|
|
|
|
Levels = Manifest#manifest.levels,
|
|
|
|
Level = array:get(LevelIdx, Levels),
|
|
|
|
UpdLevel = add_entry(LevelIdx, Level, Entry),
|
2017-01-17 10:32:15 +00:00
|
|
|
leveled_log:log("PC019", ["insert", LevelIdx, UpdLevel]),
|
2017-01-16 23:27:42 +00:00
|
|
|
Basement = max(LevelIdx, Manifest#manifest.basement),
|
|
|
|
Manifest#manifest{levels = array:set(LevelIdx, UpdLevel, Levels),
|
|
|
|
basement = Basement,
|
|
|
|
manifest_sqn = ManSQN}.
|
|
|
|
|
|
|
|
remove_manifest_entry(Manifest, ManSQN, LevelIdx, Entry) ->
|
|
|
|
Levels = Manifest#manifest.levels,
|
|
|
|
Level = array:get(LevelIdx, Levels),
|
|
|
|
UpdLevel = remove_entry(LevelIdx, Level, Entry),
|
2017-01-17 10:32:15 +00:00
|
|
|
leveled_log:log("PC019", ["remove", LevelIdx, UpdLevel]),
|
2017-01-23 11:02:54 +00:00
|
|
|
PendingDeletes = update_pendingdeletes(ManSQN,
|
|
|
|
Entry,
|
|
|
|
Manifest#manifest.pending_deletes),
|
2017-01-16 23:27:42 +00:00
|
|
|
UpdLevels = array:set(LevelIdx, UpdLevel, Levels),
|
|
|
|
case is_empty(LevelIdx, UpdLevel) of
|
|
|
|
true ->
|
|
|
|
Manifest#manifest{levels = UpdLevels,
|
|
|
|
basement = get_basement(UpdLevels),
|
|
|
|
manifest_sqn = ManSQN,
|
|
|
|
pending_deletes = PendingDeletes};
|
|
|
|
false ->
|
|
|
|
Manifest#manifest{levels = UpdLevels,
|
|
|
|
manifest_sqn = ManSQN,
|
|
|
|
pending_deletes = PendingDeletes}
|
|
|
|
end.
|
|
|
|
|
|
|
|
switch_manifest_entry(Manifest, ManSQN, SrcLevel, Entry) ->
|
|
|
|
% Move to level below - so needs to be removed but not marked as a
|
|
|
|
% pending deletion
|
|
|
|
Levels = Manifest#manifest.levels,
|
|
|
|
Level = array:get(SrcLevel, Levels),
|
|
|
|
UpdLevel = remove_entry(SrcLevel, Level, Entry),
|
|
|
|
UpdLevels = array:set(SrcLevel, UpdLevel, Levels),
|
|
|
|
insert_manifest_entry(Manifest#manifest{levels = UpdLevels},
|
|
|
|
ManSQN,
|
|
|
|
SrcLevel + 1,
|
|
|
|
Entry).
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
get_manifest_sqn(Manifest) ->
|
|
|
|
Manifest#manifest.manifest_sqn.
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
key_lookup(Manifest, LevelIdx, Key) ->
|
|
|
|
case LevelIdx > Manifest#manifest.basement of
|
2017-01-15 12:41:16 +00:00
|
|
|
true ->
|
2017-01-13 18:23:57 +00:00
|
|
|
false;
|
2017-01-15 12:41:16 +00:00
|
|
|
false ->
|
2017-01-16 23:27:42 +00:00
|
|
|
key_lookup_level(LevelIdx,
|
|
|
|
array:get(LevelIdx, Manifest#manifest.levels),
|
|
|
|
Key)
|
2017-01-13 18:23:57 +00:00
|
|
|
end.
|
2017-01-17 10:12:15 +00:00
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
range_lookup(Manifest, LevelIdx, StartKey, EndKey) ->
|
|
|
|
MakePointerFun =
|
|
|
|
fun(M) ->
|
|
|
|
{next, M, StartKey}
|
2017-01-13 18:23:57 +00:00
|
|
|
end,
|
2017-01-17 10:12:15 +00:00
|
|
|
range_lookup_int(Manifest, LevelIdx, StartKey, EndKey, MakePointerFun).
|
|
|
|
|
|
|
|
merge_lookup(Manifest, LevelIdx, StartKey, EndKey) ->
|
|
|
|
MakePointerFun =
|
|
|
|
fun(M) ->
|
|
|
|
{next, M, all}
|
|
|
|
end,
|
|
|
|
range_lookup_int(Manifest, LevelIdx, StartKey, EndKey, MakePointerFun).
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
%% An algorithm for discovering which files to merge ....
|
|
|
|
%% We can find the most optimal file:
|
|
|
|
%% - The one with the most overlapping data below?
|
|
|
|
%% - The one that overlaps with the fewest files below?
|
|
|
|
%% - The smallest file?
|
|
|
|
%% We could try and be fair in some way (merge oldest first)
|
|
|
|
%% Ultimately, there is a lack of certainty that being fair or optimal is
|
|
|
|
%% genuinely better - eventually every file has to be compacted.
|
|
|
|
%%
|
|
|
|
%% Hence, the initial implementation is to select files to merge at random
|
2017-01-16 23:27:42 +00:00
|
|
|
mergefile_selector(Manifest, LevelIdx) ->
|
|
|
|
Level = array:get(LevelIdx, Manifest#manifest.levels),
|
2017-01-17 10:12:15 +00:00
|
|
|
lists:nth(random:uniform(length(Level)), Level).
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
add_snapshot(Manifest, Pid, Timeout) ->
|
2017-01-14 20:57:43 +00:00
|
|
|
{MegaNow, SecNow, _} = os:timestamp(),
|
|
|
|
TimeToTimeout = MegaNow * 1000000 + SecNow + Timeout,
|
|
|
|
SnapEntry = {Pid, Manifest#manifest.manifest_sqn, TimeToTimeout},
|
2017-01-13 18:23:57 +00:00
|
|
|
SnapList0 = [SnapEntry|Manifest#manifest.snapshots],
|
2017-01-16 23:27:42 +00:00
|
|
|
ManSQN = Manifest#manifest.manifest_sqn,
|
|
|
|
case Manifest#manifest.min_snapshot_sqn of
|
|
|
|
0 ->
|
|
|
|
|
|
|
|
Manifest#manifest{snapshots = SnapList0,
|
|
|
|
min_snapshot_sqn = ManSQN};
|
|
|
|
N ->
|
|
|
|
N0 = min(N, ManSQN),
|
|
|
|
Manifest#manifest{snapshots = SnapList0, min_snapshot_sqn = N0}
|
|
|
|
end.
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
release_snapshot(Manifest, Pid) ->
|
2017-01-09 14:52:26 +00:00
|
|
|
FilterFun =
|
2017-01-13 18:23:57 +00:00
|
|
|
fun({P, SQN, TS}, {Acc, MinSQN}) ->
|
2017-01-09 14:52:26 +00:00
|
|
|
case P of
|
|
|
|
Pid ->
|
2017-01-17 14:02:21 +00:00
|
|
|
{Acc, MinSQN};
|
2017-01-09 14:52:26 +00:00
|
|
|
_ ->
|
2017-01-13 18:23:57 +00:00
|
|
|
{[{P, SQN, TS}|Acc], min(SQN, MinSQN)}
|
2017-01-09 14:52:26 +00:00
|
|
|
end
|
|
|
|
end,
|
2017-01-16 23:27:42 +00:00
|
|
|
{SnapList0, MinSnapSQN} = lists:foldl(FilterFun,
|
|
|
|
{[], infinity},
|
|
|
|
Manifest#manifest.snapshots),
|
2017-01-13 18:23:57 +00:00
|
|
|
leveled_log:log("P0004", [SnapList0]),
|
2017-01-16 23:27:42 +00:00
|
|
|
case SnapList0 of
|
|
|
|
[] ->
|
|
|
|
Manifest#manifest{snapshots = SnapList0,
|
|
|
|
min_snapshot_sqn = 0};
|
|
|
|
_ ->
|
|
|
|
Manifest#manifest{snapshots = SnapList0,
|
|
|
|
min_snapshot_sqn = MinSnapSQN}
|
|
|
|
end.
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
ready_to_delete(Manifest, Filename) ->
|
2017-01-17 16:54:54 +00:00
|
|
|
{ChangeSQN, _ME} = dict:fetch(Filename, Manifest#manifest.pending_deletes),
|
2017-01-17 14:02:21 +00:00
|
|
|
case Manifest#manifest.min_snapshot_sqn of
|
|
|
|
0 ->
|
|
|
|
% no shapshots
|
|
|
|
PDs = dict:erase(Filename, Manifest#manifest.pending_deletes),
|
|
|
|
{true, Manifest#manifest{pending_deletes = PDs}};
|
|
|
|
N when N >= ChangeSQN ->
|
2017-01-16 23:27:42 +00:00
|
|
|
% Every snapshot is looking at a version of history after this
|
|
|
|
% was removed
|
|
|
|
PDs = dict:erase(Filename, Manifest#manifest.pending_deletes),
|
|
|
|
{true, Manifest#manifest{pending_deletes = PDs}};
|
2017-01-17 14:02:21 +00:00
|
|
|
_N ->
|
2017-01-16 23:27:42 +00:00
|
|
|
{false, Manifest}
|
2017-01-13 18:23:57 +00:00
|
|
|
end.
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-13 18:23:57 +00:00
|
|
|
check_for_work(Manifest, Thresholds) ->
|
|
|
|
CheckLevelFun =
|
2017-01-16 23:27:42 +00:00
|
|
|
fun({LevelIdx, MaxCount}, {AccL, AccC}) ->
|
|
|
|
case LevelIdx > Manifest#manifest.basement of
|
|
|
|
true ->
|
|
|
|
{AccL, AccC};
|
|
|
|
false ->
|
|
|
|
Level = array:get(LevelIdx, Manifest#manifest.levels),
|
|
|
|
S = size(LevelIdx, Level),
|
|
|
|
case S > MaxCount of
|
|
|
|
true ->
|
|
|
|
{[LevelIdx|AccL], AccC + S - MaxCount};
|
|
|
|
false ->
|
|
|
|
{AccL, AccC}
|
|
|
|
end
|
2017-01-13 18:23:57 +00:00
|
|
|
end
|
|
|
|
end,
|
2017-01-16 23:27:42 +00:00
|
|
|
lists:foldr(CheckLevelFun, {[], 0}, Thresholds).
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
is_basement(Manifest, Level) ->
|
2017-01-15 12:41:16 +00:00
|
|
|
Level >= Manifest#manifest.basement.
|
2017-01-13 18:23:57 +00:00
|
|
|
|
|
|
|
levelzero_present(Manifest) ->
|
2017-01-16 23:27:42 +00:00
|
|
|
not is_empty(0, array:get(0, Manifest#manifest.levels)).
|
2017-01-09 14:52:26 +00:00
|
|
|
|
|
|
|
%%%============================================================================
|
|
|
|
%%% Internal Functions
|
|
|
|
%%%============================================================================
|
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
%% All these internal functions that work on a level are also passed LeveIdx
|
|
|
|
%% even if this is not presently relevant. Currnetly levels are lists, but
|
|
|
|
%% future branches may make lower levels trees or skiplists to improve fetch
|
|
|
|
%% efficiency
|
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
load_level(LevelIdx, Level, PidFun, SQNFun) ->
|
|
|
|
HigherLevelLoadFun =
|
2017-01-16 23:27:42 +00:00
|
|
|
fun(ME, {L_Out, L_MaxSQN}) ->
|
|
|
|
FN = ME#manifest_entry.filename,
|
2017-01-17 10:12:15 +00:00
|
|
|
P = PidFun(FN),
|
2017-01-16 23:27:42 +00:00
|
|
|
SQN = SQNFun(P),
|
|
|
|
{[ME#manifest_entry{owner=P}|L_Out], max(SQN, L_MaxSQN)}
|
|
|
|
end,
|
2017-01-22 23:36:16 +00:00
|
|
|
LowerLevelLoadFun =
|
|
|
|
fun({EK, ME}, {L_Out, L_MaxSQN}) ->
|
|
|
|
FN = ME#manifest_entry.filename,
|
|
|
|
P = PidFun(FN),
|
|
|
|
SQN = SQNFun(P),
|
|
|
|
{[{EK, ME#manifest_entry{owner=P}}|L_Out], max(SQN, L_MaxSQN)}
|
|
|
|
end,
|
|
|
|
case LevelIdx =< 1 of
|
|
|
|
true ->
|
|
|
|
lists:foldr(HigherLevelLoadFun, {[], 0}, Level);
|
|
|
|
false ->
|
|
|
|
{L0, MaxSQN} = lists:foldr(LowerLevelLoadFun,
|
|
|
|
{[], 0},
|
|
|
|
leveled_tree:to_list(Level)),
|
2017-01-23 00:22:53 +00:00
|
|
|
{leveled_tree:from_orderedlist(L0, ?TREE_TYPE, ?TREE_WIDTH), MaxSQN}
|
2017-01-22 23:36:16 +00:00
|
|
|
end.
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
close_level(LevelIdx, Level, CloseEntryFun) when LevelIdx =< 1 ->
|
|
|
|
lists:foreach(CloseEntryFun, Level);
|
2017-01-17 10:12:15 +00:00
|
|
|
close_level(_LevelIdx, Level, CloseEntryFun) ->
|
2017-01-22 23:36:16 +00:00
|
|
|
lists:foreach(CloseEntryFun, leveled_tree:to_list(Level)).
|
2017-01-16 23:27:42 +00:00
|
|
|
|
|
|
|
is_empty(_LevelIdx, []) ->
|
|
|
|
true;
|
2017-01-22 23:36:16 +00:00
|
|
|
is_empty(LevelIdx, _Level) when LevelIdx =< 1 ->
|
|
|
|
false;
|
|
|
|
is_empty(_LevelIdx, Level) ->
|
|
|
|
leveled_tree:tsize(Level) == 0.
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
size(LevelIdx, Level) when LevelIdx =< 1 ->
|
|
|
|
length(Level);
|
2017-01-16 23:27:42 +00:00
|
|
|
size(_LevelIdx, Level) ->
|
2017-01-22 23:36:16 +00:00
|
|
|
leveled_tree:tsize(Level).
|
|
|
|
|
|
|
|
pred_fun(LevelIdx, StartKey, _EndKey) when LevelIdx =< 1 ->
|
|
|
|
fun(ME) ->
|
|
|
|
ME#manifest_entry.start_key < StartKey
|
|
|
|
end;
|
|
|
|
pred_fun(_LevelIdx, _StartKey, EndKey) ->
|
|
|
|
fun({EK, _ME}) ->
|
|
|
|
EK < EndKey
|
|
|
|
end.
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-23 00:22:53 +00:00
|
|
|
add_entry(_LevelIdx, Level, []) ->
|
|
|
|
Level;
|
2017-01-22 23:36:16 +00:00
|
|
|
add_entry(LevelIdx, Level, Entries) when is_list(Entries) ->
|
|
|
|
FirstEntry = lists:nth(1, Entries),
|
|
|
|
PredFun = pred_fun(LevelIdx,
|
|
|
|
FirstEntry#manifest_entry.start_key,
|
|
|
|
FirstEntry#manifest_entry.end_key),
|
|
|
|
case LevelIdx =< 1 of
|
|
|
|
true ->
|
|
|
|
{LHS, RHS} = lists:splitwith(PredFun, Level),
|
|
|
|
lists:append([LHS, Entries, RHS]);
|
|
|
|
false ->
|
|
|
|
{LHS, RHS} = lists:splitwith(PredFun, leveled_tree:to_list(Level)),
|
|
|
|
MapFun =
|
|
|
|
fun(ME) ->
|
|
|
|
{ME#manifest_entry.end_key, ME}
|
|
|
|
end,
|
|
|
|
Entries0 = lists:map(MapFun, Entries),
|
|
|
|
leveled_tree:from_orderedlist(lists:append([LHS, Entries0, RHS]),
|
|
|
|
?TREE_TYPE,
|
|
|
|
?TREE_WIDTH)
|
|
|
|
end;
|
|
|
|
add_entry(LevelIdx, Level, Entry) ->
|
|
|
|
add_entry(LevelIdx, Level, [Entry]).
|
|
|
|
|
2017-01-23 11:02:54 +00:00
|
|
|
remove_entry(LevelIdx, Level, Entries) ->
|
2017-01-16 23:27:42 +00:00
|
|
|
% We're assuming we're removing a sorted sublist
|
2017-01-23 11:02:54 +00:00
|
|
|
{RemLength, FirstRemoval} = measure_removals(Entries),
|
|
|
|
remove_section(LevelIdx, Level, FirstRemoval, RemLength).
|
|
|
|
|
|
|
|
measure_removals(Removals) ->
|
|
|
|
case is_list(Removals) of
|
|
|
|
true ->
|
|
|
|
{length(Removals), lists:nth(1, Removals)};
|
|
|
|
false ->
|
|
|
|
{1, Removals}
|
|
|
|
end.
|
2017-01-22 23:36:16 +00:00
|
|
|
|
|
|
|
remove_section(LevelIdx, Level, FirstEntry, SectionLength) ->
|
|
|
|
PredFun = pred_fun(LevelIdx,
|
|
|
|
FirstEntry#manifest_entry.start_key,
|
|
|
|
FirstEntry#manifest_entry.end_key),
|
|
|
|
case LevelIdx =< 1 of
|
|
|
|
true ->
|
|
|
|
{LHS, RHS} = lists:splitwith(PredFun, Level),
|
|
|
|
Post = lists:nthtail(SectionLength, RHS),
|
|
|
|
lists:append([LHS, Post]);
|
|
|
|
false ->
|
|
|
|
{LHS, RHS} = lists:splitwith(PredFun, leveled_tree:to_list(Level)),
|
|
|
|
Post = lists:nthtail(SectionLength, RHS),
|
|
|
|
leveled_tree:from_orderedlist(lists:append([LHS, Post]),
|
|
|
|
?TREE_TYPE,
|
|
|
|
?TREE_WIDTH)
|
|
|
|
end.
|
2017-01-13 18:23:57 +00:00
|
|
|
|
2017-01-23 11:02:54 +00:00
|
|
|
replace_entry(LevelIdx, Level, Removals, Additions) when LevelIdx =< 1 ->
|
|
|
|
{SectionLength, FirstEntry} = measure_removals(Removals),
|
|
|
|
PredFun = pred_fun(LevelIdx,
|
|
|
|
FirstEntry#manifest_entry.start_key,
|
|
|
|
FirstEntry#manifest_entry.end_key),
|
|
|
|
{LHS, RHS} = lists:splitwith(PredFun, Level),
|
|
|
|
Post = lists:nthtail(SectionLength, RHS),
|
|
|
|
case is_list(Additions) of
|
|
|
|
true ->
|
|
|
|
lists:append([LHS, Additions, Post]);
|
|
|
|
false ->
|
|
|
|
lists:append([LHS, [Additions], Post])
|
|
|
|
end;
|
|
|
|
replace_entry(LevelIdx, Level, Removals, Additions) ->
|
|
|
|
{SectionLength, FirstEntry} = measure_removals(Removals),
|
|
|
|
PredFun = pred_fun(LevelIdx,
|
|
|
|
FirstEntry#manifest_entry.start_key,
|
|
|
|
FirstEntry#manifest_entry.end_key),
|
|
|
|
{LHS, RHS} = lists:splitwith(PredFun, leveled_tree:to_list(Level)),
|
|
|
|
Post = lists:nthtail(SectionLength, RHS),
|
|
|
|
UpdList =
|
|
|
|
case is_list(Additions) of
|
|
|
|
true ->
|
|
|
|
MapFun =
|
|
|
|
fun(ME) ->
|
|
|
|
{ME#manifest_entry.end_key, ME}
|
|
|
|
end,
|
|
|
|
Additions0 = lists:map(MapFun, Additions),
|
|
|
|
lists:append([LHS, Additions0, Post]);
|
|
|
|
false ->
|
|
|
|
lists:append([LHS,
|
|
|
|
[{Additions#manifest_entry.end_key,
|
|
|
|
Additions}],
|
|
|
|
Post])
|
|
|
|
end,
|
|
|
|
leveled_tree:from_orderedlist(UpdList, ?TREE_TYPE, ?TREE_WIDTH).
|
|
|
|
|
|
|
|
|
|
|
|
update_pendingdeletes(ManSQN, Removals, PendingDeletes) ->
|
|
|
|
DelFun =
|
|
|
|
fun(E, Acc) ->
|
|
|
|
dict:store(E#manifest_entry.filename,
|
|
|
|
{ManSQN, E},
|
|
|
|
Acc)
|
|
|
|
end,
|
|
|
|
Entries =
|
|
|
|
case is_list(Removals) of
|
|
|
|
true ->
|
|
|
|
Removals;
|
|
|
|
false ->
|
|
|
|
[Removals]
|
|
|
|
end,
|
|
|
|
lists:foldl(DelFun, PendingDeletes, Entries).
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
key_lookup_level(LevelIdx, [], _Key) when LevelIdx =< 1 ->
|
2017-01-16 23:27:42 +00:00
|
|
|
false;
|
2017-01-22 23:36:16 +00:00
|
|
|
key_lookup_level(LevelIdx, [Entry|Rest], Key) when LevelIdx =< 1 ->
|
2017-01-16 23:27:42 +00:00
|
|
|
case Entry#manifest_entry.end_key >= Key of
|
|
|
|
true ->
|
|
|
|
case Key >= Entry#manifest_entry.start_key of
|
2017-01-13 18:23:57 +00:00
|
|
|
true ->
|
2017-01-16 23:27:42 +00:00
|
|
|
Entry#manifest_entry.owner;
|
2017-01-13 18:23:57 +00:00
|
|
|
false ->
|
2017-01-16 23:27:42 +00:00
|
|
|
false
|
2017-01-13 18:23:57 +00:00
|
|
|
end;
|
2017-01-16 23:27:42 +00:00
|
|
|
false ->
|
|
|
|
key_lookup_level(LevelIdx, Rest, Key)
|
2017-01-22 23:36:16 +00:00
|
|
|
end;
|
|
|
|
key_lookup_level(_LevelIdx, Level, Key) ->
|
|
|
|
StartKeyFun =
|
|
|
|
fun(ME) ->
|
|
|
|
ME#manifest_entry.start_key
|
|
|
|
end,
|
|
|
|
case leveled_tree:search(Key, Level, StartKeyFun) of
|
|
|
|
none ->
|
|
|
|
false;
|
|
|
|
{_EK, ME} ->
|
|
|
|
ME#manifest_entry.owner
|
2017-01-13 18:23:57 +00:00
|
|
|
end.
|
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
|
2017-01-17 10:12:15 +00:00
|
|
|
range_lookup_int(Manifest, LevelIdx, StartKey, EndKey, MakePointerFun) ->
|
|
|
|
Range =
|
|
|
|
case LevelIdx > Manifest#manifest.basement of
|
|
|
|
true ->
|
|
|
|
[];
|
|
|
|
false ->
|
|
|
|
range_lookup_level(LevelIdx,
|
|
|
|
array:get(LevelIdx,
|
|
|
|
Manifest#manifest.levels),
|
|
|
|
StartKey,
|
|
|
|
EndKey)
|
|
|
|
end,
|
|
|
|
lists:map(MakePointerFun, Range).
|
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
range_lookup_level(LevelIdx, Level, QStartKey, QEndKey) when LevelIdx =< 1 ->
|
2017-01-16 23:27:42 +00:00
|
|
|
BeforeFun =
|
|
|
|
fun(M) ->
|
|
|
|
QStartKey > M#manifest_entry.end_key
|
|
|
|
end,
|
|
|
|
NotAfterFun =
|
|
|
|
fun(M) ->
|
|
|
|
not leveled_codec:endkey_passed(QEndKey,
|
|
|
|
M#manifest_entry.start_key)
|
|
|
|
end,
|
|
|
|
{_Before, MaybeIn} = lists:splitwith(BeforeFun, Level),
|
|
|
|
{In, _After} = lists:splitwith(NotAfterFun, MaybeIn),
|
2017-01-22 23:36:16 +00:00
|
|
|
In;
|
|
|
|
range_lookup_level(_LevelIdx, Level, QStartKey, QEndKey) ->
|
|
|
|
StartKeyFun =
|
|
|
|
fun(ME) ->
|
|
|
|
ME#manifest_entry.start_key
|
|
|
|
end,
|
|
|
|
Range = leveled_tree:search_range(QStartKey, QEndKey, Level, StartKeyFun),
|
|
|
|
MapFun =
|
|
|
|
fun({_EK, ME}) ->
|
|
|
|
ME
|
|
|
|
end,
|
|
|
|
lists:map(MapFun, Range).
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
|
|
|
|
get_basement(Levels) ->
|
|
|
|
GetBaseFun =
|
|
|
|
fun(L, Acc) ->
|
|
|
|
case is_empty(L, array:get(L, Levels)) of
|
2017-01-09 14:52:26 +00:00
|
|
|
false ->
|
2017-01-16 23:27:42 +00:00
|
|
|
max(L, Acc);
|
2017-01-09 14:52:26 +00:00
|
|
|
true ->
|
2017-01-16 23:27:42 +00:00
|
|
|
Acc
|
2017-01-09 14:52:26 +00:00
|
|
|
end
|
|
|
|
end,
|
2017-01-16 23:27:42 +00:00
|
|
|
lists:foldl(GetBaseFun, 0, lists:seq(0, ?MAX_LEVELS)).
|
|
|
|
|
2017-01-09 14:52:26 +00:00
|
|
|
|
|
|
|
filepath(RootPath, manifest) ->
|
2017-01-14 19:41:09 +00:00
|
|
|
MFP = RootPath ++ "/" ++ ?MANIFEST_FP ++ "/",
|
|
|
|
filelib:ensure_dir(MFP),
|
|
|
|
MFP.
|
2017-01-09 14:52:26 +00:00
|
|
|
|
|
|
|
filepath(RootPath, NewMSN, current_manifest) ->
|
|
|
|
filepath(RootPath, manifest) ++ "nonzero_"
|
|
|
|
++ integer_to_list(NewMSN) ++ "." ++ ?MANIFEST_FILEX.
|
|
|
|
|
|
|
|
|
|
|
|
open_manifestfile(_RootPath, []) ->
|
|
|
|
leveled_log:log("P0013", []),
|
2017-01-17 10:12:15 +00:00
|
|
|
new_manifest();
|
2017-01-09 14:52:26 +00:00
|
|
|
open_manifestfile(_RootPath, [0]) ->
|
|
|
|
leveled_log:log("P0013", []),
|
2017-01-17 10:12:15 +00:00
|
|
|
new_manifest();
|
2017-01-09 14:52:26 +00:00
|
|
|
open_manifestfile(RootPath, [TopManSQN|Rest]) ->
|
|
|
|
CurrManFile = filepath(RootPath, TopManSQN, current_manifest),
|
2017-01-17 10:12:15 +00:00
|
|
|
{ok, FileBin} = file:read_file(CurrManFile),
|
2017-01-16 23:27:42 +00:00
|
|
|
<<CRC:32/integer, BinaryOfTerm/binary>> = FileBin,
|
|
|
|
case erlang:crc32(BinaryOfTerm) of
|
|
|
|
CRC ->
|
2017-01-09 14:52:26 +00:00
|
|
|
leveled_log:log("P0012", [TopManSQN]),
|
2017-01-16 23:27:42 +00:00
|
|
|
binary_to_term(BinaryOfTerm);
|
|
|
|
_ ->
|
|
|
|
leveled_log:log("P0033", [CurrManFile, "crc wonky"]),
|
|
|
|
open_manifestfile(RootPath, Rest)
|
2017-01-09 14:52:26 +00:00
|
|
|
end.
|
|
|
|
|
2017-01-22 23:36:16 +00:00
|
|
|
|
2017-01-09 14:52:26 +00:00
|
|
|
%%%============================================================================
|
|
|
|
%%% Test
|
|
|
|
%%%============================================================================
|
|
|
|
|
|
|
|
-ifdef(TEST).
|
|
|
|
|
2017-01-14 16:36:05 +00:00
|
|
|
initial_setup() ->
|
2017-01-09 14:52:26 +00:00
|
|
|
E1 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld1"}, "K8"},
|
|
|
|
end_key={i, "Bucket1", {"Idx1", "Fld9"}, "K93"},
|
2017-01-14 16:36:05 +00:00
|
|
|
filename="Z1",
|
|
|
|
owner="pid_z1"},
|
2017-01-09 14:52:26 +00:00
|
|
|
E2 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld9"}, "K97"},
|
|
|
|
end_key={o, "Bucket1", "K71", null},
|
2017-01-14 16:36:05 +00:00
|
|
|
filename="Z2",
|
|
|
|
owner="pid_z2"},
|
2017-01-09 14:52:26 +00:00
|
|
|
E3 = #manifest_entry{start_key={o, "Bucket1", "K75", null},
|
|
|
|
end_key={o, "Bucket1", "K993", null},
|
2017-01-14 16:36:05 +00:00
|
|
|
filename="Z3",
|
|
|
|
owner="pid_z3"},
|
2017-01-09 14:52:26 +00:00
|
|
|
E4 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld1"}, "K8"},
|
|
|
|
end_key={i, "Bucket1", {"Idx1", "Fld7"}, "K93"},
|
2017-01-14 16:36:05 +00:00
|
|
|
filename="Z4",
|
|
|
|
owner="pid_z4"},
|
2017-01-09 14:52:26 +00:00
|
|
|
E5 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld7"}, "K97"},
|
|
|
|
end_key={o, "Bucket1", "K78", null},
|
2017-01-14 16:36:05 +00:00
|
|
|
filename="Z5",
|
|
|
|
owner="pid_z5"},
|
2017-01-09 14:52:26 +00:00
|
|
|
E6 = #manifest_entry{start_key={o, "Bucket1", "K81", null},
|
|
|
|
end_key={o, "Bucket1", "K996", null},
|
2017-01-14 16:36:05 +00:00
|
|
|
filename="Z6",
|
|
|
|
owner="pid_z6"},
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-14 16:36:05 +00:00
|
|
|
Man0 = new_manifest(),
|
|
|
|
% insert_manifest_entry(Manifest, ManSQN, Level, Entry)
|
|
|
|
Man1 = insert_manifest_entry(Man0, 1, 1, E1),
|
|
|
|
Man2 = insert_manifest_entry(Man1, 1, 1, E2),
|
|
|
|
Man3 = insert_manifest_entry(Man2, 1, 1, E3),
|
|
|
|
Man4 = insert_manifest_entry(Man3, 1, 2, E4),
|
|
|
|
Man5 = insert_manifest_entry(Man4, 1, 2, E5),
|
|
|
|
Man6 = insert_manifest_entry(Man5, 1, 2, E6),
|
|
|
|
{Man0, Man1, Man2, Man3, Man4, Man5, Man6}.
|
|
|
|
|
|
|
|
changeup_setup(Man6) ->
|
|
|
|
E1 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld1"}, "K8"},
|
|
|
|
end_key={i, "Bucket1", {"Idx1", "Fld9"}, "K93"},
|
|
|
|
filename="Z1",
|
|
|
|
owner="pid_z1"},
|
|
|
|
E2 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld9"}, "K97"},
|
|
|
|
end_key={o, "Bucket1", "K71", null},
|
|
|
|
filename="Z2",
|
|
|
|
owner="pid_z2"},
|
|
|
|
E3 = #manifest_entry{start_key={o, "Bucket1", "K75", null},
|
|
|
|
end_key={o, "Bucket1", "K993", null},
|
|
|
|
filename="Z3",
|
|
|
|
owner="pid_z3"},
|
|
|
|
|
2017-01-09 14:52:26 +00:00
|
|
|
E1_2 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld4"}, "K8"},
|
|
|
|
end_key={i, "Bucket1", {"Idx1", "Fld9"}, "K62"},
|
2017-01-14 16:36:05 +00:00
|
|
|
owner="pid_y1",
|
2017-01-09 14:52:26 +00:00
|
|
|
filename="Y1"},
|
|
|
|
E2_2 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld9"}, "K67"},
|
2017-01-14 16:36:05 +00:00
|
|
|
end_key={o, "Bucket1", "K45", null},
|
|
|
|
owner="pid_y2",
|
|
|
|
filename="Y2"},
|
2017-01-09 14:52:26 +00:00
|
|
|
E3_2 = #manifest_entry{start_key={o, "Bucket1", "K47", null},
|
|
|
|
end_key={o, "Bucket1", "K812", null},
|
2017-01-14 16:36:05 +00:00
|
|
|
owner="pid_y3",
|
2017-01-09 14:52:26 +00:00
|
|
|
filename="Y3"},
|
|
|
|
E4_2 = #manifest_entry{start_key={o, "Bucket1", "K815", null},
|
|
|
|
end_key={o, "Bucket1", "K998", null},
|
2017-01-14 16:36:05 +00:00
|
|
|
owner="pid_y4",
|
2017-01-09 14:52:26 +00:00
|
|
|
filename="Y4"},
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
Man7 = remove_manifest_entry(Man6, 2, 1, E1),
|
|
|
|
Man8 = remove_manifest_entry(Man7, 2, 1, E2),
|
|
|
|
Man9 = remove_manifest_entry(Man8, 2, 1, E3),
|
|
|
|
|
|
|
|
Man10 = insert_manifest_entry(Man9, 2, 1, E1_2),
|
|
|
|
Man11 = insert_manifest_entry(Man10, 2, 1, E2_2),
|
|
|
|
Man12 = insert_manifest_entry(Man11, 2, 1, E3_2),
|
|
|
|
Man13 = insert_manifest_entry(Man12, 2, 1, E4_2),
|
2017-01-14 16:36:05 +00:00
|
|
|
% remove_manifest_entry(Manifest, ManSQN, Level, Entry)
|
2017-01-16 23:27:42 +00:00
|
|
|
|
2017-01-14 16:36:05 +00:00
|
|
|
{Man7, Man8, Man9, Man10, Man11, Man12, Man13}.
|
|
|
|
|
|
|
|
keylookup_manifest_test() ->
|
|
|
|
{Man0, Man1, Man2, Man3, _Man4, _Man5, Man6} = initial_setup(),
|
|
|
|
LK1_1 = {o, "Bucket1", "K711", null},
|
|
|
|
LK1_2 = {o, "Bucket1", "K70", null},
|
|
|
|
LK1_3 = {o, "Bucket1", "K71", null},
|
|
|
|
LK1_4 = {o, "Bucket1", "K75", null},
|
|
|
|
LK1_5 = {o, "Bucket1", "K76", null},
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-14 16:36:05 +00:00
|
|
|
?assertMatch(false, key_lookup(Man0, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man1, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man2, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man3, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man6, 1, LK1_1)),
|
|
|
|
|
|
|
|
?assertMatch("pid_z2", key_lookup(Man6, 1, LK1_2)),
|
|
|
|
?assertMatch("pid_z2", key_lookup(Man6, 1, LK1_3)),
|
|
|
|
?assertMatch("pid_z3", key_lookup(Man6, 1, LK1_4)),
|
|
|
|
?assertMatch("pid_z3", key_lookup(Man6, 1, LK1_5)),
|
|
|
|
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_2)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_3)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_4)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_5)),
|
|
|
|
|
|
|
|
{_Man7, _Man8, _Man9, _Man10, _Man11, _Man12,
|
|
|
|
Man13} = changeup_setup(Man6),
|
|
|
|
|
|
|
|
?assertMatch(false, key_lookup(Man0, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man1, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man2, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man3, 1, LK1_1)),
|
|
|
|
?assertMatch(false, key_lookup(Man6, 1, LK1_1)),
|
|
|
|
|
|
|
|
?assertMatch("pid_z2", key_lookup(Man6, 1, LK1_2)),
|
|
|
|
?assertMatch("pid_z2", key_lookup(Man6, 1, LK1_3)),
|
|
|
|
?assertMatch("pid_z3", key_lookup(Man6, 1, LK1_4)),
|
|
|
|
?assertMatch("pid_z3", key_lookup(Man6, 1, LK1_5)),
|
|
|
|
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_2)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_3)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_4)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man6, 2, LK1_5)),
|
|
|
|
|
|
|
|
?assertMatch("pid_y3", key_lookup(Man13, 1, LK1_4)),
|
|
|
|
?assertMatch("pid_z5", key_lookup(Man13, 2, LK1_4)).
|
|
|
|
|
|
|
|
|
|
|
|
rangequery_manifest_test() ->
|
|
|
|
{_Man0, _Man1, _Man2, _Man3, _Man4, _Man5, Man6} = initial_setup(),
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
PidMapFun =
|
|
|
|
fun(Pointer) ->
|
|
|
|
{next, ME, _SK} = Pointer,
|
|
|
|
ME#manifest_entry.owner
|
|
|
|
end,
|
|
|
|
|
2017-01-14 16:36:05 +00:00
|
|
|
SK1 = {o, "Bucket1", "K711", null},
|
|
|
|
EK1 = {o, "Bucket1", "K999", null},
|
2017-01-16 23:27:42 +00:00
|
|
|
RL1_1 = lists:map(PidMapFun, range_lookup(Man6, 1, SK1, EK1)),
|
|
|
|
?assertMatch(["pid_z3"], RL1_1),
|
|
|
|
RL1_2 = lists:map(PidMapFun, range_lookup(Man6, 2, SK1, EK1)),
|
|
|
|
?assertMatch(["pid_z5", "pid_z6"], RL1_2),
|
2017-01-14 16:36:05 +00:00
|
|
|
SK2 = {i, "Bucket1", {"Idx1", "Fld8"}, null},
|
|
|
|
EK2 = {i, "Bucket1", {"Idx1", "Fld8"}, null},
|
2017-01-16 23:27:42 +00:00
|
|
|
RL2_1 = lists:map(PidMapFun, range_lookup(Man6, 1, SK2, EK2)),
|
|
|
|
?assertMatch(["pid_z1"], RL2_1),
|
|
|
|
RL2_2 = lists:map(PidMapFun, range_lookup(Man6, 2, SK2, EK2)),
|
|
|
|
?assertMatch(["pid_z5"], RL2_2),
|
2017-01-14 16:36:05 +00:00
|
|
|
|
|
|
|
SK3 = {o, "Bucket1", "K994", null},
|
|
|
|
EK3 = {o, "Bucket1", "K995", null},
|
2017-01-16 23:27:42 +00:00
|
|
|
RL3_1 = lists:map(PidMapFun, range_lookup(Man6, 1, SK3, EK3)),
|
2017-01-14 16:36:05 +00:00
|
|
|
?assertMatch([], RL3_1),
|
2017-01-16 23:27:42 +00:00
|
|
|
RL3_2 = lists:map(PidMapFun, range_lookup(Man6, 2, SK3, EK3)),
|
|
|
|
?assertMatch(["pid_z6"], RL3_2),
|
2017-01-14 16:36:05 +00:00
|
|
|
|
|
|
|
{_Man7, _Man8, _Man9, _Man10, _Man11, _Man12,
|
|
|
|
Man13} = changeup_setup(Man6),
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
RL1_1A = lists:map(PidMapFun, range_lookup(Man6, 1, SK1, EK1)),
|
|
|
|
?assertMatch(["pid_z3"], RL1_1A),
|
|
|
|
RL2_1A = lists:map(PidMapFun, range_lookup(Man6, 1, SK2, EK2)),
|
|
|
|
?assertMatch(["pid_z1"], RL2_1A),
|
|
|
|
RL3_1A = lists:map(PidMapFun, range_lookup(Man6, 1, SK3, EK3)),
|
2017-01-14 16:36:05 +00:00
|
|
|
?assertMatch([], RL3_1A),
|
|
|
|
|
2017-01-16 23:27:42 +00:00
|
|
|
RL1_1B = lists:map(PidMapFun, range_lookup(Man13, 1, SK1, EK1)),
|
|
|
|
?assertMatch(["pid_y3", "pid_y4"], RL1_1B),
|
|
|
|
RL2_1B = lists:map(PidMapFun, range_lookup(Man13, 1, SK2, EK2)),
|
|
|
|
?assertMatch(["pid_y1"], RL2_1B),
|
|
|
|
RL3_1B = lists:map(PidMapFun, range_lookup(Man13, 1, SK3, EK3)),
|
|
|
|
?assertMatch(["pid_y4"], RL3_1B).
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-14 19:41:09 +00:00
|
|
|
levelzero_present_test() ->
|
|
|
|
E0 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld1"}, "K8"},
|
|
|
|
end_key={o, "Bucket1", "Key996", null},
|
|
|
|
filename="Z0",
|
|
|
|
owner="pid_z0"},
|
|
|
|
|
|
|
|
Man0 = new_manifest(),
|
|
|
|
?assertMatch(false, levelzero_present(Man0)),
|
|
|
|
% insert_manifest_entry(Manifest, ManSQN, Level, Entry)
|
|
|
|
Man1 = insert_manifest_entry(Man0, 1, 0, E0),
|
|
|
|
?assertMatch(true, levelzero_present(Man1)).
|
2017-01-09 14:52:26 +00:00
|
|
|
|
2017-01-17 14:02:21 +00:00
|
|
|
snapshot_release_test() ->
|
|
|
|
Man6 = element(7, initial_setup()),
|
|
|
|
E1 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld1"}, "K8"},
|
|
|
|
end_key={i, "Bucket1", {"Idx1", "Fld9"}, "K93"},
|
|
|
|
filename="Z1",
|
|
|
|
owner="pid_z1"},
|
|
|
|
E2 = #manifest_entry{start_key={i, "Bucket1", {"Idx1", "Fld9"}, "K97"},
|
|
|
|
end_key={o, "Bucket1", "K71", null},
|
|
|
|
filename="Z2",
|
|
|
|
owner="pid_z2"},
|
|
|
|
E3 = #manifest_entry{start_key={o, "Bucket1", "K75", null},
|
|
|
|
end_key={o, "Bucket1", "K993", null},
|
|
|
|
filename="Z3",
|
|
|
|
owner="pid_z3"},
|
|
|
|
|
|
|
|
Man7 = add_snapshot(Man6, "pid_a1", 3600),
|
|
|
|
Man8 = remove_manifest_entry(Man7, 2, 1, E1),
|
|
|
|
Man9 = add_snapshot(Man8, "pid_a2", 3600),
|
|
|
|
Man10 = remove_manifest_entry(Man9, 3, 1, E2),
|
|
|
|
Man11 = add_snapshot(Man10, "pid_a3", 3600),
|
|
|
|
Man12 = remove_manifest_entry(Man11, 4, 1, E3),
|
|
|
|
Man13 = add_snapshot(Man12, "pid_a4", 3600),
|
|
|
|
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man8, "Z1"))),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man10, "Z2"))),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man12, "Z3"))),
|
|
|
|
|
|
|
|
Man14 = release_snapshot(Man13, "pid_a1"),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man14, "Z2"))),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man14, "Z3"))),
|
|
|
|
{Bool14, Man15} = ready_to_delete(Man14, "Z1"),
|
|
|
|
?assertMatch(true, Bool14),
|
|
|
|
|
|
|
|
%This doesn't change anything - released snaphsot not the min
|
|
|
|
Man16 = release_snapshot(Man15, "pid_a4"),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man16, "Z2"))),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man16, "Z3"))),
|
|
|
|
|
|
|
|
Man17 = release_snapshot(Man16, "pid_a2"),
|
|
|
|
?assertMatch(false, element(1, ready_to_delete(Man17, "Z3"))),
|
|
|
|
{Bool17, Man18} = ready_to_delete(Man17, "Z2"),
|
|
|
|
?assertMatch(true, Bool17),
|
|
|
|
|
|
|
|
Man19 = release_snapshot(Man18, "pid_a3"),
|
|
|
|
|
|
|
|
io:format("MinSnapSQN ~w~n", [Man19#manifest.min_snapshot_sqn]),
|
|
|
|
|
|
|
|
{Bool19, _Man20} = ready_to_delete(Man19, "Z3"),
|
|
|
|
?assertMatch(true, Bool19).
|
|
|
|
|
|
|
|
|
|
|
|
|
2017-01-09 14:52:26 +00:00
|
|
|
-endif.
|