GO语言实现websocket

在Go语言中实现WebSocket通信是相对简单的。首先，你需要导入`net/http`和`github.com/gorilla/websocket`两个包。然后，你可以按照以下步骤实现WebSocket：

1. 创建一个HTTP处理器函数，用于升级HTTP连接为WebSocket连接。

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为WebSocket连接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }

    // 在这里处理WebSocket连接
}

2. 创建一个`Upgrader`对象，用于配置WebSocket连接的选项。

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

3. 在WebSocket连接中实现读取和写入数据的逻辑。

for {
    // 读取客户端发送的消息
    messageType, p, err := conn.ReadMessage()
    if err != nil {
        log.Println(err)
        return
    }

    // 处理接收到的消息

    // 发送消息给客户端
    err = conn.WriteMessage(messageType, p)
    if err != nil {
        log.Println(err)
        return
    }
}

4. 创建一个HTTP服务器，并将WebSocket处理器注册到指定的路径上。

http.HandleFunc("/websocket", websocketHandler)
log.Fatal(http.ListenAndServe(":8080", nil))

这样，你就实现了一个简单的WebSocket服务器。你可以根据需要在`websocketHandler`函数中处理接收到的消息，以及在循环中发送消息给客户端。记得根据你的实际需求进行错误处理和逻辑编写。