2021-08-21 01:20:40 -04:00
|
|
|
//go:build !confonly
|
2019-02-01 14:08:21 -05:00
|
|
|
// +build !confonly
|
|
|
|
|
2018-11-07 15:08:20 -05:00
|
|
|
package router
|
|
|
|
|
|
|
|
import (
|
2021-04-08 15:55:25 -04:00
|
|
|
"context"
|
2021-01-29 19:31:11 -05:00
|
|
|
"github.com/v2fly/v2ray-core/v4/features/routing"
|
2021-04-08 15:55:25 -04:00
|
|
|
|
|
|
|
"github.com/v2fly/v2ray-core/v4/features/extension"
|
2021-02-16 15:31:50 -05:00
|
|
|
"github.com/v2fly/v2ray-core/v4/features/outbound"
|
2018-11-07 15:08:20 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
type BalancingStrategy interface {
|
|
|
|
PickOutbound([]string) string
|
|
|
|
}
|
|
|
|
|
|
|
|
type Balancer struct {
|
2021-01-29 19:31:11 -05:00
|
|
|
selectors []string
|
|
|
|
strategy routing.BalancingStrategy
|
|
|
|
ohm outbound.Manager
|
|
|
|
fallbackTag string
|
|
|
|
|
2021-02-21 10:02:42 -05:00
|
|
|
override override
|
2018-11-07 15:08:20 -05:00
|
|
|
}
|
|
|
|
|
2021-01-29 19:31:11 -05:00
|
|
|
// PickOutbound picks the tag of a outbound
|
2018-11-07 15:08:20 -05:00
|
|
|
func (b *Balancer) PickOutbound() (string, error) {
|
2021-01-29 19:31:11 -05:00
|
|
|
candidates, err := b.SelectOutbounds()
|
|
|
|
if err != nil {
|
|
|
|
if b.fallbackTag != "" {
|
|
|
|
newError("fallback to [", b.fallbackTag, "], due to error: ", err).AtInfo().WriteToLog()
|
|
|
|
return b.fallbackTag, nil
|
|
|
|
}
|
|
|
|
return "", err
|
2018-11-07 15:08:20 -05:00
|
|
|
}
|
2021-01-29 19:31:11 -05:00
|
|
|
var tag string
|
|
|
|
if o := b.override.Get(); o != nil {
|
|
|
|
tag = b.strategy.Pick(o.selects)
|
|
|
|
} else {
|
|
|
|
tag = b.strategy.SelectAndPick(candidates)
|
2018-11-07 15:08:20 -05:00
|
|
|
}
|
2019-06-14 09:47:28 -04:00
|
|
|
if tag == "" {
|
2021-01-29 19:31:11 -05:00
|
|
|
if b.fallbackTag != "" {
|
|
|
|
newError("fallback to [", b.fallbackTag, "], due to empty tag returned").AtInfo().WriteToLog()
|
|
|
|
return b.fallbackTag, nil
|
|
|
|
}
|
|
|
|
// will use default handler
|
2018-11-07 15:08:20 -05:00
|
|
|
return "", newError("balancing strategy returns empty tag")
|
|
|
|
}
|
|
|
|
return tag, nil
|
|
|
|
}
|
2021-05-19 17:28:52 -04:00
|
|
|
|
2021-04-08 15:55:25 -04:00
|
|
|
func (b *Balancer) InjectContext(ctx context.Context) {
|
|
|
|
if contextReceiver, ok := b.strategy.(extension.ContextReceiver); ok {
|
|
|
|
contextReceiver.InjectContext(ctx)
|
|
|
|
}
|
|
|
|
}
|
2021-01-29 19:31:11 -05:00
|
|
|
|
|
|
|
// SelectOutbounds select outbounds with selectors of the Balancer
|
|
|
|
func (b *Balancer) SelectOutbounds() ([]string, error) {
|
|
|
|
hs, ok := b.ohm.(outbound.HandlerSelector)
|
|
|
|
if !ok {
|
|
|
|
return nil, newError("outbound.Manager is not a HandlerSelector")
|
|
|
|
}
|
|
|
|
tags := hs.Select(b.selectors)
|
|
|
|
return tags, nil
|
|
|
|
}
|