From 9310bb5572392e411801b61078b6aabf58fa0b80 Mon Sep 17 00:00:00 2001
From: Craig <craigbett@gmail.com>
Date: Fri, 14 Feb 2020 03:52:14 +0000
Subject: [PATCH] add WebSocketHandler to WebSocket struct

---
 server/internal/websocket/socket.go    | 2 +-
 server/internal/websocket/websocket.go | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/server/internal/websocket/socket.go b/server/internal/websocket/socket.go
index 2439cd21..90dbbe77 100644
--- a/server/internal/websocket/socket.go
+++ b/server/internal/websocket/socket.go
@@ -10,7 +10,7 @@ import (
 
 type WebSocket struct {
 	id         string
-	ws         WebSocketHandler
+	ws         *WebSocketHandler
 	connection *websocket.Conn
 	mu         sync.Mutex
 }
diff --git a/server/internal/websocket/websocket.go b/server/internal/websocket/websocket.go
index 57cf34e9..ba12a655 100644
--- a/server/internal/websocket/websocket.go
+++ b/server/internal/websocket/websocket.go
@@ -142,6 +142,7 @@ func (ws *WebSocketHandler) Upgrade(w http.ResponseWriter, r *http.Request) erro
 
 	socket := &WebSocket{
 		id:         id,
+		ws:         ws,
 		connection: connection,
 	}