From ddae80a5f4282a840d02dc65bd0ef80c3924c7cd Mon Sep 17 00:00:00 2001
From: Daniel <845765@qq.com>
Date: Tue, 9 Dec 2025 12:20:57 +0800
Subject: [PATCH] :art: Improve exit

Signed-off-by: Daniel <845765@qq.com>
---
 kernel/server/serve.go   | 1 +
 kernel/util/websocket.go | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/kernel/server/serve.go b/kernel/server/serve.go
index 479f903f1..6c06e479c 100644
--- a/kernel/server/serve.go
+++ b/kernel/server/serve.go
@@ -571,6 +571,7 @@ func serveDebug(ginServer *gin.Engine) {
 }
 
 func serveWebSocket(ginServer *gin.Engine) {
+	util.WebSocketServer = melody.New()
 	util.WebSocketServer.Config.MaxMessageSize = 1024 * 1024 * 8
 
 	ginServer.GET("/ws", func(c *gin.Context) {
diff --git a/kernel/util/websocket.go b/kernel/util/websocket.go
index 2e1f3c022..29aaa2937 100644
--- a/kernel/util/websocket.go
+++ b/kernel/util/websocket.go
@@ -26,7 +26,7 @@ import (
 )
 
 var (
-	WebSocketServer = melody.New()
+	WebSocketServer *melody.Melody
 
 	// map[string]map[string]*melody.Session{}
 	sessions = sync.Map{} // {appId, {sessionId, session}}