add hub struct for connections

Author: Euler-B

main.go

@@ -55,5 +55,6 @@ func BindRoutes(s server.Server, r *mux.Router) {
 	r.HandleFunc("/posts/{id}", handlers.DeletePostHandler(s)).Methods(http.MethodDelete)
 	r.HandleFunc("/posts", handlers.ListPostHandler(s)).Methods(http.MethodGet)
 
-	r.HandleFunc("/ws", hub.HandleWebSocket(s)) 
+	go hub.Run()
+	r.HandleFunc("/ws", hub.HandleWebSocket) 
 }

websocket/client.go

@@ -2,7 +2,6 @@ package websocket
 
 import (
 	"github.com/gorilla/websocket"
-	"golang.org/x/text/message"
 )
 
 type Client struct {
@@ -23,7 +22,7 @@ func NewClient (hub *Hub, socket *websocket.Conn) *Client {
 func (c *Client) Write() {
 	for {
 		select {
-		case message, ok: <- c.outbound:
+		case message, ok := <- c.outbound:
 			if !ok {
 				c.socket.WriteMessage(websocket.CloseMessage, []byte{})
 				return

websocket/hub.go

@@ -38,4 +38,41 @@ func (hub *Hub) HandleWebSocket(w http.ResponseWriter, r *http.Request) {
 	hub.register <- client
 
 	go client.Write()
+}
+
+func (hub *Hub) Run() {
+	for {
+		select {
+		case client := <- hub.register:
+			hub.onConnect(client)
+		case client := <- hub.unregister:
+			hub.onDisconnect(client) 
+		}
+	}
+}
+
+func (hub *Hub) onConnect(client *Client) {
+	log.Println("Client Connected", client.socket.RemoteAddr())
+
+	hub.mutex.Lock()
+	defer hub.mutex.Unlock()
+	client.id = client.socket.RemoteAddr().String()
+	hub.clients = append(hub.clients, client)
+}
+
+func (hub *Hub) onDisconnect(client *Client) {
+	log.Println("Client Disconnected", client.socket.RemoteAddr())
+	client.socket.Close()
+	hub.mutex.Lock()
+	defer hub.mutex.Unlock()
+
+	i := -1
+	for j, c := range hub.clients {
+		if c.id == client.id {
+			i = j
+		}
+	}
+	copy(hub.clients[i:], hub.clients[i+1:])
+	hub.clients[len(hub.clients) - 1] = nil 
+	hub.clients = hub.clients[:len(hub.clients) - 1]
 }