ginuerzh
effba3b690
Add godoc comments for all 62 exported symbols across 20 files in config/parsing/: 23 MDKey constants, 35 Parse*/List/Default* functions, 3 TLS helpers, 1 Args struct with 7 fields, and the package doc.
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package selector
|
|
|
|
import (
|
|
"github.com/go-gost/core/chain"
|
|
"github.com/go-gost/core/selector"
|
|
"github.com/go-gost/x/config"
|
|
xs "github.com/go-gost/x/selector"
|
|
)
|
|
|
|
// ParseChainSelector creates a chain-level selector from a SelectorConfig. If
|
|
// cfg is nil it returns nil. Strategy defaults to round-robin when unrecognized
|
|
// or unset.
|
|
func ParseChainSelector(cfg *config.SelectorConfig) selector.Selector[chain.Chainer] {
|
|
if cfg == nil {
|
|
return nil
|
|
}
|
|
|
|
var strategy selector.Strategy[chain.Chainer]
|
|
switch cfg.Strategy {
|
|
case "round", "rr":
|
|
strategy = xs.RoundRobinStrategy[chain.Chainer]()
|
|
case "random", "rand":
|
|
strategy = xs.RandomStrategy[chain.Chainer]()
|
|
case "fifo", "ha":
|
|
strategy = xs.FIFOStrategy[chain.Chainer]()
|
|
case "hash":
|
|
strategy = xs.HashStrategy[chain.Chainer]()
|
|
default:
|
|
strategy = xs.RoundRobinStrategy[chain.Chainer]()
|
|
}
|
|
return xs.NewSelector(
|
|
strategy,
|
|
xs.FailFilter[chain.Chainer](cfg.MaxFails, cfg.FailTimeout),
|
|
xs.BackupFilter[chain.Chainer](),
|
|
)
|
|
}
|
|
|
|
// ParseNodeSelector creates a node-level selector from a SelectorConfig. If
|
|
// cfg is nil it returns nil. Supported strategies: round/rr, random/rand,
|
|
// fifo/ha, hash, and parallel. Defaults to round-robin.
|
|
func ParseNodeSelector(cfg *config.SelectorConfig) selector.Selector[*chain.Node] {
|
|
if cfg == nil {
|
|
return nil
|
|
}
|
|
|
|
var strategy selector.Strategy[*chain.Node]
|
|
switch cfg.Strategy {
|
|
case "round", "rr":
|
|
strategy = xs.RoundRobinStrategy[*chain.Node]()
|
|
case "random", "rand":
|
|
strategy = xs.RandomStrategy[*chain.Node]()
|
|
case "fifo", "ha":
|
|
strategy = xs.FIFOStrategy[*chain.Node]()
|
|
case "hash":
|
|
strategy = xs.HashStrategy[*chain.Node]()
|
|
case "parallel":
|
|
strategy = xs.ParallelStrategy[*chain.Node]()
|
|
default:
|
|
strategy = xs.RoundRobinStrategy[*chain.Node]()
|
|
}
|
|
|
|
return xs.NewSelector(
|
|
strategy,
|
|
xs.FailFilter[*chain.Node](cfg.MaxFails, cfg.FailTimeout),
|
|
xs.BackupFilter[*chain.Node](),
|
|
)
|
|
}
|
|
|
|
// DefaultNodeSelector returns a node selector using round-robin strategy with
|
|
// default fail filter settings.
|
|
func DefaultNodeSelector() selector.Selector[*chain.Node] {
|
|
return xs.NewSelector(
|
|
xs.RoundRobinStrategy[*chain.Node](),
|
|
xs.FailFilter[*chain.Node](xs.DefaultMaxFails, xs.DefaultFailTimeout),
|
|
xs.BackupFilter[*chain.Node](),
|
|
)
|
|
}
|
|
|
|
// DefaultChainSelector returns a chain selector using round-robin strategy
|
|
// with default fail filter settings.
|
|
func DefaultChainSelector() selector.Selector[chain.Chainer] {
|
|
return xs.NewSelector(
|
|
xs.RoundRobinStrategy[chain.Chainer](),
|
|
xs.FailFilter[chain.Chainer](xs.DefaultMaxFails, xs.DefaultFailTimeout),
|
|
xs.BackupFilter[chain.Chainer](),
|
|
)
|
|
}
|