Fix duplicated netty pipeline handler error

kuvasz-uptime · Aug 29, 2020 · e07ca6c · e07ca6c
1 parent 52303f0
commit e07ca6c
Showing 1 changed file with 15 additions and 10 deletions.
diff --git a/src/main/kotlin/com/kuvaszuptime/kuvasz/services/HttpCommunicationLogger.kt b/src/main/kotlin/com/kuvaszuptime/kuvasz/services/HttpCommunicationLogger.kt
@@ -16,22 +16,27 @@ class HttpCommunicationLogger(private val logbook: Logbook) : BeanCreatedEventLi
 
     override fun onCreated(event: BeanCreatedEvent<ChannelPipelineCustomizer>): ChannelPipelineCustomizer {
         val customizer = event.bean
+        val handlerName = "logbook"
         if (customizer.isServerChannel) {
             customizer.doOnConnect { pipeline: ChannelPipeline ->
-                pipeline.addAfter(
-                    ChannelPipelineCustomizer.HANDLER_HTTP_SERVER_CODEC,
-                    "logbook",
-                    LogbookServerHandler(logbook)
-                )
+                if (pipeline.get(handlerName) == null) {
+                    pipeline.addAfter(
+                        ChannelPipelineCustomizer.HANDLER_HTTP_SERVER_CODEC,
+                        handlerName,
+                        LogbookServerHandler(logbook)
+                    )
+                }
                 pipeline
             }
         } else {
             customizer.doOnConnect { pipeline: ChannelPipeline ->
-                pipeline.addAfter(
-                    ChannelPipelineCustomizer.HANDLER_HTTP_CLIENT_CODEC,
-                    "logbook",
-                    LogbookClientHandler(logbook)
-                )
+                if (pipeline.get(handlerName) == null) {
+                    pipeline.addAfter(
+                        ChannelPipelineCustomizer.HANDLER_HTTP_CLIENT_CODEC,
+                        handlerName,
+                        LogbookClientHandler(logbook)
+                    )
+                }
                 pipeline
             }
         }