fix: 修复wsConf的func参数问题

main v1.1.2
NoahLan 2 years ago
parent 40860ee0f9
commit ba3df17bc4

@ -25,6 +25,9 @@ type (
TLSCertificate string `json:",optional"`
// TLS 证书key地址
TLSKey string `json:",optional"`
}
WSServerFullConf struct {
WSServerConf
// check origin
CheckOrigin func(*http.Request) bool `json:",optional"`
}

@ -11,16 +11,16 @@ import (
"strings"
)
type WsConfOption func(conf config.WSServerConf)
type WsConfOption func(conf config.WSServerFullConf)
func WithWSCheckOrigin(fn func(*http.Request) bool) WsConfOption {
return func(conf config.WSServerConf) {
return func(conf config.WSServerFullConf) {
conf.CheckOrigin = fn
}
}
// ListenWebsocket 开始监听Websocket
func (ngin *Engine) ListenWebsocket(conf config.WSServerConf, opts ...WsConfOption) error {
func (ngin *Engine) ListenWebsocket(conf config.WSServerFullConf, opts ...WsConfOption) error {
for _, opt := range opts {
opt(conf)
}
@ -51,7 +51,7 @@ func (ngin *Engine) handleWS(conn *websocket.Conn) {
ngin.handle(wsConn)
}
func (ngin *Engine) upgradeWebsocket(conf config.WSServerConf) {
func (ngin *Engine) upgradeWebsocket(conf config.WSServerFullConf) {
upgrade := websocket.Upgrader{
HandshakeTimeout: conf.HandshakeTimeout,
ReadBufferSize: conf.ReadBufferSize,

Loading…
Cancel
Save