90 lines
1.9 KiB
Go
90 lines
1.9 KiB
Go
package overflow_gateway_websocket
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"git.loafle.net/overflow/overflow_gateway_websocket/websocket"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
type ()
|
|
|
|
type Server interface {
|
|
ListenAndServe(addr string) error
|
|
HandleClient(pattern string, o *ClientOptions)
|
|
}
|
|
|
|
type server struct {
|
|
_option *ServerOptions
|
|
_upgrader *websocket.Upgrader
|
|
_handlers map[string]*ClientOptions
|
|
_clients map[string]Client
|
|
_cMtx sync.Mutex
|
|
}
|
|
|
|
func NewServer(o *ServerOptions) Server {
|
|
s := &server{
|
|
_option: o.Validate(),
|
|
_handlers: make(map[string]*ClientOptions, 1),
|
|
_clients: make(map[string]Client, 100),
|
|
}
|
|
|
|
s._upgrader = &websocket.Upgrader{
|
|
HandshakeTimeout: s._option.HandshakeTimeout,
|
|
ReadBufferSize: s._option.ReadBufferSize,
|
|
WriteBufferSize: s._option.WriteBufferSize,
|
|
CheckOrigin: s._option.OnCheckOrigin,
|
|
Error: s._option.OnError,
|
|
EnableCompression: s._option.EnableCompression,
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func (s *server) onPush(cb OnPushFunc) {
|
|
|
|
}
|
|
|
|
func (s *server) onDisconnected(c Client) {
|
|
delete(s._clients, c.ID())
|
|
|
|
s._option.OnDisconnected(c)
|
|
}
|
|
|
|
func (s *server) onConnection(ctx *fasthttp.RequestCtx) {
|
|
path := string(ctx.Path())
|
|
co, ok := s._handlers[path]
|
|
if !ok {
|
|
ctx.Response.Header.Set("Sec-Websocket-Version", "13")
|
|
ctx.Error(http.StatusText(fasthttp.StatusNotFound), fasthttp.StatusNotFound)
|
|
|
|
log.Printf("Path[%s] is not exist.", path)
|
|
return
|
|
}
|
|
|
|
s._upgrader.Upgrade(ctx, nil, func(conn *websocket.Conn, err error) {
|
|
if err != nil {
|
|
log.Print("upgrade:", err)
|
|
return
|
|
}
|
|
s._cMtx.Lock()
|
|
cid := s._option.IDGenerator(ctx)
|
|
c := NewClient(cid, path, co, conn)
|
|
s._clients[cid] = c
|
|
s._cMtx.Unlock()
|
|
|
|
s._option.OnConnection(c)
|
|
})
|
|
}
|
|
|
|
func (s *server) HandleClient(pattern string, o *ClientOptions) {
|
|
|
|
s._handlers[pattern] = o.Validate()
|
|
}
|
|
|
|
func (s *server) ListenAndServe(addr string) error {
|
|
return fasthttp.ListenAndServe(addr, s.onConnection)
|
|
}
|