Ayodele 5 лет назад
Родитель
Сommit
2b481dabe7
5 измененных файлов с 8 добавлено и 5 удалено
  1. 4 2
      apps/emqx_management/src/emqx_mgmt_api_banned.erl
  2. 1 0
      include/emqx.hrl
  3. 1 1
      src/emqx.erl
  4. 1 1
      src/emqx_hooks.erl
  5. 1 1
      src/emqx_types.erl

+ 4 - 2
apps/emqx_management/src/emqx_mgmt_api_banned.erl

@@ -59,7 +59,7 @@ create(_Bindings, Params) ->
             {ok, Banned} = pack_banned(NParams),
             ok = emqx_mgmt:create_banned(Banned),
             return({ok, maps:from_list(Params)});
-        {error, Code, Message} -> 
+        {error, Code, Message} ->
             return({error, Code, Message})
     end.
 
@@ -71,7 +71,7 @@ delete(#{as := As, who := Who}, _) ->
         {ok, NParams} ->
             do_delete(get_value(<<"as">>, NParams), get_value(<<"who">>, NParams)),
             return();
-        {error, Code, Message} -> 
+        {error, Code, Message} ->
             return({error, Code, Message})
     end.
 
@@ -105,6 +105,8 @@ validate_params(Params) ->
             {error, ?ERROR8, Msg}
     end.
 
+%% TODO who and reason is undefined - causing dialyzer errors. fix later
+-dialyzer({nowarn_function,pack_banned/1}).
 pack_banned(Params) ->
     Now = erlang:system_time(second),
     do_pack_banned(Params, #banned{by = <<"user">>,

+ 1 - 0
include/emqx.hrl

@@ -145,6 +145,7 @@
 
 -record(banned, {
           who    :: {clientid,  binary()}
+                  | {peerhost, inet:ip_address()}
                   | {username,   binary()}
                   | {ip_address, inet:ip_address()},
           by     :: binary(),

+ 1 - 1
src/emqx.erl

@@ -184,7 +184,7 @@ hook(HookPoint, Action, InitArgs) when is_list(InitArgs) ->
 hook(HookPoint, Action, Filter, Priority) ->
     emqx_hooks:add(HookPoint, Action, Filter, Priority).
 
--spec(unhook(emqx_hooks:hookpoint(), fun() | {atom(), atom()} | {atom(), atom(), [any()]}) -> ok).
+-spec(unhook(emqx_hooks:hookpoint(), emqx_hooks:action() | {module(), atom()}) -> ok).
 unhook(HookPoint, Action) ->
     emqx_hooks:del(HookPoint, Action).
 

+ 1 - 1
src/emqx_hooks.erl

@@ -121,7 +121,7 @@ put(HookPoint, Callback) ->
     end.
 
 %% @doc Unregister a callback.
--spec(del(hookpoint(), function() | {module(), atom()}) -> ok).
+-spec(del(hookpoint(), action() | {module(), atom()}) -> ok).
 del(HookPoint, Action) ->
     gen_server:cast(?SERVER, {del, HookPoint, Action}).
 

+ 1 - 1
src/emqx_types.erl

@@ -127,7 +127,7 @@
                       expiry_interval => non_neg_integer(),
                       atom() => term()
                      }).
--type(clientinfo() :: #{zone         := zone(),
+-type(clientinfo() :: #{zone         := maybe(zone()),
                         protocol     := protocol(),
                         peerhost     := peerhost(),
                         sockport     := non_neg_integer(),