zirconium-go/core/websocket_server.go

50 lines
1.1 KiB
Go
Raw Normal View History

package core
import (
"fmt"
"github.com/google/logger"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
"net/http"
)
var wsUpgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
type WebsocketServer struct {
r *mux.Router
connHandler *ConnectionHandler
cfg *Config
}
func NewWebsocketServer(cfg *Config, connHandler *ConnectionHandler) *WebsocketServer {
wss := &WebsocketServer{}
wss.connHandler = connHandler
wss.cfg = cfg
r := mux.NewRouter()
wss.r = r
r.HandleFunc("/", func(response http.ResponseWriter, request *http.Request) {
_, _ = response.Write([]byte("Zirconium server is up and running!"))
}).Methods("GET")
r.HandleFunc(cfg.Websocket.Endpoint, func(w http.ResponseWriter, r *http.Request) {
ws, err := wsUpgrader.Upgrade(w, r, nil)
if err != nil {
logger.Errorf(err.Error())
return
}
wss.connHandler.HandleNewConnection(ws)
})
return wss
}
func (wss *WebsocketServer) Run() error {
addr := fmt.Sprintf("%s:%d", wss.cfg.Websocket.Host, wss.cfg.Websocket.Port)
return http.ListenAndServe(addr, wss.r)
}