Start pingreq handler on handler added

swift-server-community · Nov 22, 2021 · 55da431 · 55da431
1 parent b7957d3
commit 55da431
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Sources/MQTTNIO/ChannelHandlers/MQTTMessageHandler.swift b/Sources/MQTTNIO/ChannelHandlers/MQTTMessageHandler.swift
@@ -34,14 +34,12 @@ class MQTTMessageHandler: ChannelDuplexHandler {
         self.decoder = .init(.init(version: client.configuration.version))
     }
 
-    func channelActive(context: ChannelHandlerContext) {
+    func handlerAdded(context: ChannelHandlerContext) {
         self.pingreqHandler?.start(context: context)
-        context.fireChannelActive()
     }
 
-    func channelInactive(context: ChannelHandlerContext) {
+    func handlerRemoved(context: ChannelHandlerContext) {
         self.pingreqHandler?.stop()
-        context.fireChannelInactive()
     }
 
     func write(context: ChannelHandlerContext, data: NIOAny, promise: EventLoopPromise<Void>?) {