chore(emqx_mgmt_api_banned): do_pack_banned takes map instead of impropr record

This commit is contained in:
z8674558 2021-03-01 12:53:20 +09:00 committed by tigercl
parent 85b43af2ca
commit 4b43b6532c
1 changed files with 11 additions and 15 deletions

View File

@ -105,37 +105,33 @@ validate_params(Params) ->
{error, ?ERROR8, Msg} {error, ?ERROR8, Msg}
end. end.
%% TODO who and reason is undefined - causing dialyzer errors. fix later
-dialyzer({nowarn_function,pack_banned/1}).
pack_banned(Params) -> pack_banned(Params) ->
Now = erlang:system_time(second), Now = erlang:system_time(second),
do_pack_banned(Params, #banned{by = <<"user">>, do_pack_banned(Params, #{by => <<"user">>, at => Now, until => Now + 300}).
at = Now,
until = Now + 300}).
do_pack_banned([], Banned) -> do_pack_banned([], #{who := Who, by := By, reason := Reason, at := At, until := Until}) ->
{ok, Banned}; {ok, #banned{who = Who, by = By, reason = Reason, at = At, until = Until}};
do_pack_banned([{<<"who">>, Who} | Params], Banned) -> do_pack_banned([{<<"who">>, Who} | Params], Banned) ->
case lists:keytake(<<"as">>, 1, Params) of case lists:keytake(<<"as">>, 1, Params) of
{value, {<<"as">>, <<"peerhost">>}, Params2} -> {value, {<<"as">>, <<"peerhost">>}, Params2} ->
{ok, IPAddress} = inet:parse_address(str(Who)), {ok, IPAddress} = inet:parse_address(str(Who)),
do_pack_banned(Params2, Banned#banned{who = {peerhost, IPAddress}}); do_pack_banned(Params2, Banned#{who => {peerhost, IPAddress}});
{value, {<<"as">>, <<"clientid">>}, Params2} -> {value, {<<"as">>, <<"clientid">>}, Params2} ->
do_pack_banned(Params2, Banned#banned{who = {clientid, Who}}); do_pack_banned(Params2, Banned#{who => {clientid, Who}});
{value, {<<"as">>, <<"username">>}, Params2} -> {value, {<<"as">>, <<"username">>}, Params2} ->
do_pack_banned(Params2, Banned#banned{who = {username, Who}}) do_pack_banned(Params2, Banned#{who => {username, Who}})
end; end;
do_pack_banned([P1 = {<<"as">>, _}, P2 | Params], Banned) -> do_pack_banned([P1 = {<<"as">>, _}, P2 | Params], Banned) ->
do_pack_banned([P2, P1 | Params], Banned); do_pack_banned([P2, P1 | Params], Banned);
do_pack_banned([{<<"by">>, By} | Params], Banned) -> do_pack_banned([{<<"by">>, By} | Params], Banned) ->
do_pack_banned(Params, Banned#banned{by = By}); do_pack_banned(Params, Banned#{by => By});
do_pack_banned([{<<"reason">>, Reason} | Params], Banned) -> do_pack_banned([{<<"reason">>, Reason} | Params], Banned) ->
do_pack_banned(Params, Banned#banned{reason = Reason}); do_pack_banned(Params, Banned#{reason => Reason});
do_pack_banned([{<<"at">>, At} | Params], Banned) -> do_pack_banned([{<<"at">>, At} | Params], Banned) ->
do_pack_banned(Params, Banned#banned{at = At}); do_pack_banned(Params, Banned#{at => At});
do_pack_banned([{<<"until">>, Until} | Params], Banned) -> do_pack_banned([{<<"until">>, Until} | Params], Banned) ->
do_pack_banned(Params, Banned#banned{until = Until}); do_pack_banned(Params, Banned#{until => Until});
do_pack_banned([_P | Params], Banned) -> %% ingore other params do_pack_banned([_P | Params], Banned) -> %% ignore other params
do_pack_banned(Params, Banned). do_pack_banned(Params, Banned).
do_delete(<<"peerhost">>, Who) -> do_delete(<<"peerhost">>, Who) ->