This repository has been archived by the owner on Nov 23, 2022. It is now read-only.
/
ws.go
73 lines (66 loc) · 1.9 KB
/
ws.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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package pusher
import (
"fmt"
"github.com/everstake/teztracker/api/render"
"github.com/everstake/teztracker/models"
"github.com/everstake/teztracker/ws"
wsmodels "github.com/everstake/teztracker/ws/models"
"strings"
)
type WSPusher struct {
ws *ws.Hub
}
func NewWSPusher(ws *ws.Hub) *WSPusher {
return &WSPusher{
ws: ws,
}
}
func (p WSPusher) Push(event wsmodels.EventType, data interface{}) (err error) {
switch event {
case wsmodels.EventTypeBlock:
err = p.sendBlock(data)
case wsmodels.EventTypeOperation:
err = p.sendOperation(data)
case wsmodels.EventTypeAccountCreated:
err = p.sendAccount(data)
default:
return nil
}
if err != nil {
return fmt.Errorf("send[%s]: %s", event, err.Error())
}
return nil
}
func (p WSPusher) sendBlock(data interface{}) error {
block, ok := data.(models.Block)
if !ok {
return fmt.Errorf("wrong data")
}
apiBlock := render.Block(block)
p.ws.Broadcast(wsmodels.BasicMessage{Event: wsmodels.EventTypeBlock, Data: apiBlock})
return nil
}
func (p WSPusher) sendOperation(data interface{}) error {
op, ok := data.(models.Operation)
if !ok {
return fmt.Errorf("wrong data")
}
apiOperation := render.Operation(op, nil)
p.ws.Broadcast(wsmodels.BasicMessage{Event: wsmodels.EventTypeOperation, Data: apiOperation})
p.ws.Broadcast(wsmodels.BasicMessage{Event: wsmodels.EventType(fmt.Sprint(op.Kind.String, "s")), Data: apiOperation})
return nil
}
func (p WSPusher) sendAccount(data interface{}) error {
acc, ok := data.(models.AccountListView)
if !ok {
return fmt.Errorf("wrong data")
}
apiAccount := render.Account(acc)
accountType := "accounts"
if strings.Contains(acc.AccountID.String, models.ContractAccountPrefix) {
accountType = "contracts"
}
p.ws.Broadcast(wsmodels.BasicMessage{Event: wsmodels.EventTypeAccountCreated, Data: apiAccount})
p.ws.Broadcast(wsmodels.BasicMessage{Event: wsmodels.EventType(accountType), Data: apiAccount})
return nil
}