mirror of https://github.com/v2fly/v2ray-core.git
39 lines
1006 B
Go
39 lines
1006 B
Go
package router
|
|
|
|
import (
|
|
"github.com/v2fly/v2ray-core/v4/common/dice"
|
|
"github.com/v2fly/v2ray-core/v4/features/routing"
|
|
)
|
|
|
|
// RandomStrategy represents a random balancing strategy
|
|
type RandomStrategy struct{}
|
|
|
|
// GetInformation implements the routing.BalancingStrategy.
|
|
func (s *RandomStrategy) GetInformation(tags []string) *routing.StrategyInfo {
|
|
items := make([]*routing.OutboundInfo, 0)
|
|
for _, tag := range tags {
|
|
items = append(items, &routing.OutboundInfo{Tag: tag})
|
|
}
|
|
return &routing.StrategyInfo{
|
|
Settings: []string{"random"},
|
|
ValueTitles: nil,
|
|
Selects: items,
|
|
Others: nil,
|
|
}
|
|
}
|
|
|
|
// SelectAndPick implements the routing.BalancingStrategy.
|
|
func (s *RandomStrategy) SelectAndPick(candidates []string) string {
|
|
return s.Pick(candidates)
|
|
}
|
|
|
|
// Pick implements the routing.BalancingStrategy.
|
|
func (s *RandomStrategy) Pick(candidates []string) string {
|
|
count := len(candidates)
|
|
if count == 0 {
|
|
// goes to fallbackTag
|
|
return ""
|
|
}
|
|
return candidates[dice.Roll(count)]
|
|
}
|