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)
 }

backend/models/models.go

@@ -26,3 +26,12 @@ type Post struct {
 	Author    *User          `gorm:"foreignKey:AuthorID" json:"author"`
 	Content   string         `json:"content"`
 }
+
+type Message struct {
+	ID        uint           `gorm:"primarykey" json:"id"`
+	Content   string         `json:"text"`
+	AuthorID  uint           `json:"-"`
+	Author    *User          `gorm:"foreignKey:AuthorID" json:"author"`
+	CreatedAt time.Time      `json:"createdAt"`
+	DeletedAt gorm.DeletedAt `gorm:"index" json:"deletedAt"`
+}

backend/services/websocket.go

@@ -1,6 +1,11 @@
 package services
 
 import (
+	"sync"
+	"time"
+
+	"adam-french.co.uk/backend/models"
+
 	"github.com/gorilla/websocket"
 )
 
@@ -8,3 +13,52 @@ var Upgrader = websocket.Upgrader{
 	ReadBufferSize:  1024,
 	WriteBufferSize: 1024,
 }
+
+var (
+	clients  = make(map[*websocket.Conn]bool)
+	messages = make([]models.Message, 0)
+	mu       sync.Mutex
+)
+
+func HandleWebSocket(conn *websocket.Conn) {
+	defer conn.Close()
+
+	mu.Lock()
+	clients[conn] = true
+
+	// Send existing message history to new client
+	for _, msg := range messages {
+		if err := conn.WriteJSON(msg); err != nil {
+			mu.Unlock()
+			return
+		}
+	}
+	mu.Unlock()
+
+	// Read loop
+	for {
+		var incoming models.Message
+		if err := conn.ReadJSON(&incoming); err != nil {
+			break // client disconnected
+		}
+
+		incoming.CreatedAt = time.Now()
+
+		// Store and broadcast
+		mu.Lock()
+		messages = append(messages, incoming)
+
+		for client := range clients {
+			if err := client.WriteJSON(incoming); err != nil {
+				client.Close()
+				delete(clients, client)
+			}
+		}
+		mu.Unlock()
+	}
+
+	// Cleanup on disconnect
+	mu.Lock()
+	delete(clients, conn)
+	mu.Unlock()
+}