1. 完善websocket通信逻辑

2. 实现Switch接口

internal/config/config.go

@@ -17,6 +17,9 @@ type Config struct {
 
 	// Database 数据库配置
 	Database DatabaseConfig `yaml:"database"`
+
+	// WebSocket WebSocket配置
+	WebSocket WebSocketConfig `yaml:"websocket"`
 }
 
 // ServerConfig 代表服务器配置
@@ -67,9 +70,19 @@ type DatabaseConfig struct {
 	ConnMaxLifetime int `yaml:"conn_max_lifetime"`
 }
 
+// WebSocketConfig 代表WebSocket配置
+type WebSocketConfig struct {
+	// Timeout WebSocket请求超时时间(秒)
+	Timeout int `yaml:"timeout"`
+}
+
 // NewConfig 创建并返回一个新的配置实例
 func NewConfig() *Config {
-	return &Config{}
+	return &Config{
+		WebSocket: WebSocketConfig{
+			Timeout: 5, // 默认5秒超时
+		},
+	}
 }
 
 // Load 从指定路径加载配置文件
@@ -101,3 +114,11 @@ func (c *Config) GetDatabaseConnectionString() string {
 		c.Database.SSLMode,
 	)
 }
+
+// GetWebSocketTimeout 获取WebSocket超时时间(秒)
+func (c *Config) GetWebSocketTimeout() int {
+	if c.WebSocket.Timeout <= 0 {
+		return 5 // 默认5秒超时
+	}
+	return c.WebSocket.Timeout
+}