我有一个gen_server模块,我使用 gun 作为 http 客户端与 http 服务器建立长拉取连接,所以我在我的模块的 init 中调用 gun:open,但如果 gun:open 失败,我的模块就会失败,所以我的应用程序无法启动。正确的方法是什么。以下是我的代码:
init() ->
lager:debug("http_api_client: connecting to admin server...~n"),
{ok, ConnPid} = gun:open("localhost", 5001),
{ok, Protocol} = gun:await_up(ConnPid),
{ok, #state{conn_pid = ConnPid, streams = #{},protocol = Protocol}}.
基本上,您有两种选择:要么您的进程要求 HTTP 服务器可用(您当前的解决方案(,要么不需要,并在与 HTTP 服务器的连接正常关闭时处理请求(通过返回错误响应(。这篇博文更雄辩地提出了这个想法:https://ferd.ca/it-s-about-the-guarantees.html
您可以通过将此代码分离到一个单独的函数中来做到这一点,如果连接失败,该函数不会崩溃:
try_connect(State) ->
lager:debug("http_api_client: connecting to admin server...~n"),
case gun:open("localhost", 5001) of
{ok, ConnPid} ->
{ok, Protocol} = gun:await_up(ConnPid),
State#state{conn_pid = ConnPid, streams = #{},protocol = Protocol};
{error, _} ->
State#state{conn_pid = undefined}
end.
并从init
调用此函数。也就是说,无论您是否可以连接,您的gen_server都将启动。
init(_) ->
{ok, try_connect(#state{})}.
然后,当您向此gen_server发出需要存在连接的请求时,请检查它是否undefined
:
handle_call(foo, _, State = #state{conn_pid = undefined}) ->
{reply, {error, not_connected}, State};
handle_call(foo, _, State = #state{conn_pid = ConnPid}) ->
%% make a request through ConnPid here
{reply, ok, State};
当然,这意味着如果连接在启动时失败,您的gen_server将永远不会再次尝试连接。您可以添加计时器,也可以添加显式reconnect
命令:
handle_call(reconnect, _, State = #state{conn_pid = undefined}) ->
NewState = try_connect(State),
Result = case NewState of
#state{conn_pid = undefined} ->
reconnect_failed;
_ ->
ok
end,
{reply, Result, NewState};
handle_call(reconnect, _, State) ->
{reply, already_connected, State}.
上面的代码不处理gen_server运行时连接断开的情况。 您可以显式处理,也可以在这种情况下让gen_server进程崩溃,以便它重新启动到"未连接"状态。