You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
nnet/server_ws.go

67 lines
1.9 KiB
Go

package nnet
import (
"fmt"
"git.noahlan.cn/noahlan/nnet/config"
"git.noahlan.cn/noahlan/nnet/connection"
"git.noahlan.cn/noahlan/ntool/nlog"
"github.com/gorilla/websocket"
"net/http"
"os"
"strings"
)
// ListenWebsocket 开始监听Websocket
func (ngin *Engine) ListenWebsocket(conf config.WSServerConf) error {
err := ngin.setup()
if err != nil {
nlog.Errorf("%s failed to setup server, err:%v", ngin.LogPrefix(), err)
return err
}
nlog.Infof("%s now listening websocket at %s.", ngin.LogPrefix(), conf.Addr)
ngin.upgradeWebsocket(conf)
if conf.IsTLS() {
if err := http.ListenAndServeTLS(conf.Addr, conf.TLSCertificate, conf.TLSKey, nil); err != nil {
nlog.Errorf("%s failed to listening websocket with TLS at %s %v", ngin.LogPrefix(), conf.Addr, err)
return err
}
} else {
if err := http.ListenAndServe(conf.Addr, nil); err != nil {
nlog.Errorf("%s failed to listening websocket at %s %v", ngin.LogPrefix(), conf.Addr, err)
return err
}
}
return nil
}
func (ngin *Engine) handleWS(conn *websocket.Conn) {
wsConn := connection.NewWSConn(conn)
ngin.handle(wsConn)
}
func (ngin *Engine) upgradeWebsocket(conf config.WSServerConf) {
upgrade := websocket.Upgrader{
HandshakeTimeout: conf.HandshakeTimeout,
ReadBufferSize: conf.ReadBufferSize,
WriteBufferSize: conf.WriteBufferSize,
CheckOrigin: conf.CheckOrigin,
EnableCompression: conf.Compression,
}
path := fmt.Sprintf("/%s", strings.TrimPrefix(conf.Path, "/"))
http.HandleFunc(path, func(writer http.ResponseWriter, request *http.Request) {
conn, err := upgrade.Upgrade(writer, request, nil)
if err != nil {
nlog.Errorf("%s Upgrade failure, URI=%ng, Error=%ng", ngin.LogPrefix(), request.RequestURI, err.Error())
return
}
err = ngin.goPool.Submit(func() {
ngin.handleWS(conn)
})
if err != nil {
nlog.Errorf("%s submit conn pool err: %v", ngin.LogPrefix(), err.Error())
os.Exit(1)
}
})
}