Fix memory leak

softwaremill · Mar 27, 2024 · 83fa2f8 · 83fa2f8
1 parent b4e7192
commit 83fa2f8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...server/src/main/scala/sttp/tapir/server/netty/internal/WebSocketControlFrameHandler.scala b/...server/src/main/scala/sttp/tapir/server/netty/internal/WebSocketControlFrameHandler.scala
@@ -16,10 +16,14 @@ class NettyControlFrameHandler(ignorePong: Boolean, autoPongOnPing: Boolean, dec
       case ping: PingWebSocketFrame =>
         if (autoPongOnPing) {
           val _ = ctx.writeAndFlush(new PongWebSocketFrame(ping.content().retain()))
+        } else {
+          val _ = ping.content().release()
         }
-      case pong: PongWebSocketFrame => 
+      case pong: PongWebSocketFrame =>
         if (!ignorePong) {
           val _ = ctx.fireChannelRead(pong)
+        } else {
+          val _ = pong.content().release()
         }
       case close: CloseWebSocketFrame =>
         if (decodeCloseRequests) {