/
rpc_node.go
96 lines (83 loc) · 2.25 KB
/
rpc_node.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
89
90
91
92
93
94
95
96
/*
* Xenon
*
* Copyright 2018 The Xenon Authors.
* Code is licensed under the GPLv3.
*
*/
package server
import (
"model"
)
type NodeRPC struct {
server *Server
}
func (s *Server) GetNodeRPC() *NodeRPC {
return &NodeRPC{s}
}
func (n *NodeRPC) AddNodes(req *model.NodeRPCRequest, rsp *model.NodeRPCResponse) error {
log := n.server.log
rsp.RetCode = model.OK
nodes := req.GetNodes()
log.Warning("server.rpc.node.add:%+v", req)
for _, node := range nodes {
if err := n.server.raft.AddPeer(node); err != nil {
rsp.RetCode = err.Error()
log.Error("rpc.add.peer[%v].error[%v]", node, err)
return nil
}
}
return nil
}
func (n *NodeRPC) AddIdleNodes(req *model.NodeRPCRequest, rsp *model.NodeRPCResponse) error {
log := n.server.log
rsp.RetCode = model.OK
nodes := req.GetNodes()
log.Warning("server.rpc.node.add:%+v", req)
for _, node := range nodes {
if err := n.server.raft.AddIdlePeer(node); err != nil {
rsp.RetCode = err.Error()
log.Error("rpc.add.idle.peer[%v].error[%v]", node, err)
return nil
}
}
return nil
}
func (n *NodeRPC) RemoveNodes(req *model.NodeRPCRequest, rsp *model.NodeRPCResponse) error {
log := n.server.log
rsp.RetCode = model.OK
nodes := req.GetNodes()
log.Warning("server.rpc.node.remove:%+v", req)
for _, node := range nodes {
if err := n.server.raft.RemovePeer(node); err != nil {
rsp.RetCode = err.Error()
log.Error("rpc.remove.peer[%v].error[%v]", node, err)
return nil
}
}
return nil
}
func (n *NodeRPC) RemoveIdleNodes(req *model.NodeRPCRequest, rsp *model.NodeRPCResponse) error {
log := n.server.log
rsp.RetCode = model.OK
nodes := req.GetNodes()
log.Warning("server.rpc.node.remove:%+v", req)
for _, node := range nodes {
if err := n.server.raft.RemoveIdlePeer(node); err != nil {
rsp.RetCode = err.Error()
log.Error("rpc.remove.idle.peer[%v].error[%v]", node, err)
return nil
}
}
return nil
}
func (n *NodeRPC) GetNodes(req *model.NodeRPCRequest, rsp *model.NodeRPCResponse) error {
rsp.RetCode = model.OK
rsp.Leader = n.server.raft.GetLeader()
rsp.ViewID = n.server.raft.GetVewiID()
rsp.EpochID = n.server.raft.GetEpochID()
rsp.State = n.server.raft.GetState().String()
nodes := n.server.raft.GetAllPeers()
rsp.Nodes = append(rsp.Nodes, nodes...)
return nil
}