feat(utils): add `emqx_utils_ets:keyfoldl/3` function
Designed to be used with `bag` / `duplicate_bag` tables.
This commit is contained in:
parent
4097585f5d
commit
5bd9ee5c7f
|
@ -26,6 +26,8 @@
|
||||||
lookup_value/3
|
lookup_value/3
|
||||||
]).
|
]).
|
||||||
|
|
||||||
|
-export([keyfoldl/3]).
|
||||||
|
|
||||||
-export([delete/1]).
|
-export([delete/1]).
|
||||||
|
|
||||||
%% Create an ets table.
|
%% Create an ets table.
|
||||||
|
@ -57,6 +59,24 @@ lookup_value(Tab, Key, Def) ->
|
||||||
error:badarg -> Def
|
error:badarg -> Def
|
||||||
end.
|
end.
|
||||||
|
|
||||||
|
-spec keyfoldl(fun((_Key :: term(), Acc) -> Acc), Acc, ets:tab()) -> Acc.
|
||||||
|
keyfoldl(F, Acc, Tab) ->
|
||||||
|
true = ets:safe_fixtable(Tab, true),
|
||||||
|
First = ets:first(Tab),
|
||||||
|
try
|
||||||
|
keyfoldl(F, Acc, First, Tab)
|
||||||
|
after
|
||||||
|
ets:safe_fixtable(Tab, false)
|
||||||
|
end.
|
||||||
|
|
||||||
|
keyfoldl(F, Acc, Key, Tab) ->
|
||||||
|
case Key of
|
||||||
|
'$end_of_table' ->
|
||||||
|
Acc;
|
||||||
|
_ ->
|
||||||
|
keyfoldl(F, F(Key, Acc), ets:next(Tab, Key), Tab)
|
||||||
|
end.
|
||||||
|
|
||||||
%% Delete the ets table.
|
%% Delete the ets table.
|
||||||
-spec delete(ets:tab()) -> ok.
|
-spec delete(ets:tab()) -> ok.
|
||||||
delete(Tab) ->
|
delete(Tab) ->
|
||||||
|
|
Loading…
Reference in New Issue