gin开发WebSocket并开启SSL

原创

已于 2024-06-16 10:54:27 修改 · 840 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#gin #websocket #golang #后端

于 2024-06-16 09:00:00 首次发布

这段时间工作需求用到实现双工通信，因此选择WebSocket来开发，简单记录一下。

1.安装WebSocket

在 Go 语言中操作 WebSocket 可以通过使用 gorilla/websocket 包来实现。

go get github.com/gorilla/websocket

2.服务端代码

// Upgrader 用于升级 HTTP 连接到 WebSocket 连接
var upgrader = websocket.Upgrader{
   
   
    CheckOrigin: func(r *http.Request) bool {
   
   
        return true // 允许所有来源的连接，生产环境应进行适当的来源验证
    },
}

// WebSocketHandler 处理 WebSocket 连接
func WebSocketHandler(c *gin.Context) {
   
   
    conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
    if err != nil {
   
   
        fmt.Println("Upgrade error:", err)
        return
    }
    defer conn.Close()

    for {
   
   
        // 读取消息
        messageType, message, err := conn.ReadMessage()
        if err