-
Notifications
You must be signed in to change notification settings - Fork 75
/
monitor.go
88 lines (68 loc) · 1.81 KB
/
monitor.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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package relay
import (
"net"
"github.com/hashicorp/memberlist"
"github.com/sonm-io/core/proto"
"github.com/sonm-io/core/util"
"github.com/sonm-io/core/util/xgrpc"
"go.uber.org/zap"
"golang.org/x/net/context"
"google.golang.org/grpc"
)
type monitor struct {
cfg MonitorConfig
certificate util.HitlessCertRotator
server *grpc.Server
listener net.Listener
cluster *memberlist.Memberlist
metrics *metrics
log *zap.Logger
}
func newMonitor(cfg MonitorConfig, cluster *memberlist.Memberlist, metrics *metrics, log *zap.Logger) (*monitor, error) {
certificate, TLSConfig, err := util.NewHitlessCertRotator(context.Background(), cfg.PrivateKey)
if err != nil {
return nil, err
}
credentials := util.NewTLS(TLSConfig)
server := xgrpc.NewServer(log,
xgrpc.Credentials(credentials),
xgrpc.DefaultTraceInterceptor(),
xgrpc.VerifyInterceptor(),
)
m := &monitor{
cfg: cfg,
certificate: certificate,
server: server,
cluster: cluster,
metrics: metrics,
log: log,
}
return m, nil
}
func (m *monitor) Cluster(ctx context.Context, request *sonm.Empty) (*sonm.RelayClusterReply, error) {
membersCluster := m.cluster.Members()
members := make([]string, 0, len(membersCluster))
for _, member := range membersCluster {
members = append(members, member.Address())
}
return &sonm.RelayClusterReply{
Members: members,
}, nil
}
func (m *monitor) Metrics(ctx context.Context, request *sonm.Empty) (*sonm.RelayMetrics, error) {
return m.metrics.Dump(), nil
}
func (m *monitor) Serve() error {
listener, err := net.Listen("tcp", m.cfg.Endpoint)
if err != nil {
return err
}
sonm.RegisterRelayServer(m.server, m)
m.listener = listener
return m.server.Serve(m.listener)
}
func (m *monitor) Close() error {
m.server.Stop()
m.certificate.Close()
return nil
}