Go Broadcast solution

broadcast.go

// Make it extend sync to handle multiple go routines
type Broadcaster struct {
	sync.Mutex
	clients map[string]chan models.CreatePostEvent
}

// Constructor
func NewBroadcaster() *Broadcaster {
	return &Broadcaster{
		clients: make(map[string]chan models.CreatePostEvent),
	}
}

func (b *Broadcaster) Broadcast(post models.CreatePostEvent) {
	b.Lock()
	defer b.Unlock()
	for _, client := range b.clients {
		client <- post
	}
}

// Function to add a client to the broadcaster
func (b *Broadcaster) AddClient(key string, client chan models.CreatePostEvent) {
	b.Lock()
	defer b.Unlock()
	b.clients[key] = client
}

// Function to remove a client from the broadcaster
func (b *Broadcaster) RemoveClient(key string) {
	b.Lock()
	defer b.Unlock()
	delete(b.clients, key)
}