From 30648c3e0767d07bb1d728afb92ce67bedc4f20c Mon Sep 17 00:00:00 2001
From: Shelikhoo <xiaokangwang@outlook.com>
Date: Tue, 7 Sep 2021 11:42:45 +0100
Subject: [PATCH] added custom balancer settings processing logic

---
 app/router/config.go       | 34 ++++++++++++++++++++++++++++++++++
 infra/conf/v5cfg/common.go |  5 +++++
 2 files changed, 39 insertions(+)

diff --git a/app/router/config.go b/app/router/config.go
index 45eff96a7..f14560a89 100644
--- a/app/router/config.go
+++ b/app/router/config.go
@@ -1,11 +1,15 @@
 package router
 
 import (
+	"context"
+	"encoding/json"
+	"github.com/golang/protobuf/jsonpb"
 	"github.com/v2fly/v2ray-core/v4/app/router/routercommon"
 	"github.com/v2fly/v2ray-core/v4/common/net"
 	"github.com/v2fly/v2ray-core/v4/common/serial"
 	"github.com/v2fly/v2ray-core/v4/features/outbound"
 	"github.com/v2fly/v2ray-core/v4/features/routing"
+	"github.com/v2fly/v2ray-core/v4/infra/conf/v5cfg"
 )
 
 // CIDRList is an alias of []*CIDR to provide sort.Interface.
@@ -203,3 +207,33 @@ func (br *BalancingRule) Build(ohm outbound.Manager, dispatcher routing.Dispatch
 		return nil, newError("unrecognized balancer type")
 	}
 }
+
+func (br *BalancingRule) UnmarshalJSONPB(unmarshaler *jsonpb.Unmarshaler, bytes []byte) error {
+	type BalancingRuleStub struct {
+		Tag              string          `protobuf:"bytes,1,opt,name=tag,proto3" json:"tag,omitempty"`
+		OutboundSelector []string        `protobuf:"bytes,2,rep,name=outbound_selector,json=outboundSelector,proto3" json:"outbound_selector,omitempty"`
+		Strategy         string          `protobuf:"bytes,3,opt,name=strategy,proto3" json:"strategy,omitempty"`
+		StrategySettings json.RawMessage `protobuf:"bytes,4,opt,name=strategy_settings,json=strategySettings,proto3" json:"strategy_settings,omitempty"`
+		FallbackTag      string          `protobuf:"bytes,5,opt,name=fallback_tag,json=fallbackTag,proto3" json:"fallback_tag,omitempty"`
+	}
+
+	var stub BalancingRuleStub
+	if err := json.Unmarshal(bytes, &stub); err != nil {
+		return err
+	}
+	if stub.Strategy == "" {
+		stub.Strategy = "random"
+	}
+	settingsPack, err := v5cfg.LoadHeterogeneousConfigFromRawJson(context.TODO(), "balancer", stub.Strategy, stub.StrategySettings)
+	if err != nil {
+		return err
+	}
+	br.StrategySettings = serial.ToTypedMessage(settingsPack)
+
+	br.Tag = stub.Tag
+	br.Strategy = stub.Strategy
+	br.OutboundSelector = stub.OutboundSelector
+	br.FallbackTag = stub.FallbackTag
+
+	return nil
+}
diff --git a/infra/conf/v5cfg/common.go b/infra/conf/v5cfg/common.go
index 97ff4ca6a..b5567ccfd 100644
--- a/infra/conf/v5cfg/common.go
+++ b/infra/conf/v5cfg/common.go
@@ -14,3 +14,8 @@ func loadHeterogeneousConfigFromRawJson(interfaceType, name string, rawJson json
 	ctx := envctx.ContextWithEnvironment(context.TODO(), fsdef)
 	return registry.LoadImplementationByAlias(ctx, interfaceType, name, []byte(rawJson))
 }
+
+// LoadHeterogeneousConfigFromRawJson private API
+func LoadHeterogeneousConfigFromRawJson(ctx context.Context, interfaceType, name string, rawJson json.RawMessage) (proto.Message, error) {
+	return loadHeterogeneousConfigFromRawJson(interfaceType, name, rawJson)
+}