From a47f6e23bc476ee4c223e5cdc2cad34b9fe38bd2 Mon Sep 17 00:00:00 2001
From: terry-xiaoyu <506895667@qq.com>
Date: Thu, 9 Apr 2020 21:28:20 +0800
Subject: [PATCH] Fix add_reconn_callbk failed

---
 src/ecpool_worker.erl | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/ecpool_worker.erl b/src/ecpool_worker.erl
index fe048e9d1..182a6c497 100644
--- a/src/ecpool_worker.erl
+++ b/src/ecpool_worker.erl
@@ -135,9 +135,15 @@ handle_call(Req, _From, State) ->
 handle_cast({set_reconn_callbk, OnReconnect}, State) ->
     {noreply, State#state{on_reconnect = OnReconnect}};
 
-handle_cast({add_reconn_callbk, OnReconnect}, State = #state{on_reconnect = OnReconnectList}) ->
+handle_cast({add_reconn_callbk, OnReconnect}, State = #state{on_reconnect = OnReconnectList}) when is_list(OnReconnectList) ->
     {noreply, State#state{on_reconnect = [OnReconnect | OnReconnectList]}};
 
+handle_cast({add_reconn_callbk, OnReconnect}, State = #state{on_reconnect = undefined}) ->
+    {noreply, State#state{on_reconnect = [OnReconnect]}};
+
+handle_cast({add_reconn_callbk, OnReconnect}, State = #state{on_reconnect = OnReconnect0}) ->
+    {noreply, State#state{on_reconnect = [OnReconnect, OnReconnect0]}};
+
 handle_cast(_Msg, State) ->
     {noreply, State}.