48 lines
820 B
Go
48 lines
820 B
Go
package chain
|
|
|
|
type Chainable interface {
|
|
WithChain(chain *Chain)
|
|
}
|
|
|
|
type Chain struct {
|
|
groups []*NodeGroup
|
|
}
|
|
|
|
func (c *Chain) AddNodeGroup(group *NodeGroup) {
|
|
c.groups = append(c.groups, group)
|
|
}
|
|
|
|
func (c *Chain) GetRoute() (r *Route) {
|
|
return c.GetRouteFor("tcp", "")
|
|
}
|
|
|
|
func (c *Chain) GetRouteFor(network, address string) (r *Route) {
|
|
if c == nil || len(c.groups) == 0 {
|
|
return
|
|
}
|
|
|
|
r = &Route{}
|
|
for _, group := range c.groups {
|
|
node := group.Next()
|
|
if node == nil {
|
|
return
|
|
}
|
|
if node.bypass != nil && node.bypass.Contains(address) {
|
|
break
|
|
}
|
|
|
|
if node.transport.IsMultiplex() {
|
|
tr := node.transport.Copy().WithRoute(r)
|
|
node = node.Copy().WithTransport(tr)
|
|
r = &Route{}
|
|
}
|
|
|
|
r.AddNode(node)
|
|
}
|
|
return r
|
|
}
|
|
|
|
func (c *Chain) IsEmpty() bool {
|
|
return c == nil || len(c.groups) == 0
|
|
}
|