deprecated_overflow_gateway.../server.go

143 lines
2.9 KiB
Go
Raw Normal View History

2017-08-18 11:30:29 +00:00
package overflow_gateway_websocket
2017-08-21 10:23:45 +00:00
import (
2017-08-25 02:51:27 +00:00
"net/http"
2017-08-25 02:57:12 +00:00
"sync"
2017-08-21 10:23:45 +00:00
2017-08-24 09:24:38 +00:00
"git.loafle.net/overflow/overflow_gateway_websocket/websocket"
"github.com/valyala/fasthttp"
2017-08-18 11:30:29 +00:00
)
2017-08-24 09:24:38 +00:00
type ()
2017-08-25 09:52:05 +00:00
type sendRequest struct {
s Socket
m []byte
}
2017-08-18 11:30:29 +00:00
type Server interface {
2017-08-24 09:24:38 +00:00
ListenAndServe(addr string) error
2017-08-25 08:13:21 +00:00
HandleSocket(pattern string, o *SocketOptions)
2017-08-25 09:52:05 +00:00
Send(soc Socket, m []byte)
SendAll(m []byte)
2017-08-18 11:30:29 +00:00
}
type server struct {
2017-08-25 09:52:05 +00:00
_option *ServerOptions
_upgrader *websocket.Upgrader
_handlers map[string]*SocketOptions
_sockets map[string]Socket
_sIdMtx sync.Mutex
_addSocketCh chan Socket
_removeSocketCh chan Socket
_sendCh chan sendRequest
_sendAllCh chan []byte
2017-08-18 11:30:29 +00:00
}
2017-08-25 02:33:00 +00:00
func NewServer(o *ServerOptions) Server {
2017-08-18 11:30:29 +00:00
s := &server{
2017-08-24 11:49:51 +00:00
_option: o.Validate(),
2017-08-25 08:13:21 +00:00
_handlers: make(map[string]*SocketOptions, 1),
_sockets: make(map[string]Socket, 100),
2017-08-21 10:23:45 +00:00
}
2017-08-24 10:42:40 +00:00
s._upgrader = &websocket.Upgrader{
2017-08-25 01:57:38 +00:00
HandshakeTimeout: s._option.HandshakeTimeout,
ReadBufferSize: s._option.ReadBufferSize,
WriteBufferSize: s._option.WriteBufferSize,
CheckOrigin: s._option.OnCheckOrigin,
2017-08-25 05:40:04 +00:00
Error: s.onError,
2017-08-25 01:57:38 +00:00
EnableCompression: s._option.EnableCompression,
2017-08-24 10:42:40 +00:00
}
2017-08-21 10:23:45 +00:00
2017-08-18 11:30:29 +00:00
return s
}
2017-08-25 09:52:05 +00:00
func (s *server) addSocket(soc Socket) {
s._addSocketCh <- soc
}
func (s *server) removeSocket(soc Socket) {
s._removeSocketCh <- soc
}
2017-08-25 05:40:04 +00:00
func (s *server) onError(ctx *fasthttp.RequestCtx, status int, reason error) {
2017-08-25 07:40:17 +00:00
ctx.Response.Header.Set("Sec-Websocket-Version", "13")
ctx.Error(http.StatusText(status), status)
2017-08-25 05:40:04 +00:00
2017-08-25 07:40:17 +00:00
s._option.OnError(ctx, status, reason)
2017-08-25 05:40:04 +00:00
}
2017-08-25 08:13:21 +00:00
func (s *server) onDisconnected(soc Socket) {
2017-08-25 09:52:05 +00:00
s.removeSocket(soc)
2017-08-25 08:13:21 +00:00
s._option.OnDisconnected(soc)
2017-08-18 11:30:29 +00:00
}
2017-08-24 10:42:40 +00:00
func (s *server) onConnection(ctx *fasthttp.RequestCtx) {
2017-08-24 14:50:14 +00:00
path := string(ctx.Path())
2017-08-25 02:57:12 +00:00
co, ok := s._handlers[path]
if !ok {
2017-08-25 07:40:17 +00:00
s.onError(ctx, fasthttp.StatusNotFound, nil)
2017-08-25 02:57:12 +00:00
return
}
2017-08-24 14:50:14 +00:00
2017-08-24 10:42:40 +00:00
s._upgrader.Upgrade(ctx, nil, func(conn *websocket.Conn, err error) {
2017-08-24 09:24:38 +00:00
if err != nil {
2017-08-25 07:40:17 +00:00
s.onError(ctx, fasthttp.StatusInternalServerError, err)
2017-08-24 09:24:38 +00:00
return
}
2017-08-25 09:52:05 +00:00
s._sIdMtx.Lock()
2017-08-25 08:13:21 +00:00
id := s._option.IDGenerator(ctx)
2017-08-25 09:52:05 +00:00
s._sIdMtx.Unlock()
2017-08-25 08:13:21 +00:00
soc := NewSocket(id, path, co, conn)
2017-08-25 09:52:05 +00:00
s.addSocket(soc)
2017-08-25 08:13:21 +00:00
s._option.OnConnection(soc)
2017-08-25 06:10:00 +00:00
2017-08-25 08:13:21 +00:00
soc.run()
2017-08-24 09:24:38 +00:00
})
}
2017-08-18 11:30:29 +00:00
2017-08-25 09:52:05 +00:00
func (s *server) listenHandler() {
for {
select {
// Add new a socket
case soc := <-s._addSocketCh:
s._sockets[soc.ID()] = soc
// remove a socket
case soc := <-s._removeSocketCh:
delete(s._sockets, soc.ID())
case sr := <-s._sendCh:
sr.s.Write(sr.m)
case m := <-s._sendAllCh:
for _, soc := range s._sockets {
soc.Write(m)
}
}
}
}
func (s *server) Send(soc Socket, m []byte) {
r := sendRequest{
s: soc,
m: m,
}
s._sendCh <- r
}
func (s *server) SendAll(m []byte) {
s._sendAllCh <- m
}
2017-08-25 08:13:21 +00:00
func (s *server) HandleSocket(pattern string, o *SocketOptions) {
2017-08-25 05:40:04 +00:00
o.onDisconnected = s.onDisconnected
2017-08-25 02:33:00 +00:00
2017-08-24 11:29:22 +00:00
s._handlers[pattern] = o.Validate()
}
2017-08-24 09:24:38 +00:00
func (s *server) ListenAndServe(addr string) error {
2017-08-25 09:52:05 +00:00
go s.listenHandler()
2017-08-24 10:42:40 +00:00
return fasthttp.ListenAndServe(addr, s.onConnection)
2017-08-18 11:30:29 +00:00
}