如何在Gin框架中使用Websocket？

Toormi2024-05-06 10:56

在Gin框架中使用Websocket，你需要使用一个第三方库，例如github.com/gorilla/websocket。以下是一个基本的示例，展示了如何在Gin中设置一个Websocket服务器：

首先，你需要安装gorilla/websocket包，你可以使用以下命令进行安装：

bash 复制代码

go get github.com/gorilla/websocket

然后，你可以创建一个新的Gin路由，并将其绑定到一个Websocket处理函数上：

go 复制代码

package main

import (
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"net/http"
)

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

func main() {
	r := gin.Default()

	r.GET("/ws", func(c *gin.Context) {
		handleWebsocket(c.Writer, c.Request)
	})

	r.Run(":8080")
}

func handleWebsocket(w http.ResponseWriter, r *http.Request) {
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		return
	}
	defer conn.Close()

	for {
		messageType, p, err := conn.ReadMessage()
		if err != nil {
			return
		}
		if err := conn.WriteMessage(messageType, p); err != nil {
			return
		}
	}
}

在这个例子中，我们创建了一个新的Gin路由/ws，当客户端向这个路径发送GET请求时，我们的服务器就会将这个HTTP连接升级为一个Websocket连接。然后，我们进入一个无限循环，不断地从Websocket连接中读取消息，然后将这个消息原样写回。这就是一个最基本的Websocket回声服务器。

你可以根据你的需求修改handleWebsocket函数，例如，你可以在接收到消息时调用你的业务逻辑函数，而不是简单地将消息原样写回。