Hub implementation in Go

hub.go

type Subscriber struct {
	send chan string
}

type Hub struct {
	clients    map[*Subscriber]bool
	broadcast  chan string
	register   chan *Subscriber
	unregister chan *Subscriber
}

func NewHub() *Hub {
	return &Hub{
		clients:    make(map[*Subscriber]bool),
		broadcast:  make(chan string),
		register:   make(chan *Subscriber),
		unregister: make(chan *Subscriber),
	}
}

func (h *Hub) Run() {
	for {
		select {
		case client := <-h.register:
			h.clients[client] = true
		case client := <-h.unregister:
			delete(h.clients, client)
			close(client.send)
		case msg := <-h.broadcast:
			for client := range h.clients {
				select {
				case client.send <- msg:
				default:
					close(client.send)
					delete(h.clients, client)
				}
			}
		}
	}
}