1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-07-26 11:44:22 -04:00

add V4 json support for observer tag config

This commit is contained in:
Shelikhoo 2021-06-19 11:15:48 +01:00
parent e6f5db1c93
commit 2dc539cc70
No known key found for this signature in database
GPG Key ID: C4D5E79D22B25316
3 changed files with 68 additions and 0 deletions

View File

@ -1,8 +1,11 @@
package conf package conf
import ( import (
"encoding/json"
"github.com/golang/protobuf/proto" "github.com/golang/protobuf/proto"
"github.com/v2fly/v2ray-core/v4/app/observatory/burst" "github.com/v2fly/v2ray-core/v4/app/observatory/burst"
"github.com/v2fly/v2ray-core/v4/app/observatory/multiObservatory"
"github.com/v2fly/v2ray-core/v4/common/serial"
"github.com/v2fly/v2ray-core/v4/app/observatory" "github.com/v2fly/v2ray-core/v4/app/observatory"
"github.com/v2fly/v2ray-core/v4/infra/conf/cfgcommon/duration" "github.com/v2fly/v2ray-core/v4/infra/conf/cfgcommon/duration"
@ -31,3 +34,47 @@ func (b BurstObservatoryConfig) Build() (proto.Message, error) {
return nil, err 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{}
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)
break
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
}

View File

@ -17,6 +17,7 @@ var (
strategyConfigLoader = NewJSONConfigLoader(ConfigCreatorCache{ strategyConfigLoader = NewJSONConfigLoader(ConfigCreatorCache{
strategyRandom: func() interface{} { return new(strategyEmptyConfig) }, strategyRandom: func() interface{} { return new(strategyEmptyConfig) },
strategyLeastLoad: func() interface{} { return new(strategyLeastLoadConfig) }, strategyLeastLoad: func() interface{} { return new(strategyLeastLoadConfig) },
strategyLeastPing: func() interface{} { return new(strategyLeastPingConfig) },
}, "type", "settings") }, "type", "settings")
) )
@ -38,6 +39,8 @@ type strategyLeastLoadConfig struct {
MaxRTT Duration `json:"maxRTT,omitempty"` MaxRTT Duration `json:"maxRTT,omitempty"`
// acceptable failure rate // acceptable failure rate
Tolerance float64 `json:"tolerance,omitempty"` Tolerance float64 `json:"tolerance,omitempty"`
ObserverTag string `json:"observerTag,omitempty"`
} }
// healthCheckSettings holds settings for health Checker // healthCheckSettings holds settings for health Checker
@ -64,6 +67,7 @@ func (v *strategyLeastLoadConfig) Build() (proto.Message, error) {
config := &router.StrategyLeastLoadConfig{} config := &router.StrategyLeastLoadConfig{}
config.Costs = v.Costs config.Costs = v.Costs
config.Tolerance = float32(v.Tolerance) config.Tolerance = float32(v.Tolerance)
config.ObserverTag = v.ObserverTag
if config.Tolerance < 0 { if config.Tolerance < 0 {
config.Tolerance = 0 config.Tolerance = 0
} }
@ -87,3 +91,11 @@ func (v *strategyLeastLoadConfig) Build() (proto.Message, error) {
} }
return config, nil return config, nil
} }
type strategyLeastPingConfig struct {
ObserverTag string `json:"observerTag,omitempty"`
}
func (s strategyLeastPingConfig) Build() (proto.Message, error) {
return &router.StrategyLeastPingConfig{ObserverTag: s.ObserverTag}, nil
}

View File

@ -352,6 +352,7 @@ type Config struct {
BrowserForwarder *BrowserForwarderConfig `json:"browserForwarder"` BrowserForwarder *BrowserForwarderConfig `json:"browserForwarder"`
Observatory *ObservatoryConfig `json:"observatory"` Observatory *ObservatoryConfig `json:"observatory"`
BurstObservatory *BurstObservatoryConfig `json:"burstObservatory"` BurstObservatory *BurstObservatoryConfig `json:"burstObservatory"`
MultiObservatory *MultiObservatoryConfig `json:"multiObservatory"`
Services map[string]*json.RawMessage `json:"services"` Services map[string]*json.RawMessage `json:"services"`
} }
@ -500,6 +501,14 @@ func (c *Config) Build() (*core.Config, error) {
config.App = append(config.App, serial.ToTypedMessage(r)) config.App = append(config.App, serial.ToTypedMessage(r))
} }
if c.MultiObservatory != nil {
r, err := c.MultiObservatory.Build()
if err != nil {
return nil, err
}
config.App = append(config.App, serial.ToTypedMessage(r))
}
// Load Additional Services that do not have a json translator // Load Additional Services that do not have a json translator
if msg, err := c.BuildServices(c.Services); err != nil { if msg, err := c.BuildServices(c.Services); err != nil {