refactor(websocket): 移除构造函数并在注册时初始化连接

将WebSocket模块的初始化逻辑从构造函数移动到Register方法中

Author: Zing

js_engine/model/autogo/websocket/websocket_inject.go

@@ -16,14 +16,6 @@ type WebSocketModule struct {
 	mu          sync.Mutex
 }
 
-// New 创建新的 WebSocket 模块实例
-func New() *WebSocketModule {
-	return &WebSocketModule{
-		connections: make(map[int]*websocket.Conn),
-		nextHandle:  1,
-	}
-}
-
 // Name 返回模块名称
 func (m *WebSocketModule) Name() string {
 	return "websocket"
@@ -38,6 +30,13 @@ func (m *WebSocketModule) IsAvailable() bool {
 func (m *WebSocketModule) Register(engine model.Engine) error {
 	vm := engine.GetVM()
 
+	m.mu.Lock()
+	if m.connections == nil {
+		m.connections = make(map[int]*websocket.Conn)
+		m.nextHandle = 1
+	}
+	m.mu.Unlock()
+
 	wsObj := vm.NewObject()
 	vm.Set("websocket", wsObj)
 

lua_engine/model/autogo/websocket/websocket_inject.go

@@ -16,14 +16,6 @@ type WebSocketModule struct {
 	mu          sync.Mutex
 }
 
-// New 创建新的 WebSocket 模块实例
-func New() *WebSocketModule {
-	return &WebSocketModule{
-		connections: make(map[int]*websocket.Conn),
-		nextHandle:  1,
-	}
-}
-
 // Name 返回模块名称
 func (m *WebSocketModule) Name() string {
 	return "websocket"
@@ -38,6 +30,13 @@ func (m *WebSocketModule) IsAvailable() bool {
 func (m *WebSocketModule) Register(engine model.Engine) error {
 	state := engine.GetState()
 
+	m.mu.Lock()
+	if m.connections == nil {
+		m.connections = make(map[int]*websocket.Conn)
+		m.nextHandle = 1
+	}
+	m.mu.Unlock()
+
 	wsObj := state.NewTable()
 	state.SetGlobal("websocket", wsObj)