From dd6630ed74f48d4964ff4a8774803212a04094b2 Mon Sep 17 00:00:00 2001
From: Mark4z <36187602+mark4z@users.noreply.github.com>
Date: Mon, 5 Feb 2024 17:07:27 +0800
Subject: [PATCH] specify loadbalance strategy on dubbo proxy (#613)

---
 pixiu/pkg/client/dubbo/config.go | 2 ++
 pixiu/pkg/client/dubbo/dubbo.go  | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/pixiu/pkg/client/dubbo/config.go b/pixiu/pkg/client/dubbo/config.go
index 85db58926..a5d498ff2 100644
--- a/pixiu/pkg/client/dubbo/config.go
+++ b/pixiu/pkg/client/dubbo/config.go
@@ -33,4 +33,6 @@ type DubboProxyConfig struct {
 	AutoResolve bool `yaml:"auto_resolve" json:"auto_resolve,omitempty"`
 	// Protoset path to load protoset files
 	Protoset []string `yaml:"protoset" json:"protoset,omitempty"`
+	// Load
+	LoadBalance string `yaml:"load_balance"  json:"load_balance,omitempty"`
 }
diff --git a/pixiu/pkg/client/dubbo/dubbo.go b/pixiu/pkg/client/dubbo/dubbo.go
index 4ea994167..0998e3d5b 100644
--- a/pixiu/pkg/client/dubbo/dubbo.go
+++ b/pixiu/pkg/client/dubbo/dubbo.go
@@ -27,7 +27,12 @@ import (
 
 import (
 	_ "dubbo.apache.org/dubbo-go/v3/cluster/cluster/failover"
+	_ "dubbo.apache.org/dubbo-go/v3/cluster/loadbalance/consistenthashing"
+	_ "dubbo.apache.org/dubbo-go/v3/cluster/loadbalance/leastactive"
+	_ "dubbo.apache.org/dubbo-go/v3/cluster/loadbalance/p2c"
 	_ "dubbo.apache.org/dubbo-go/v3/cluster/loadbalance/random"
+	_ "dubbo.apache.org/dubbo-go/v3/cluster/loadbalance/ringhash"
+	_ "dubbo.apache.org/dubbo-go/v3/cluster/loadbalance/roundrobin"
 	"dubbo.apache.org/dubbo-go/v3/common/constant"
 	_ "dubbo.apache.org/dubbo-go/v3/common/proxy/proxy_factory"
 	dg "dubbo.apache.org/dubbo-go/v3/config"
@@ -313,6 +318,7 @@ func (dc *Client) create(key string, irequest fc.IntegrationRequest) *generic.Ge
 		Generic:       "true",
 		Version:       irequest.DubboBackendConfig.Version,
 		Group:         irequest.Group,
+		Loadbalance:   dc.dubboProxyConfig.LoadBalance,
 	}
 
 	if len(irequest.DubboBackendConfig.Retries) == 0 {