From 402b8f1a0877abcc073bedd69e472a19902a42a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Miroslav=20=C5=A0ediv=C3=BD?= Date: Sun, 29 Aug 2021 23:01:51 +0200 Subject: [PATCH] rename WebSocket handler. --- internal/types/websocket.go | 4 ++-- internal/websocket/manager.go | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/internal/types/websocket.go b/internal/types/websocket.go index a6b9cf5a..2eea24aa 100644 --- a/internal/types/websocket.go +++ b/internal/types/websocket.go @@ -10,7 +10,7 @@ type WebSocketMessage struct { Payload json.RawMessage `json:"payload"` } -type HandlerFunction func(Session, WebSocketMessage) bool +type WebSocketHandler func(Session, WebSocketMessage) bool type CheckOrigin func(r *http.Request) bool @@ -22,6 +22,6 @@ type WebSocketPeer interface { type WebSocketManager interface { Start() Shutdown() error - AddHandler(handler HandlerFunction) + AddHandler(handler WebSocketHandler) Upgrade(w http.ResponseWriter, r *http.Request, checkOrigin CheckOrigin) } diff --git a/internal/websocket/manager.go b/internal/websocket/manager.go index 9983d1b4..93e30719 100644 --- a/internal/websocket/manager.go +++ b/internal/websocket/manager.go @@ -28,7 +28,7 @@ func New( sessions: sessions, desktop: desktop, handler: handler.New(sessions, desktop, capture, webrtc), - handlers: []types.HandlerFunction{}, + handlers: []types.WebSocketHandler{}, } } @@ -40,7 +40,7 @@ type WebSocketManagerCtx struct { sessions types.SessionManager desktop types.DesktopManager handler *handler.MessageHandlerCtx - handlers []types.HandlerFunction + handlers []types.WebSocketHandler } func (manager *WebSocketManagerCtx) Start() { @@ -156,7 +156,7 @@ func (manager *WebSocketManagerCtx) Shutdown() error { return nil } -func (manager *WebSocketManagerCtx) AddHandler(handler types.HandlerFunction) { +func (manager *WebSocketManagerCtx) AddHandler(handler types.WebSocketHandler) { manager.handlers = append(manager.handlers, handler) }