From 0569388f0f81f52fcff92d23a72dc019a60f3e5f Mon Sep 17 00:00:00 2001
From: terry-xiaoyu <506895667@qq.com>
Date: Sat, 20 Jul 2019 19:02:49 +0800
Subject: [PATCH] Fix websocket still send willmsg even after disconnect is
 received

---
 src/emqx_ws_channel.erl | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/emqx_ws_channel.erl b/src/emqx_ws_channel.erl
index 23a4d8d4c..e8c86202a 100644
--- a/src/emqx_ws_channel.erl
+++ b/src/emqx_ws_channel.erl
@@ -296,8 +296,8 @@ websocket_info({binary, Data}, State) ->
 websocket_info({shutdown, Reason}, State) ->
     shutdown(Reason, State);
 
-websocket_info({stop, Reason}, State) ->
-    {stop, State#state{shutdown = Reason}};
+websocket_info(stop, State) ->
+    {stop, State};
 
 websocket_info(Info = {'EXIT', SessionPid, Reason}, State = #state{proto_state = ProtoState}) ->
     case emqx_protocol:session(ProtoState) of
@@ -363,12 +363,12 @@ ensure_stats_timer(State) ->
     State.
 
 shutdown(Reason = {shutdown, _}, State) ->
-    self() ! {stop, Reason},
-    {ok, State};
+    self() ! stop,
+    {ok, State#state{shutdown = Reason}};
 shutdown(Reason, State) ->
     %% Fix the issue#2591(https://github.com/emqx/emqx/issues/2591#issuecomment-500278696)
-    self() ! {stop, {shutdown, Reason}},
-    {ok, State}.
+    self() ! stop,
+    {ok, State#state{shutdown = {shutdown, Reason}}}.
 
 wsock_stats() ->
     [{Key, emqx_pd:get_counter(Key)} || Key <- ?SOCK_STATS].