golang WebSocket教程：如何实现实时通信

Golang WebSocket教程：如何实现实时通信

首先，让我们来介绍一下什么是WebSocket。WebSocket是一种在Web浏览器和服务器之间进行全双工通信的协议。与HTTP协议不同，WebSocket允许服务器向客户端推送数据，并且可以实现实时通信、事件驱动等功能。

在本教程中，我们将使用Golang来实现一个简单的WebSocket服务器。我们将演示如何建立WebSocket连接、发送和接收消息，并通过具体的代码示例来说明。

首先，我们需要导入Golang的WebSocket包：

import (
"log"
"net/http"
"github.com/gorilla/websocket"
)

然后，我们需要定义一个WebSocket升级器（upgrader），用于将HTTP协议升级为WebSocket协议：

var upgrader = websocket.Upgrader{
ReadBufferSize:  1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool { return true },
}

接下来，我们需要定义一个处理WebSocket连接的函数。这个函数将被注册为HTTP处理器，并将在每次有WebSocket连接请求时调用：

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
// 将HTTP连接升级为WebSocket连接
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
// 在新的goroutine中处理WebSocket连接
go handleWebSocketConnection(conn)
}

在处理WebSocket连接的函数中，我们需要定义一个循环来处理接收和发送消息：

func handleWebSocketConnection(conn *websocket.Conn) {
for {
// 读取客户端发送的消息
_, msg, err := conn.ReadMessage()
if err != nil {
log.Println(err)
return
}
// 打印接收到的消息
log.Println("Received message:", string(msg))
// 发送消息给客户端
err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, client!"))
if err != nil {
log.Println(err)
return
}
}
}

现在，我们只需要将上述函数注册为HTTP处理器，并启动一个HTTP服务器即可：

func main() {
// 注册WebSocket处理器
http.HandleFunc("/ws", handleWebSocket)
// 启动HTTP服务器
log.Println("WebSocket server started on localhost:8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal(err)
}
}

至此，我们已经完成了一个简单的WebSocket服务器。当有客户端通过WebSocket连接到服务器时，服务器将会接收客户端发送的消息，并回复一个固定的消息。

要测试服务器，我们可以使用一个浏览器插件，如”Simple WebSocket Client”，输入服务器地址ws://localhost:8080/ws并连接即可。在插件的消息框内输入消息并发送，我们将能在终端中看到接收到的消息，并同时向客户端回复一条消息。