gin 使用gorilla 关闭 websocket 时候设置close code 和message

gin 使用gorilla 时建立websocket的教程很多博客都有写，但是很少有人讲关闭websocket时自定义 close code 和message。主要是使用：websocket.FormatCloseMessage， WriteControl 两个方法。可用close code 的范围见文档：https://developer.mozilla.org/zh-CN/docs/Web/API/CloseEvent#属性。下面给出示例：

import (

	"fmt"

	"net/http"

	"sync"

	"github.com/gin-gonic/gin"

	"github.com/gorilla/websocket"

)

var (

	upGrader = websocket.Upgrader{

		CheckOrigin: func(r *http.Request) bool {

			return true

		},

	}

	wsPriceOnce sync.Once

)

func Test(c *gin.Context) {

	//升级get请求为webSocket协议

	ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)

	if err != nil {

		c.Writer.Write([]byte(err.Error()))

		return

	}

	m := map[string]interface{}{

		"msg" :"ok",

	}

	er := ws.WriteJSON(m)

	fmt.Println(er)

	// 自定义关闭消息

	closeCode := 4001

	message := "server closed ! "

	closeMessage := websocket.FormatCloseMessage(closeCode, message)

	deadline := time.Now().Add(time.Second)

	e := ws.WriteControl(websocket.CloseMessage, closeMessage, deadline)

	if e != nil {

		fmt.Println(e)

	}

}

在postman中测试效果图，我们看到close code 为4001， message 也是我们设置的 "server closed !" ：

巴特西

gin 使用gorilla 关闭 websocket 时候设置close code 和message

最新文章

热门文章