Golang實現(xiàn)WebSocket服務的項目實踐

更新時間：2023年05月22日 10:52:43 作者：夢想畫家

本文介紹如何使用Golang實現(xiàn)實時后端WebSocket服務，首先使用Gin框架搭建http服務，然后使用gorilla/websocket庫實現(xiàn)簡單后端WebSocket服務，具有一定的參考價值，感興趣的可以了解一下

我們每天接觸到各類應用，如社交、在線文檔、直播等，后端都需要使用WebSocket技術提供實時通信能力。本文介紹如何使用Golang實現(xiàn)實時后端WebSocket服務，首先使用Gin框架搭建http服務，然后使用gorilla/websocket庫實現(xiàn)簡單后端WebSocket服務，示例實現(xiàn)從0到1的過程，適合初學者快速入門。

Gin實現(xiàn)Web應用

創(chuàng)建Go工程，新增hello.go文件，編寫下面代碼：

package main
import "fmt"
func main() {
? ? ?fmt.Println("Hello, World!")
}

運行可以看到輸出結果。下面增加Web服務功能。安裝Gin庫依賴：

go get -u github.com/gin-gonic/gin

修改上面hello.go文件內容：

package main
import (
? "fmt"
? "github.com/gin-gonic/gin"
)
func main() {
? ? fmt.Println("Hello, World!")
? ? // 創(chuàng)建Gin引擎實例
? ? r := gin.Default()
? ? // 增加路由ping，簡單返回pong
? ? r.GET("/ping", func(c *gin.Context) {
? ? ? ? c.JSON(200, gin.H{
? ? ? ? ? ? "message": "pong",
? ? ? ? })
? ? })
? ? // listen and serve on 0.0.0.0:8080
? ? // 啟動HTTP服務
? ? r.Run()?
}

運行程序，訪問地址：http://localhost:8080/ping，返回結果：

{
"message": "pong"
}

實現(xiàn)WebSocket服務

創(chuàng)建server.go，修改hello.go文件的main方法為hello()，同一包下不能有兩個main方法。需要安裝相應的庫：go get -u github.com/gorilla/websocket,增加下面內容:

package main
import (
? "fmt"
? "log"
? "net/http"
? "github.com/gin-gonic/gin"
? "github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
? // 解決跨域問題
? CheckOrigin: func(r *http.Request) bool {
? ? return true
? },
}?
func ws(c *gin.Context) {
? //更新get請求使用WebSocket協(xié)議
? ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
? if err != nil {
? ?log. Print("upgrade:", err)
? ?return
? }
? defer ws.Close()
? // 監(jiān)聽 ws 消息
? for {
? ?// 從 ws 讀取數據
? ?mt, message, err := ws.ReadMessage()
? ?if err != nil {
? ? log.Println("read:", err)
? ? break
? ?}
? ?log.Printf("recv: %s", message)
? ?//往 ws 寫數據
? ?err = ws.WriteMessage(mt, message)
? ?if err != nil {
? ? log.Println("write:", err)
? ? break
? ?}
? }
}
func main() {
? ? fmt.Println("Websocket Server!")
? ? r := gin.Default()
? ? r.GET("/ws", ws)
? ? r.Run("localhost:8448")
}

核心邏輯為：收到消息后，往客戶端寫回相同內容。代碼增加了注釋，主要差異就是for循環(huán)中的讀寫消息。下面需要html文件進行交互，新建index.html文件：

<!DOCTYPE html>
<html lang="en">
<head>
? ? ?<meta charset="UTF-8">
? ? ?<meta http-equiv="X-UA-Compatible" content="IE=edge">
? ? ?<meta name="viewport" content="width=device-width, initial-scale=1.0">
? ? ?<title>Univer Server</title>
</head>
<body>
? ? ?<script>
? ? ? ? ?var ws = new WebSocket("ws://localhost:8448/ws");
? ? ? ? ?//打開連接時觸發(fā)
? ? ? ? ?ws.onopen = function(evt) {
? ? ? ? ? ? ?console.log("Connection open...");
? ? ? ? ? ? ?ws.send("Hello WebSockets!");
? ? ? ? ?};
? ? ? ? ?//接收到消息時觸發(fā)
? ? ? ? ?ws.onmessage = function(evt) {
? ? ? ? ? ? ?console.log("Received Message: " + evt.data);
? ? ? ? ?};
? ? ? ? ?//連接關閉時觸發(fā)
? ? ? ? ?ws.onclose = function(evt) {
? ? ? ? ? ? ?console.log("Connection closed.");
? ? ? ? ?};
? ? ?</script>
? ? ?Hello WebSocket.
</body>
</html>

script標簽中定義了三個回調函數，連接ws后給服務端發(fā)送消息，服務端收到消息返回相同內容，最后時關閉連接時觸發(fā)。