22 lines
514 B
Go
22 lines
514 B
Go
|
// Package utils -----------------------------
|
|||
|
|
// @file : hertzWSUpgrade.go
|
|||
|
|
// @author : JJXu
|
|||
|
|
// @contact : wavingbear@163.com
|
|||
|
|
// @time : 2022/6/28 14:19
|
|||
|
|
// -------------------------------------------
|
|||
|
|
package ws
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"github.com/gorilla/websocket"
|
|||
|
|
"net/http"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var UpGrader = websocket.Upgrader{
|
|||
|
|
ReadBufferSize: 1024,
|
|||
|
|
WriteBufferSize: 1024,
|
|||
|
|
CheckOrigin: func(r *http.Request) bool {
|
|||
|
|
// 检查请求的来源是否允许websocket连接，可根据需求自行实现
|
|||
|
|
return true
|
|||
|
|
},
|
|||
|
|
}
|