60 lines
940 B
Go
60 lines
940 B
Go
package chain
|
|
|
|
type Node struct {
|
|
name string
|
|
addr string
|
|
transport *Transport
|
|
}
|
|
|
|
func NewNode(name, addr string) *Node {
|
|
return &Node{
|
|
name: name,
|
|
addr: addr,
|
|
}
|
|
}
|
|
|
|
func (node *Node) Name() string {
|
|
return node.name
|
|
}
|
|
|
|
func (node *Node) Addr() string {
|
|
return node.addr
|
|
}
|
|
|
|
func (node *Node) Transport() *Transport {
|
|
return node.transport
|
|
}
|
|
|
|
func (node *Node) WithTransport(tr *Transport) *Node {
|
|
node.transport = tr
|
|
return node
|
|
}
|
|
|
|
type NodeGroup struct {
|
|
nodes []*Node
|
|
selector Selector
|
|
}
|
|
|
|
func NewNodeGroup(nodes ...*Node) *NodeGroup {
|
|
return &NodeGroup{
|
|
nodes: nodes,
|
|
}
|
|
}
|
|
|
|
func (g *NodeGroup) AddNode(node *Node) {
|
|
g.nodes = append(g.nodes, node)
|
|
}
|
|
|
|
func (g *NodeGroup) WithSelector(selector Selector) {
|
|
g.selector = selector
|
|
}
|
|
|
|
func (g *NodeGroup) Next() *Node {
|
|
selector := g.selector
|
|
if selector == nil {
|
|
// selector = defaultSelector
|
|
return g.nodes[0]
|
|
}
|
|
return selector.Select(g.nodes...)
|
|
}
|