53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
|
package websocket
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
// MessageHandleFunc for handling messages
|
||
|
|
type MessageHandleFunc func(msg *Message)
|
||
|
|
|
||
|
|
// WebsocketHandlerService to handle every Message on there Subject by Handlers
|
||
|
|
type WebsocketHandlerService struct {
|
||
|
|
inputMSG chan *Message
|
||
|
|
server *Server
|
||
|
|
handlers map[string]MessageHandleFunc
|
||
|
|
FallbackHandler MessageHandleFunc
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewWebsocketHandlerService with Websocket Server
|
||
|
|
func NewWebsocketHandlerService() *WebsocketHandlerService {
|
||
|
|
ws := WebsocketHandlerService{
|
||
|
|
handlers: make(map[string]MessageHandleFunc),
|
||
|
|
inputMSG: make(chan *Message),
|
||
|
|
}
|
||
|
|
ws.server = NewServer(ws.inputMSG, NewSessionManager())
|
||
|
|
return &ws
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ws *WebsocketHandlerService) messageHandler() {
|
||
|
|
for msg := range ws.inputMSG {
|
||
|
|
if handler, ok := ws.handlers[msg.Subject]; ok {
|
||
|
|
handler(msg)
|
||
|
|
} else if ws.FallbackHandler != nil {
|
||
|
|
ws.FallbackHandler(msg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetHandler for a message type by subject
|
||
|
|
func (ws *WebsocketHandlerService) SetHandler(subject string, f MessageHandleFunc) {
|
||
|
|
ws.handlers[subject] = f
|
||
|
|
}
|
||
|
|
|
||
|
|
// Listen on net/http server at `path` and start running handling
|
||
|
|
func (ws *WebsocketHandlerService) Listen(path string) {
|
||
|
|
http.HandleFunc(path, ws.server.Handler)
|
||
|
|
go ws.messageHandler()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close webserver
|
||
|
|
func (ws *WebsocketHandlerService) Close() {
|
||
|
|
close(ws.inputMSG)
|
||
|
|
}
|