From 4d9d063aa664b57fbb6237c7c7012a95113227d2 Mon Sep 17 00:00:00 2001
From: Daishan <daishan@rancher.com>
Date: Thu, 30 Sep 2021 15:39:21 -0700
Subject: [PATCH] Add provider labels to v3 cluster

---
 .../dashboard/kubernetesprovider/kubernetesprovider.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pkg/controllers/dashboard/kubernetesprovider/kubernetesprovider.go b/pkg/controllers/dashboard/kubernetesprovider/kubernetesprovider.go
index 849396afac1..95772125cfb 100644
--- a/pkg/controllers/dashboard/kubernetesprovider/kubernetesprovider.go
+++ b/pkg/controllers/dashboard/kubernetesprovider/kubernetesprovider.go
@@ -10,6 +10,10 @@ import (
 	"k8s.io/client-go/kubernetes"
 )
 
+const (
+	providerKey = "provider.cattle.io"
+)
+
 type handler struct {
 	ctx                context.Context
 	clusters           v32.ClusterClient
@@ -32,7 +36,7 @@ func Register(ctx context.Context,
 }
 
 func (h *handler) OnChange(key string, cluster *v3.Cluster) (*v3.Cluster, error) {
-	if cluster == nil || cluster.Status.Provider != "" {
+	if cluster == nil || (cluster.Status.Provider != "" && cluster.Labels[providerKey] != "") {
 		// if cluster has windows enabled and provider is not rke.windows, continue to detect if it has windows nodes
 		if cluster == nil || !cluster.Spec.WindowsPreferedCluster || cluster.Status.Provider == "rke.windows" {
 			return cluster, nil
@@ -65,6 +69,10 @@ func (h *handler) OnChange(key string, cluster *v3.Cluster) (*v3.Cluster, error)
 		return cluster, err
 	}
 	cluster = cluster.DeepCopy()
+	if cluster.Labels == nil {
+		cluster.Labels = map[string]string{}
+	}
+	cluster.Labels[providerKey] = provider
 	cluster.Status.Provider = provider
 	return h.clusters.Update(cluster)
 }