websocket?

backend/handlers/handle_websocket.go

@@ -1,11 +1,8 @@
 package handlers
 
 import (
-	"time"
-
 	"adam-french.co.uk/backend/services"
 	"github.com/gin-gonic/gin"
-	"github.com/gorilla/websocket"
 )
 
 func (store *Store) ConnectWebSocket(ctx *gin.Context) {
@@ -14,9 +11,6 @@ func (store *Store) ConnectWebSocket(ctx *gin.Context) {
 		ctx.JSON(500, gin.H{"error": err.Error()})
 		return
 	}
-	defer conn.Close()
-	for {
-		conn.WriteMessage(websocket.TextMessage, []byte("Hello Websocket!"))
-		time.Sleep(time.Second)
-	}
+
+	services.HandleWebSocket(conn)
 }