add store labels (#3784)

* add store labels * Update pkg/apis/pingcap/v1alpha1/types.go Co-authored-by: Lonng <heng@lonng.org> * Update pkg/manager/member/tikv_member_manager.go Co-authored-by: Lonng <heng@lonng.org> * Update pkg/apis/pingcap/v1alpha1/types.go Co-authored-by: Lonng <heng@lonng.org> * fix var * fix * generate code Co-authored-by: Lonng <heng@lonng.org> Co-authored-by: DanielZhangQD <36026334+DanielZhangQD@users.noreply.github.com>
pingcap · Feb 25, 2021 · a3b0e6c · a3b0e6c
1 parent 64a6d40
commit a3b0e6c
Show file tree

Hide file tree

Showing 6 changed files with 42 additions and 3 deletions.
diff --git a/docs/api-references/docs.md b/docs/api-references/docs.md
@@ -18057,6 +18057,18 @@ Defaults to 10m</p>
 <p>StorageVolumes configure additional storage for TiKV pods.</p>
 </td>
 </tr>
+<tr>
+<td>
+<code>storeLabels</code></br>
+<em>
+[]string
+</em>
+</td>
+<td>
+<em>(Optional)</em>
+<p>StoreLabels configures additional labels for TiKV stores.</p>
+</td>
+</tr>
 </tbody>
 </table>
 <h3 id="tikvstatus">TiKVStatus</h3>

diff --git a/manifests/crd.yaml b/manifests/crd.yaml
@@ -12788,6 +12788,10 @@ spec:
                 storageVolumes:
                   items: {}
                   type: array
+                storeLabels:
+                  items:
+                    type: string
+                  type: array
                 terminationGracePeriodSeconds:
                   format: int64
                   type: integer

diff --git a/pkg/apis/pingcap/v1alpha1/openapi_generated.go b/pkg/apis/pingcap/v1alpha1/openapi_generated.go
diff --git a/pkg/apis/pingcap/v1alpha1/types.go b/pkg/apis/pingcap/v1alpha1/types.go
@@ -468,6 +468,10 @@ type TiKVSpec struct {
 	// StorageVolumes configure additional storage for TiKV pods.
 	// +optional
 	StorageVolumes []StorageVolume `json:"storageVolumes,omitempty"`
+
+	// StoreLabels configures additional labels for TiKV stores.
+	// +optional
+	StoreLabels []string `json:"storeLabels,omitempty"`
 }
 
 // TiFlashSpec contains details of TiFlash members

diff --git a/pkg/apis/pingcap/v1alpha1/zz_generated.deepcopy.go b/pkg/apis/pingcap/v1alpha1/zz_generated.deepcopy.go
diff --git a/pkg/manager/member/tikv_member_manager.go b/pkg/manager/member/tikv_member_manager.go
@@ -816,8 +816,8 @@ func (m *tikvMemberManager) setStoreLabelsForTiKV(tc *v1alpha1.TidbCluster) (int
 		return setCount, err
 	}
 
-	locationLabels := []string(config.Replication.LocationLabels)
-	if locationLabels == nil {
+	storeLabels := append(config.Replication.LocationLabels, tc.Spec.TiKV.StoreLabels...)
+	if storeLabels == nil {
 		return setCount, nil
 	}
 
@@ -843,7 +843,7 @@ func (m *tikvMemberManager) setStoreLabelsForTiKV(tc *v1alpha1.TidbCluster) (int
 		}
 
 		nodeName := pod.Spec.NodeName
-		ls, err := m.getNodeLabels(nodeName, locationLabels)
+		ls, err := m.getNodeLabels(nodeName, storeLabels)
 		if err != nil || len(ls) == 0 {
 			klog.Warningf("node: [%s] has no node labels, skipping set store labels for Pod: [%s/%s]", nodeName, ns, podName)
 			continue