feat(circuit): add circuit breaker init config

Signed-off-by: cbluebird <[email protected]>
zjutjh · Jan 24, 2025 · b9e317e · b9e317e
1 parent b6bc133
commit b9e317e
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/app/utils/circuitBreaker/circuitBreaker.go b/app/utils/circuitBreaker/circuitBreaker.go
@@ -1,11 +1,28 @@
 package circuitBreaker
 
-import "wejh-go/config/api/funnelApi"
+import (
+	liveNessConfig "wejh-go/config/LiveNess"
+	"wejh-go/config/api/funnelApi"
+)
 
 var CB CircuitBreaker
 
 func init() {
-	//Todo: add config
+	lb := LoadBalance{
+		zfLB:    &randomLB{},
+		oauthLB: &randomLB{},
+	}
+	for _, config := range liveNessConfig.GetLoadBalanceConfig() {
+		if config.Type == funnelApi.Oauth {
+			lb.oauthLB.Add(config.Url)
+		} else if config.Type == funnelApi.ZF {
+			lb.zfLB.Add(config.Url)
+		}
+	}
+	CB = CircuitBreaker{
+		lb:       lb,
+		SnapShot: make(map[string]*apiSnapShot),
+	}
 }
 
 type CircuitBreaker struct {

diff --git a/config/LiveNess/config.go b/config/LiveNess/config.go
@@ -33,9 +33,9 @@ func GetLiveNessConfig() LiveNessProbeConfig {
 }
 
 type LoadBalanceConfig struct {
-	Name string
-	Url  string
-	Type funnelApi.LoginType
+	Name string              `json:"name"`
+	Url  string              `json:"url"`
+	Type funnelApi.LoginType `json:"type"`
 }
 
 func GetLoadBalanceConfig() []LoadBalanceConfig {