client_manager.go

package routingtable

// ClientManager manages clients of routing tables (observer pattern)
type ClientManager struct {
	clients      map[RouteTableClient]struct{} // Ensures a client registers at most once
	routingTable *RoutingTable
}

// Register registers a client for updates
func (c *ClientManager) Register(client RouteTableClient) {
	if c.clients == nil {
		c.clients = make(map[RouteTableClient]struct{}, 0)
	}
	c.clients[client] = struct{}{}
	//c.routingTable.updateNewClient(client)
}

// Unregister unregisters a client
func (c *ClientManager) Unregister(client RouteTableClient) {
	if _, ok := c.clients[client]; !ok {
		return
	}
	delete(c.clients, client)
}

// Clients returns a list of registered clients
func (c *ClientManager) Clients() (ret []RouteTableClient) {
	for rtc := range c.clients {
		ret = append(ret, rtc)
	}

	return ret
}