implementing of websocket's(part one)

Author: Euler-B

main.go

@@ -9,6 +9,7 @@ import (
 	"github.com/Euler-B/API-REST_Go/handlers"
 	"github.com/Euler-B/API-REST_Go/middleware"
 	"github.com/Euler-B/API-REST_Go/server"
+	"github.com/Euler-B/API-REST_Go/websocket"
 	"github.com/gorilla/mux"
 	"github.com/joho/godotenv"
 )
@@ -38,6 +39,8 @@ func main() {
 }
 
 func BindRoutes(s server.Server, r *mux.Router) {
+	hub := websocket.NewHub()
+
 	r.Use(middleware.CheckAuthMiddleware(s))
 
 	r.HandleFunc("/", handlers.HomeHandler(s)).Methods(http.MethodGet)
@@ -51,4 +54,6 @@ func BindRoutes(s server.Server, r *mux.Router) {
 	r.HandleFunc("/posts/{id}", handlers.UpdatePostHandler(s)).Methods(http.MethodPut)
 	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)) 
 }

websocket/client.go

@@ -0,0 +1,34 @@
+package websocket
+
+import (
+	"github.com/gorilla/websocket"
+	"golang.org/x/text/message"
+)
+
+type Client struct {
+	hub *Hub
+	id string
+	socket *websocket.Conn
+	outbound chan []byte
+}
+
+func NewClient (hub *Hub, socket *websocket.Conn) *Client {
+	return &Client{
+		hub: hub,
+		socket: socket,
+		outbound: make(chan []byte),
+	}
+}
+
+func (c *Client) Write() {
+	for {
+		select {
+		case message, ok: <- c.outbound:
+			if !ok {
+				c.socket.WriteMessage(websocket.CloseMessage, []byte{})
+				return
+			}
+			c.socket.WriteMessage(websocket.TextMessage, message)
+			
+		}
+}

websocket/hub.go

@@ -0,0 +1,41 @@
+package websocket
+
+import (
+	"http"
+	"log"
+	"net/http"
+	"sync"
+
+	"github.com/gorilla/websocket"
+)
+
+var upgrader = websocket.Upgrader{
+	CheckOrigin: func(r *http.Request) bool { return true},
+}
+
+type Hub struct {
+	clients    []*Client
+	register   chan *Client
+	unregister chan *Client
+	mutex      *sync.Mutex
+}
+
+func NewHub() *Hub {
+	return &Hub{
+		clients:  make([]*Client, 0),
+		register: make(chan *Client),
+		mutex: &sync.Mutex{},
+	}
+}
+
+func (hub *Hub) HandleWebSocket(w http.ResponseWriter, r *http.Request) {
+	socket, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		log.Println(err)
+		http.Error(w, "Could not open websocket connection", http.StatusBadRequest)
+	}
+	client := NewClient(hub, socket)
+	hub.register <- client
+
+	go client.Write()
+}