fix(platform): provider platform client nil when upgarde (#1812)

cmd/tke-installer/app/installer/upgrader.go

@@ -56,6 +56,7 @@ import (
 
 	// import platform schema
 	_ "tkestack.io/tke/api/platform/install"
+	baremetalcluster "tkestack.io/tke/pkg/platform/provider/baremetal/cluster"
 )
 
 const (
@@ -347,6 +348,17 @@ func (t *TKE) prepareForUpgrade(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+
+	provider, err := baremetalcluster.NewProvider()
+	if err != nil {
+		return err
+	}
+	provider.PlatformClient = t.platformClient
+	if err := clusterprovider.ReRegister(provider.Name(), provider); err != nil {
+		return err
+	}
+	t.clusterProvider = provider
+
 	t.Cluster, err = clusterprovider.GetV1ClusterByName(ctx, t.platformClient, "global", clusterprovider.AdminUsername)
 	if err != nil {
 		return err

pkg/platform/provider/cluster/cluster.go

@@ -58,6 +58,17 @@ func Register(name string, provider Provider) {
 	providers[name] = provider
 }
 
+// re register provider, if provider's name exists, new provider will replace old provider
+func ReRegister(name string, provider Provider) error {
+	providersMu.Lock()
+	defer providersMu.Unlock()
+	if provider == nil {
+		return fmt.Errorf("cluster: Register provider is nil")
+	}
+	providers[name] = provider
+	return nil
+}
+
 // RegisterHandler register all provider's hanlder.
 func RegisterHandler(mux *mux.PathRecorderMux) {
 	for _, p := range providers {