*: Auto refresh PD addrs for PDPlacementManager, PDLabelManager, …

…`TiFlashPDPlacementManager` (#33909) (#33913) close #33908
pingcap · Jun 17, 2022 · 7bb3edf · 7bb3edf
1 parent 0ec9c36
commit 7bb3edf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 13 deletions.
diff --git a/domain/infosync/info.go b/domain/infosync/info.go
@@ -177,11 +177,7 @@ func GlobalInfoSyncerInit(ctx context.Context, id string, serverIDGetter func()
 	if err != nil {
 		return nil, err
 	}
-	if etcdCli != nil {
-		is.labelRuleManager = initLabelRuleManager(etcdCli.Endpoints())
-	} else {
-		is.labelRuleManager = initLabelRuleManager([]string{})
-	}
+	is.labelRuleManager = initLabelRuleManager(etcdCli)
 	setGlobalInfoSyncer(is)
 	return is, nil
 }
@@ -208,11 +204,11 @@ func (is *InfoSyncer) GetSessionManager() util2.SessionManager {
 	return is.manager
 }
 
-func initLabelRuleManager(addrs []string) LabelRuleManager {
-	if len(addrs) == 0 {
+func initLabelRuleManager(etcdCli *clientv3.Client) LabelRuleManager {
+	if etcdCli == nil {
 		return &mockLabelManager{labelRules: map[string][]byte{}}
 	}
-	return &PDLabelManager{addrs: addrs}
+	return &PDLabelManager{etcdCli: etcdCli}
 }
 
 // GetServerInfo gets self server static information.

diff --git a/domain/infosync/label_manager.go b/domain/infosync/label_manager.go
@@ -23,6 +23,7 @@ import (
 
 	"github.com/pingcap/tidb/ddl/label"
 	"github.com/pingcap/tidb/util/pdapi"
+	"go.etcd.io/etcd/clientv3"
 )
 
 // LabelRuleManager manages label rules
@@ -35,7 +36,7 @@ type LabelRuleManager interface {
 
 // PDLabelManager manages rules with pd
 type PDLabelManager struct {
-	addrs []string
+	etcdCli *clientv3.Client
 }
 
 // PutLabelRule implements PutLabelRule
@@ -44,7 +45,7 @@ func (lm *PDLabelManager) PutLabelRule(ctx context.Context, rule *label.Rule) er
 	if err != nil {
 		return err
 	}
-	_, err = doRequest(ctx, lm.addrs, path.Join(pdapi.Config, "region-label", "rule"), "POST", bytes.NewReader(r))
+	_, err = doRequest(ctx, lm.etcdCli.Endpoints(), path.Join(pdapi.Config, "region-label", "rule"), "POST", bytes.NewReader(r))
 	return err
 }
 
@@ -55,14 +56,14 @@ func (lm *PDLabelManager) UpdateLabelRules(ctx context.Context, patch *label.Rul
 		return err
 	}
 
-	_, err = doRequest(ctx, lm.addrs, path.Join(pdapi.Config, "region-label", "rules"), "PATCH", bytes.NewReader(r))
+	_, err = doRequest(ctx, lm.etcdCli.Endpoints(), path.Join(pdapi.Config, "region-label", "rules"), "PATCH", bytes.NewReader(r))
 	return err
 }
 
 // GetAllLabelRules implements GetAllLabelRules
 func (lm *PDLabelManager) GetAllLabelRules(ctx context.Context) ([]*label.Rule, error) {
 	var rules []*label.Rule
-	res, err := doRequest(ctx, lm.addrs, path.Join(pdapi.Config, "region-label", "rules"), "GET", nil)
+	res, err := doRequest(ctx, lm.etcdCli.Endpoints(), path.Join(pdapi.Config, "region-label", "rules"), "GET", nil)
 
 	if err == nil && res != nil {
 		err = json.Unmarshal(res, &rules)
@@ -78,7 +79,7 @@ func (lm *PDLabelManager) GetLabelRules(ctx context.Context, ruleIDs []string) (
 	}
 
 	rules := []*label.Rule{}
-	res, err := doRequest(ctx, lm.addrs, path.Join(pdapi.Config, "region-label", "rules", "ids"), "GET", bytes.NewReader(ids))
+	res, err := doRequest(ctx, lm.etcdCli.Endpoints(), path.Join(pdapi.Config, "region-label", "rules", "ids"), "GET", bytes.NewReader(ids))
 
 	if err == nil && res != nil {
 		err = json.Unmarshal(res, &rules)