阅读量:0
在Go语言中实现WebSocket通信，可以使用标准库中提供的net/http和github.com/gorilla/websocket包。以下是一个简单的示例：
package main import ( "fmt" "net/http" "github.com/gorilla/websocket" ) var upgrader = websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { return true }, } func handler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { fmt.Println("Error upgrading to WebSocket:", err) return } for { // 读取客户端发送的消息 _, message, err := conn.ReadMessage() if err != nil { fmt.Println("Error reading message:", err) break } // 输出客户端发送的消息 fmt.Printf("Received message: %sn", message) // 发送消息给客户端 err = conn.WriteMessage(websocket.TextMessage, []byte("Message received")) if err != nil { fmt.Println("Error writing message:", err) break } } } func main() { http.HandleFunc("/", handler) http.ListenAndServe(":8080", nil) } 在上面的代码中，我们首先创建了一个upgrader实例，用于将HTTP连接升级为WebSocket连接。然后定义了一个处理函数handler，当有客户端连接时会调用该函数。在处理函数中，我们通过upgrader.Upgrade方法将HTTP连接升级为WebSocket连接，并通过conn.ReadMessage读取客户端发送的消息，然后通过conn.WriteMessage发送消息给客户端。
最后在main函数中，我们通过http.HandleFunc将处理函数注册到路由上，然后通过http.ListenAndServe启动HTTP服务器，监听在8080端口上。
在客户端可以使用JavaScript来与该WebSocket服务器进行通信，例如：
const ws = new WebSocket("ws://localhost:8080/"); ws.onopen = function() { console.log("Connected to WebSocket server"); ws.send("Hello, server!"); }; ws.onmessage = function(event) { console.log("Received message from server:", event.data); }; 通过这种方式，可以使用Go语言实现WebSocket通信。
