如何在Golang 中實現WebSocket 的雙向通訊
WebSockets 是一種在客戶端和伺服器之間建立持久化連接的通訊協議，它允許雙向通信，而且在即時性和效率方面表現出色。在 Golang 中，我們可以使用標準函式庫中的 net/http 和 gorilla/websocket 套件來實作 WebSocket 的雙向通訊。本文將介紹如何在 Golang 中實現 WebSocket 的雙向通信，並提供具體的程式碼範例。
步驟一：導入依賴套件
首先，我們需要導入 net/http 和 gorilla/websocket 套件。 net/http 套件提供了 HTTP 伺服器和客戶端的功能，而 gorilla/websocket 套件則是一個 WebSocket 實作庫。
import (
"log"
"net/http"
"github.com/gorilla/websocket"
)步驟二：設定 WebSocket 處理函數
接下來，我們需要設定 WebSocket 的處理函數。這個函數將會處理客戶端的連線請求，並在連線建立成功後，處理客戶端發送的訊息。
func websocketHandler(w http.ResponseWriter, r *http.Request) {
// 将 HTTP 连接升级为 WebSocket 连接
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Error upgrading HTTP connection to WebSocket:", err)
return
}
// 在连接结束时关闭 WebSocket
defer ws.Close()
// 循环处理客户端发来的消息
for {
// 读取客户端发来的消息
messageType, message, err := ws.ReadMessage()
if err != nil {
log.Println("Error reading message from WebSocket:", err)
break
}
// 打印收到的消息
log.Printf("Received message: %s", message)
// 回复客户端收到的消息
if err = ws.WriteMessage(messageType, message); err != nil {
log.Println("Error writing message to WebSocket:", err)
break
}
}
}步驟三：設定WebSocket 升級器
在處理函數中，我們需要使用gorilla/websocket 套件的升級器(upgrader)來將HTTP 連線升級為WebSocket 連線。升級器還可以設定一些參數，例如允許的域以及握手超時時間。
var upgrader = websocket.Upgrader{
// 允许跨域访问
CheckOrigin: func(r *http.Request) bool {
return true
},
}步驟四：設定 HTTP 伺服器
最後，我們需要設定 HTTP 伺服器，將處理函數對應到對應的路徑上，以便客戶端可以透過路徑連接到 WebSocket。
func main() {
// 设置 WebSocket 处理函数
http.HandleFunc("/", websocketHandler)
// 启动 HTTP 服务器
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("Error starting HTTP server:", err)
}
}程式碼範例中的 :8080 是伺服器的監聽位址和連接埠號，你可以根據需要修改。
使用上述程式碼，我們已經成功實現了在 Golang 中透過 WebSocket 實現雙向通訊的功能。在客戶端連接成功後，可以使用類似下面的程式碼來發送和接收訊息：
const socket = new WebSocket("ws://localhost:8080/");
socket.addEventListener("open", (event) => {
socket.send("Hello, server!");
});
socket.addEventListener("message", (event) => {
console.log("Received message from server:", event.data);
});
socket.addEventListener("close", (event) => {
console.log("Server connection closed:", event);
});綜上所述，我們詳細介紹瞭如何在Golang 中實現WebSocket 的雙向通信，並提供了具體的程式碼範例。透過這些範例程式碼，你可以快速上手並實現基於 WebSocket 的即時通訊功能。
以上是如何在golang中實現WebSocket的雙向通信的詳細內容。更多資訊請關注PHP中文網其他相關文章！
