From c52b464b3c7cd36e9d3e51fdf1db992e4f6a86a5 Mon Sep 17 00:00:00 2001 From: EMQ-YangM Date: Fri, 29 Apr 2022 17:07:42 +0800 Subject: [PATCH] fix: check process alive before health check --- apps/emqx_plugin_libs/src/emqx_plugin_libs_pool.erl | 6 ++++-- apps/emqx_resource/src/emqx_resource_health_check.erl | 11 ++++++++++- apps/emqx_resource/src/emqx_resource_instance.erl | 7 ------- 3 files changed, 14 insertions(+), 10 deletions(-) diff --git a/apps/emqx_plugin_libs/src/emqx_plugin_libs_pool.erl b/apps/emqx_plugin_libs/src/emqx_plugin_libs_pool.erl index e6a7ad366..4358999a5 100644 --- a/apps/emqx_plugin_libs/src/emqx_plugin_libs_pool.erl +++ b/apps/emqx_plugin_libs/src/emqx_plugin_libs_pool.erl @@ -68,8 +68,10 @@ get_status(PoolName, CheckFunc, AutoReconn) when is_function(CheckFunc) -> Status = [ begin case ecpool_worker:client(Worker) of - {ok, Conn} -> CheckFunc(Conn); - _ -> false + {ok, Conn} -> + erlang:is_process_alive(Conn) andalso CheckFunc(Conn); + _ -> + false end end || {_WorkerName, Worker} <- ecpool:workers(PoolName) diff --git a/apps/emqx_resource/src/emqx_resource_health_check.erl b/apps/emqx_resource/src/emqx_resource_health_check.erl index ac03c3c35..265592582 100644 --- a/apps/emqx_resource/src/emqx_resource_health_check.erl +++ b/apps/emqx_resource/src/emqx_resource_health_check.erl @@ -15,6 +15,8 @@ %%-------------------------------------------------------------------- -module(emqx_resource_health_check). +-include_lib("emqx/include/logger.hrl"). + -export([ start_link/3, create_checker/3, @@ -98,7 +100,14 @@ health_check_timeout_checker(Pid, Name, SleepTime, Timeout) -> #{name => Name}, <> ), - _ = emqx_resouce_instance:set_resource_status(Name, disconnected), + ?SLOG( + error, + #{ + msg => "health check failed: timeout", + name => Name + } + ), + _ = emqx_resource_instance:set_resource_status(Name, disconnected), receive health_check_finish -> timer:sleep(SleepTime) end diff --git a/apps/emqx_resource/src/emqx_resource_instance.erl b/apps/emqx_resource/src/emqx_resource_instance.erl index 67ff4637a..eb72fff1b 100644 --- a/apps/emqx_resource/src/emqx_resource_instance.erl +++ b/apps/emqx_resource/src/emqx_resource_instance.erl @@ -345,13 +345,6 @@ do_set_resource_status_connecting(InstId) -> set_resource_status(InstId, Status) -> case lookup(InstId) of {ok, Group, #{id := _} = Data} -> - ?SLOG( - error, - #{ - msg => "health check failed: timeout", - resource_id => InstId - } - ), update_resource(InstId, Group, Data#{status => Status}); Error -> ?SLOG(