Clash.Meta/adapter/outboundgroup/fallback.go

141 lines
3.3 KiB
Go
Raw Normal View History

2019-12-08 12:17:24 +08:00
package outboundgroup
import (
"context"
"encoding/json"
2021-12-03 14:35:21 +08:00
"github.com/Dreamacro/clash/log"
"go.uber.org/atomic"
"time"
2019-12-08 12:17:24 +08:00
2021-06-10 14:05:56 +08:00
"github.com/Dreamacro/clash/adapter/outbound"
"github.com/Dreamacro/clash/component/dialer"
2019-12-08 12:17:24 +08:00
C "github.com/Dreamacro/clash/constant"
"github.com/Dreamacro/clash/constant/provider"
2019-12-08 12:17:24 +08:00
)
type Fallback struct {
*GroupBase
2021-12-03 14:35:21 +08:00
disableUDP bool
failedTimes *atomic.Int32
failedTime *atomic.Int64
2019-12-08 12:17:24 +08:00
}
func (f *Fallback) Now() string {
proxy := f.findAliveProxy(false)
2019-12-08 12:17:24 +08:00
return proxy.Name()
}
2021-04-29 11:23:14 +08:00
// DialContext implements C.ProxyAdapter
func (f *Fallback) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.Conn, error) {
proxy := f.findAliveProxy(true)
c, err := proxy.DialContext(ctx, metadata, f.Base.DialOptions(opts...)...)
2019-12-08 12:17:24 +08:00
if err == nil {
c.AppendToChains(f)
f.failedTimes.Store(-1)
f.failedTime.Store(-1)
2021-12-03 14:35:21 +08:00
} else {
f.onDialFailed()
2019-12-08 12:17:24 +08:00
}
2019-12-08 12:17:24 +08:00
return c, err
}
// ListenPacketContext implements C.ProxyAdapter
func (f *Fallback) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
proxy := f.findAliveProxy(true)
pc, err := proxy.ListenPacketContext(ctx, metadata, f.Base.DialOptions(opts...)...)
2019-12-08 12:17:24 +08:00
if err == nil {
pc.AppendToChains(f)
f.failedTimes.Store(-1)
f.failedTime.Store(-1)
2021-12-03 14:35:21 +08:00
} else {
f.onDialFailed()
2019-12-08 12:17:24 +08:00
}
2020-01-31 14:43:54 +08:00
return pc, err
2019-12-08 12:17:24 +08:00
}
2021-12-03 14:35:21 +08:00
func (f *Fallback) onDialFailed() {
if f.failedTime.Load() == -1 {
log.Warnln("%s first failed", f.Name())
now := time.Now().UnixMilli()
f.failedTime.Store(now)
f.failedTimes.Store(1)
} else {
2022-01-08 09:23:49 +08:00
if f.failedTime.Load()-time.Now().UnixMilli() > 5*time.Second.Milliseconds() {
2021-12-03 14:35:21 +08:00
f.failedTimes.Store(-1)
f.failedTime.Store(-1)
} else {
failedCount := f.failedTimes.Inc()
2021-12-03 14:35:21 +08:00
log.Warnln("%s failed count: %d", f.Name(), failedCount)
if failedCount >= 5 {
log.Warnln("because %s failed multiple times, active health check", f.Name())
2021-12-03 14:35:21 +08:00
for _, proxyProvider := range f.providers {
go proxyProvider.HealthCheck()
}
2022-01-08 09:23:49 +08:00
f.failedTimes.Store(-1)
f.failedTime.Store(-1)
2021-12-03 14:35:21 +08:00
}
}
}
}
2021-04-29 11:23:14 +08:00
// SupportUDP implements C.ProxyAdapter
2019-12-08 12:17:24 +08:00
func (f *Fallback) SupportUDP() bool {
if f.disableUDP {
return false
}
proxy := f.findAliveProxy(false)
2019-12-08 12:17:24 +08:00
return proxy.SupportUDP()
}
2021-04-29 11:23:14 +08:00
// MarshalJSON implements C.ProxyAdapter
2019-12-08 12:17:24 +08:00
func (f *Fallback) MarshalJSON() ([]byte, error) {
2022-03-23 13:48:21 +08:00
all := []string{}
for _, proxy := range f.GetProxies(false) {
2019-12-08 12:17:24 +08:00
all = append(all, proxy.Name())
}
2022-03-16 12:10:13 +08:00
return json.Marshal(map[string]any{
2019-12-08 12:17:24 +08:00
"type": f.Type().String(),
"now": f.Now(),
"all": all,
})
}
2021-04-29 11:23:14 +08:00
// Unwrap implements C.ProxyAdapter
func (f *Fallback) Unwrap(metadata *C.Metadata) C.Proxy {
proxy := f.findAliveProxy(true)
return proxy
}
func (f *Fallback) findAliveProxy(touch bool) C.Proxy {
proxies := f.GetProxies(touch)
for _, proxy := range proxies {
if proxy.Alive() {
return proxy
2019-12-08 12:17:24 +08:00
}
}
return proxies[0]
2019-12-08 12:17:24 +08:00
}
func NewFallback(option *GroupCommonOption, providers []provider.ProxyProvider) *Fallback {
2019-12-08 12:17:24 +08:00
return &Fallback{
GroupBase: NewGroupBase(GroupBaseOption{
outbound.BaseOption{
Name: option.Name,
Type: C.Fallback,
Interface: option.Interface,
RoutingMark: option.RoutingMark,
},
option.Filter,
providers,
}),
disableUDP: option.DisableUDP,
failedTimes: atomic.NewInt32(-1),
failedTime: atomic.NewInt64(-1),
2019-12-08 12:17:24 +08:00
}
}