/
portfolio_hub.go
55 lines (47 loc) · 1.38 KB
/
portfolio_hub.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package websocket
import (
"github.com/jeremyhahn/tradebot/common"
logging "github.com/op/go-logging"
)
type PortfolioHub struct {
logger *logging.Logger
clients map[*PortfolioClient]bool
broadcast chan common.Portfolio
register chan *PortfolioClient
unregister chan *PortfolioClient
}
func NewPortfolioHub(logger *logging.Logger) *PortfolioHub {
return &PortfolioHub{
broadcast: make(chan common.Portfolio),
register: make(chan *PortfolioClient),
unregister: make(chan *PortfolioClient),
clients: make(map[*PortfolioClient]bool),
logger: logger}
}
func (h *PortfolioHub) Run() {
for {
h.logger.Debug("Portfolio hub running...")
select {
case client := <-h.register:
client.ctx.GetLogger().Debugf("[PortfolioHub.run] Registering new client: %s", client.ctx.GetUser().GetUsername())
h.clients[client] = true
case client := <-h.unregister:
if _, ok := h.clients[client]; ok {
h.logger.Debugf("[PortfolioHub.run] Unregistering client: %s", client.ctx.GetUser().GetUsername())
client.disconnect()
delete(h.clients, client)
close(client.send)
}
case portfolio := <-h.broadcast:
for client := range h.clients {
h.logger.Debugf("[PortfolioHub.run] Broadcasting portfolio: %+v\n", portfolio)
select {
case client.send <- portfolio:
default:
close(client.send)
delete(h.clients, client)
}
}
}
}
}