From 775bfbffc18daff74759e731faf9dfb56206dd8b Mon Sep 17 00:00:00 2001
From: tursom <tursom@foxmail.com>
Date: Mon, 26 Apr 2021 16:27:52 +0800
Subject: [PATCH] add AutoCloseWebSocketHandler

---
 .../core/ws/AutoCloseWebSocketHandler.kt      | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 ts-core/ts-ws-client/src/main/kotlin/cn/tursom/core/ws/AutoCloseWebSocketHandler.kt

diff --git a/ts-core/ts-ws-client/src/main/kotlin/cn/tursom/core/ws/AutoCloseWebSocketHandler.kt b/ts-core/ts-ws-client/src/main/kotlin/cn/tursom/core/ws/AutoCloseWebSocketHandler.kt
new file mode 100644
index 0000000..2aa08cb
--- /dev/null
+++ b/ts-core/ts-ws-client/src/main/kotlin/cn/tursom/core/ws/AutoCloseWebSocketHandler.kt
@@ -0,0 +1,26 @@
+package cn.tursom.core.ws
+
+import io.netty.buffer.ByteBuf
+import io.netty.handler.codec.http.websocketx.TextWebSocketFrame
+
+interface AutoCloseWebSocketHandler : WebSocketHandler {
+  override fun readMessage(client: WebSocketClient, msg: TextWebSocketFrame) {
+    super.readMessage(client, msg)
+    msg.release()
+  }
+
+  override fun readMessage(client: WebSocketClient, msg: ByteBuf) {
+    super.readMessage(client, msg)
+    msg.release()
+  }
+
+  override fun readPing(client: WebSocketClient, msg: ByteBuf) {
+    super.readPing(client, msg)
+    msg.release()
+  }
+
+  override fun readPong(client: WebSocketClient, msg: ByteBuf) {
+    super.readPong(client, msg)
+    msg.release()
+  }
+}
\ No newline at end of file