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 !" ：