解决Go语言Websocket应用程序中的线程阻塞问题

type Message struct {
    // 定义消息结构
    ConnID string
    Data   []byte
}

type ConnManager struct {
    // 定义连接管理器
    connections map[string]*websocket.Conn
    broadcast   chan Message
}

func NewConnManager() *ConnManager {
    // 创建连接管理器
    return &ConnManager{
        connections: make(map[string]*websocket.Conn),
        broadcast:   make(chan Message),
    }
}

func (cm *ConnManager) Add(connID string, conn *websocket.Conn) {
    // 添加连接到管理器
    cm.connections[connID] = conn
}

func (cm *ConnManager) Remove(connID string) {
    // 从管理器中删除连接
    delete(cm.connections, connID)
}

func (cm *ConnManager) Broadcast(msg Message) {
    // 广播消息给所有连接
    for _, conn := range cm.connections {
        conn.WriteMessage(websocket.TextMessage, msg.Data)
    }
}

func (cm *ConnManager) Run() {
    // 运行连接管理器
    for {
        select {
        case msg := <-cm.broadcast:
            // 接收广播消息并发送给所有连接
            cm.Broadcast(msg)
        }
    }
}

type ConnManager struct {
    // ...
    broadcast   chan Message
}

func NewConnManager() *ConnManager {
    // ...
    return &ConnManager{
        connections: make(map[string]*websocket.Conn),
        broadcast:   make(chan Message, 10), // 设置channel的缓冲大小
    }
}

func (cm *ConnManager) HandleConnection(connID string, conn *websocket.Conn) {
    go func() {
        for {
            messageType, message, err := conn.ReadMessage()
            if err != nil {
                // 处理连接异常
                break
            }
            // 处理消息
            msg := Message{ConnID: connID, Data: message}
            select {
            case cm.broadcast <- msg:
                // 广播消息
            case <-time.After(3 * time.Second):
                // 处理超时
                break
            }
        }
        // 关闭连接
        conn.Close()
        cm.Remove(connID)
    }()
}

func main() {
    cm := NewConnManager()
    // ...
}