feat(utils): add `emqx_utils_ets:keyfoldl/3` function

Designed to be used with `bag` / `duplicate_bag` tables.
This commit is contained in:
Andrew Mayorov 2024-05-10 15:19:58 +02:00 committed by Serge Tupchii
parent 4097585f5d
commit 5bd9ee5c7f
1 changed files with 20 additions and 0 deletions

View File

@ -26,6 +26,8 @@
lookup_value/3
]).
-export([keyfoldl/3]).
-export([delete/1]).
%% Create an ets table.
@ -57,6 +59,24 @@ lookup_value(Tab, Key, Def) ->
error:badarg -> Def
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.
-spec delete(ets:tab()) -> ok.
delete(Tab) ->