Add broadcaster to notify clients and message changes

main.go

@@ -51,3 +51,9 @@ func StartSever(port string, path string) {
 func GetClients() map[string]*websocket.Client {
 	return pool.GetClients()
 }
+
+func ListenClientAddChanges() chan websocket.Client {
+	c := make(chan websocket.Client)
+	websocket.CBR.AddWorker(c)
+	return c
+}

websocket/broadcast.go

@@ -0,0 +1,58 @@
+package websocket
+
+import "sync"
+
+var mutex1 sync.Mutex
+var mutex2 sync.Mutex
+
+type MessageBCastChannel struct {
+	clientList []chan Message
+}
+
+type ClientBCastChannel struct {
+	clientList []chan Client
+}
+
+func NewMessageBCastChannel(size int) *MessageBCastChannel {
+	return &MessageBCastChannel{clientList: make([]chan Message, size)}
+}
+
+func NewClientBCastChannel(size int) *ClientBCastChannel {
+	return &ClientBCastChannel{clientList: make([]chan Client, size)}
+}
+
+func (bc *MessageBCastChannel) AddWorker(c chan Message) {
+	mutex1.Lock()
+	bc.clientList = append(bc.clientList, c)
+	mutex1.Unlock()
+}
+
+func (bc *ClientBCastChannel) AddWorker(c chan Client) {
+	mutex2.Lock()
+	bc.clientList = append(bc.clientList, c)
+	mutex2.Unlock()
+}
+
+func (bc *MessageBCastChannel) BroadCast(data Message) {
+	mutex1.Lock()
+	for _, v := range bc.clientList {
+		if v != nil {
+			go func() {
+				v <- data
+			}()
+		}
+	}
+	mutex1.Unlock()
+}
+
+func (bc *ClientBCastChannel) BroadCast(data Client) {
+	mutex2.Lock()
+	for _, v := range bc.clientList {
+		if v != nil {
+			go func() {
+				v <- data
+			}()
+		}
+	}
+	mutex2.Unlock()
+}

websocket/pool.go

@@ -23,6 +23,10 @@ func NewPool() *Pool {
 
 }
 
+var CBR = NewClientBCastChannel(2)
+var CBU = NewClientBCastChannel(2)
+var CBM = NewMessageBCastChannel(2)
+
 func (pool *Pool) GetClients() map[string]*Client {
 	return pool.Clients
 }
@@ -35,6 +39,8 @@ func (pool *Pool) Start() {
 			pool.Clients[client.ID] = client
 			fmt.Println("New user", client.ID)
 			fmt.Println("Size of Connection Pool: ", len(pool.Clients))
+			CBR.BroadCast(*client)
+
 			for _, client2 := range pool.Clients {
 				if err := client2.Conn.WriteJSON(Message{Type: 2, Body: MessageBody{Message: client.ID}}); err != nil {
 					log.Fatal("Error on write")
@@ -46,6 +52,7 @@ func (pool *Pool) Start() {
 		case client := <-pool.Unregister:
 			delete(pool.Clients, client.ID)
 			fmt.Println("Size of Connection Pool: ", len(pool.Clients))
+			CBU.BroadCast(*client)
 			for _, client := range pool.Clients {
 				if err := client.Conn.WriteJSON(Message{Type: 3, Body: MessageBody{Message: client.ID}}); err != nil {
 					log.Fatal("Error on write")
@@ -55,6 +62,7 @@ func (pool *Pool) Start() {
 			break
 		case message := <-pool.Broadcast:
 
+			CBM.BroadCast(*message)
 			if message.Body.To != "" {
 				fmt.Println("Sending message to", message.Body.To)