yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(mgmt_api): `node_query`.

This commit is contained in:
JimMoen 2021-10-08 14:58:18 +08:00
parent e26804b9ac
commit 77516d3854
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
apps/emqx_management/src/emqx_mgmt_api.erl
View File

@ -120,16 +120,27 @@ node_query(Node, Params, Tab, QsSchema, QueryFun) ->
{CodCnt, Qs} = params2qs(Params, QsSchema), {CodCnt, Qs} = params2qs(Params, QsSchema),
Limit = b2i(limit(Params)), Limit = b2i(limit(Params)),
Page = b2i(page(Params)), Page = b2i(page(Params)),
Start = if Page > 1 -> (Page-1) * Limit; PageStart = page_start(Page, Limit),
true -> 0 %% Rows so big, fixme.
end, Rows = do_node_query(Node, Tab, Qs, QueryFun, _Continuation = ?FRESH_SELECT, Limit, []),
{_, Rows} = do_query(Node, Tab, Qs, QueryFun, Start, Limit+1), Data = lists:sublist(Rows, PageStart, Limit),
Meta = #{page => Page, limit => Limit}, Meta = #{page => Page, limit => Limit},
NMeta = case CodCnt =:= 0 of NMeta = case CodCnt =:= 0 of
true -> Meta#{count => count(Tab)}; true -> Meta#{count => count(Tab)};
_ -> Meta#{count => length(Rows)} _ -> Meta#{count => length(Rows)}
end, end,
#{meta => NMeta, data => lists:sublist(Rows, Limit)}. #{meta => NMeta, data => Data}.
%% @private
do_node_query(Node, Tab, Qs, QueryFun, Continuation, Limit, Acc) ->
{Rows, NContinuation} = do_query(Node, Tab, Qs, QueryFun, Continuation, Limit),
NAcc = [Rows | Acc],
case NContinuation of
?FRESH_SELECT ->
lists:append(lists:reverse(Acc));
_ ->
do_node_query(Node, Tab, Qs, QueryFun, NContinuation, Limit, NAcc)
end.
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% Cluster Query %% Cluster Query