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

fix(exhook): set trap_exit flag

This commit is contained in:
JianBo He 2021-08-11 18:21:16 +08:00
parent b3db4d0f7c
commit 1e4ca14476
3 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/emqx_exhook/src/emqx_exhook_mngr.erl
View File

@ -101,6 +101,7 @@ call(Pid, Req) ->
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
init([Servers, AutoReconnect, ReqOpts]) -> init([Servers, AutoReconnect, ReqOpts]) ->
process_flag(trap_exit, true),
%% XXX: Due to the ExHook Module in the enterprise, %% XXX: Due to the ExHook Module in the enterprise,
%% this process may start multiple times and they will share this table %% this process may start multiple times and they will share this table
try try
@ -182,11 +183,11 @@ handle_info({timeout, _Ref, {reload, Name}}, State) ->
handle_info(_Info, State) -> handle_info(_Info, State) ->
{noreply, State}. {noreply, State}.
terminate(_Reason, State = #state{stopped = Stopped}) -> terminate(_Reason, State = #state{running = Running}) ->
_ = maps:fold(fun(Name, _, AccIn) -> _ = maps:fold(fun(Name, _, AccIn) ->
{ok, NAccIn} = do_unload_server(Name, AccIn), {ok, NAccIn} = do_unload_server(Name, AccIn),
NAccIn NAccIn
end, State, Stopped), end, State, Running),
_ = unload_exhooks(), _ = unload_exhooks(),
ok. ok.

1
apps/emqx_exhook/src/emqx_exhook_sup.erl
View File

@ -30,6 +30,7 @@
#{ id => Mod #{ id => Mod
, start => {Mod, start_link, Args} , start => {Mod, start_link, Args}
, type => Type , type => Type
, shutdown => 15000
} }
). ).

11
apps/emqx_exhook/test/emqx_exhook_SUITE.erl
View File

@ -52,17 +52,12 @@ set_special_cfgs(emqx_exhook) ->
t_noserver_nohook(_) -> t_noserver_nohook(_) ->
emqx_exhook:disable(default), emqx_exhook:disable(default),
?assertEqual([], ets:tab2list(emqx_hooks)), ?assertEqual([], ets:tab2list(emqx_hooks)),
ok = emqx_exhook:enable(default),
Opts = proplists:get_value(
default,
application:get_env(emqx_exhook, servers, [])
),
ok = emqx_exhook:enable(default, Opts),
?assertNotEqual([], ets:tab2list(emqx_hooks)). ?assertNotEqual([], ets:tab2list(emqx_hooks)).
t_cli_list(_) -> t_cli_list(_) ->
meck_print(), meck_print(),
?assertEqual( [[emqx_exhook_server:format(Svr) || Svr <- emqx_exhook:list()]] ?assertEqual( [[emqx_exhook_server:format(emqx_exhook_mngr:server(Name)) || Name <- emqx_exhook:list()]]
, emqx_exhook_cli:cli(["server", "list"]) , emqx_exhook_cli:cli(["server", "list"])
), ),
unmeck_print(). unmeck_print().
@ -71,7 +66,7 @@ t_cli_enable_disable(_) ->
meck_print(), meck_print(),
?assertEqual([already_started], emqx_exhook_cli:cli(["server", "enable", "default"])), ?assertEqual([already_started], emqx_exhook_cli:cli(["server", "enable", "default"])),
?assertEqual(ok, emqx_exhook_cli:cli(["server", "disable", "default"])), ?assertEqual(ok, emqx_exhook_cli:cli(["server", "disable", "default"])),
?assertEqual([], emqx_exhook_cli:cli(["server", "list"])), ?assertEqual([["name=default, hooks=#{}, active=false"]], emqx_exhook_cli:cli(["server", "list"])),
?assertEqual([not_running], emqx_exhook_cli:cli(["server", "disable", "default"])), ?assertEqual([not_running], emqx_exhook_cli:cli(["server", "disable", "default"])),
?assertEqual(ok, emqx_exhook_cli:cli(["server", "enable", "default"])), ?assertEqual(ok, emqx_exhook_cli:cli(["server", "enable", "default"])),