Go语言(Golang)提供了内置的标准库net/http来实现WebSocket通信。以下是一个简单的示例代码,演示了如何在Go中使用WebSocket:
package main
import (
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{} // WebSocket升级器
func echo(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil) // 将HTTP连接升级为WebSocket连接
if err != nil {
log.Println(err)
return
}
defer conn.Close()
for {
messageType, message, err := conn.ReadMessage() // 读取客户端发送的消息
if err != nil {
log.Println(err)
return
}
log.Printf("Received: %s", message)
err = conn.WriteMessage(messageType, message) // 将消息回送给客户端
if err != nil {
log.Println(err)
return
}
}
}
func main() {
http.HandleFunc("/echo", echo)
log.Fatal(http.ListenAndServe(":8080", nil)) // 监听并启动Web服务
}
在上述示例中,我们通过github.com/gorilla/websocket包提供的Upgrader将HTTP连接升级为WebSocket连接。然后,我们使用conn.ReadMessage()方法从客户端接收消息,并使用conn.WriteMessage()方法将消息回送给客户端。
你可以根据自己的需求进一步扩展和修改该示例代码来实现更复杂的功能。
内容由零声教学AI助手提供,问题来源于学员提问




