fix

pingcap · Oct 17, 2023 · 8626df3 · 8626df3
1 parent cda0829
commit 8626df3
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/pkg/cluster/manager/patch.go b/pkg/cluster/manager/patch.go
@@ -105,7 +105,7 @@ func (m *Manager) Patch(name string, packagePath string, opt operator.Options, o
 				return nil
 			}
 			// TBD: should patch be treated as an upgrade?
-			return operator.Upgrade(ctx, topo, opt, tlsCfg, base.Version, false)
+			return operator.Upgrade(ctx, topo, opt, tlsCfg, base.Version, base.Version)
 		}).
 		Build()
 

diff --git a/pkg/cluster/manager/reload.go b/pkg/cluster/manager/reload.go
@@ -124,7 +124,7 @@ func (m *Manager) Reload(name string, gOpt operator.Options, skipRestart, skipCo
 			return err
 		}
 		b.Func("Upgrade Cluster", func(ctx context.Context) error {
-			return operator.Upgrade(ctx, topo, gOpt, tlsCfg, base.Version, false)
+			return operator.Upgrade(ctx, topo, gOpt, tlsCfg, base.Version, base.Version)
 		})
 	}
 

diff --git a/pkg/cluster/manager/upgrade.go b/pkg/cluster/manager/upgrade.go
@@ -237,7 +237,7 @@ Do you want to continue? [y/N]:`,
 			if offline {
 				return nil
 			}
-			return operator.Upgrade(ctx, topo, opt, tlsCfg, base.Version, true)
+			return operator.Upgrade(ctx, topo, opt, tlsCfg, base.Version, clusterVersion)
 		}).
 		Build()
 

diff --git a/pkg/cluster/operation/upgrade.go b/pkg/cluster/operation/upgrade.go
@@ -46,7 +46,7 @@ func Upgrade(
 	options Options,
 	tlsCfg *tls.Config,
 	currentVersion string,
-	isUpgrade bool,
+	targetVersion string,
 ) error {
 	roleFilter := set.NewStringSet(options.Roles...)
 	nodeFilter := set.NewStringSet(options.Nodes...)
@@ -109,7 +109,7 @@ func Upgrade(
 				endpoints = append(endpoints, utils.JoinHostPort(db.GetManageHost(), db.StatusPort))
 			}
 
-			if isUpgrade && tidbver.TiDBSupportUpgradeAPI(currentVersion) {
+			if currentVersion != targetVersion && tidbver.TiDBSupportUpgradeAPI(currentVersion) && tidbver.TiDBSupportUpgradeAPI(targetVersion) {
 				tidbClient = api.NewTiDBClient(ctx, endpoints, 10*time.Second, tlsCfg)
 				err = tidbClient.StartUpgrade()
 				if err != nil {
@@ -198,7 +198,7 @@ func Upgrade(
 
 		switch component.Name() {
 		case spec.ComponentTiDB:
-			if isUpgrade && tidbver.TiDBSupportUpgradeAPI(currentVersion) {
+			if currentVersion != targetVersion && tidbver.TiDBSupportUpgradeAPI(currentVersion) && tidbver.TiDBSupportUpgradeAPI(targetVersion) {
 				err = tidbClient.FinishUpgrade()
 				if err != nil {
 					return err