update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package v4 
 
import ( 
    "encoding/json" 
 
    "github.com/golang/protobuf/proto" 
    "google.golang.org/protobuf/types/known/anypb" 
 
    "github.com/v2fly/v2ray-core/v5/app/observatory" 
    "github.com/v2fly/v2ray-core/v5/app/observatory/burst" 
    "github.com/v2fly/v2ray-core/v5/app/observatory/multiobservatory" 
    "github.com/v2fly/v2ray-core/v5/common/serial" 
    "github.com/v2fly/v2ray-core/v5/common/taggedfeatures" 
    "github.com/v2fly/v2ray-core/v5/infra/conf/cfgcommon/duration" 
    "github.com/v2fly/v2ray-core/v5/infra/conf/synthetic/router" 
) 
 
type ObservatoryConfig struct { 
    SubjectSelector []string          `json:"subjectSelector"` 
    ProbeURL        string            `json:"probeURL"` 
    ProbeInterval   duration.Duration `json:"probeInterval"` 
} 
 
func (o *ObservatoryConfig) Build() (proto.Message, error) { 
    return &observatory.Config{SubjectSelector: o.SubjectSelector, ProbeUrl: o.ProbeURL, ProbeInterval: int64(o.ProbeInterval)}, nil 
} 
 
type BurstObservatoryConfig struct { 
    SubjectSelector []string `json:"subjectSelector"` 
    // health check settings 
    HealthCheck *router.HealthCheckSettings `json:"pingConfig,omitempty"` 
} 
 
func (b BurstObservatoryConfig) Build() (proto.Message, error) { 
    result, err := b.HealthCheck.Build() 
    if err == nil { 
        return &burst.Config{SubjectSelector: b.SubjectSelector, PingConfig: result.(*burst.HealthPingConfig)}, nil 
    } 
    return nil, err 
} 
 
type MultiObservatoryItem struct { 
    MemberType string          `json:"type"` 
    Tag        string          `json:"tag"` 
    Value      json.RawMessage `json:"settings"` 
} 
 
type MultiObservatoryConfig struct { 
    Observers []MultiObservatoryItem `json:"observers"` 
} 
 
func (o *MultiObservatoryConfig) Build() (proto.Message, error) { 
    ret := &multiobservatory.Config{Holders: &taggedfeatures.Config{Features: make(map[string]*anypb.Any)}} 
    for _, v := range o.Observers { 
        switch v.MemberType { 
        case "burst": 
            var burstObservatoryConfig BurstObservatoryConfig 
            err := json.Unmarshal(v.Value, &burstObservatoryConfig) 
            if err != nil { 
                return nil, err 
            } 
            burstObservatoryConfigPb, err := burstObservatoryConfig.Build() 
            if err != nil { 
                return nil, err 
            } 
            ret.Holders.Features[v.Tag] = serial.ToTypedMessage(burstObservatoryConfigPb) 
        case "default": 
            fallthrough 
        default: 
            var observatoryConfig ObservatoryConfig 
            err := json.Unmarshal(v.Value, &observatoryConfig) 
            if err != nil { 
                return nil, err 
            } 
            observatoryConfigPb, err := observatoryConfig.Build() 
            if err != nil { 
                return nil, err 
            } 
            ret.Holders.Features[v.Tag] = serial.ToTypedMessage(observatoryConfigPb) 
        } 
    } 
    return ret, nil 
}