Add pingreq handler when active, not when handler is added

swift-server-community · Dec 14, 2021 · 7d343da · 7d343da
1 parent 49c60c4
commit 7d343da
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/MQTTNIO/ChannelHandlers/MQTTMessageHandler.swift b/Sources/MQTTNIO/ChannelHandlers/MQTTMessageHandler.swift
@@ -35,10 +35,16 @@ class MQTTMessageHandler: ChannelDuplexHandler {
     }
 
     func handlerAdded(context: ChannelHandlerContext) {
+        if context.channel.isActive {
+            self.pingreqHandler?.start(context: context)
+        }
+    }
+
+    func channelActive(context: ChannelHandlerContext) {
         self.pingreqHandler?.start(context: context)
     }
 
-    func handlerRemoved(context: ChannelHandlerContext) {
+    func channelInactive(context: ChannelHandlerContext) {
         self.pingreqHandler?.stop()
     }