From 40977e529a9a8a29fdf4d260408ea0e861186d88 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=91=A8=E5=AD=90=E5=8D=9A?= <349832309@qq.com>
Date: Sat, 8 Sep 2018 13:35:23 +0800
Subject: [PATCH] Retain flag in retained message must set to 1

---
 src/emqx_session.erl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/emqx_session.erl b/src/emqx_session.erl
index 05142297e..8e6e150c5 100644
--- a/src/emqx_session.erl
+++ b/src/emqx_session.erl
@@ -726,9 +726,10 @@ run_dispatch_steps([{qos, SubQoS}|Steps], Msg = #message{qos = PubQoS}, State =
     run_dispatch_steps(Steps, Msg#message{qos = min(SubQoS, PubQoS)}, State);
 run_dispatch_steps([{qos, SubQoS}|Steps], Msg = #message{qos = PubQoS}, State = #state{upgrade_qos = true}) ->
     run_dispatch_steps(Steps, Msg#message{qos = max(SubQoS, PubQoS)}, State);
+run_dispatch_steps([{rap, _Rap}|Steps], Msg = #message{flags = Flags, headers = #{retained := true}}, State = #state{}) ->
+    run_dispatch_steps(Steps, Msg#message{flags = maps:put(retain, true, Flags)}, State);
 run_dispatch_steps([{rap, 0}|Steps], Msg = #message{flags = Flags}, State = #state{}) ->
-    Flags1 = maps:put(retain, false, Flags),
-    run_dispatch_steps(Steps, Msg#message{flags = Flags1}, State);
+    run_dispatch_steps(Steps, Msg#message{flags = maps:put(retain, false, Flags)}, State);
 run_dispatch_steps([{rap, _}|Steps], Msg, State) ->
     run_dispatch_steps(Steps, Msg, State);
 run_dispatch_steps([{subid, SubId}|Steps], Msg, State) ->