From 56a3b50136a1cbdd32493328be7f05e60f94c4e5 Mon Sep 17 00:00:00 2001 From: Lunny Xiao Date: Fri, 3 Dec 2021 01:12:11 +0800 Subject: [PATCH] Check if column exist before rename if exist, just return with no error (#17870) (#17882) * Check if column exist before rename if exist, just return with no error * Also check if errors column exist * Add comment for migration * Fix sqlite test --- models/migrations/v184.go | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/models/migrations/v184.go b/models/migrations/v184.go index b7be342b8..97bc72d5d 100644 --- a/models/migrations/v184.go +++ b/models/migrations/v184.go @@ -5,6 +5,7 @@ package migrations import ( + "context" "fmt" "code.gitea.io/gitea/modules/setting" @@ -19,6 +20,22 @@ func renameTaskErrorsToMessage(x *xorm.Engine) error { Status int `xorm:"index"` } + // This migration maybe rerun so that we should check if it has been run + messageExist, err := x.Dialect().IsColumnExist(x.DB(), context.Background(), "task", "message") + if err != nil { + return err + } + + if messageExist { + errorsExist, err := x.Dialect().IsColumnExist(x.DB(), context.Background(), "task", "errors") + if err != nil { + return err + } + if !errorsExist { + return nil + } + } + sess := x.NewSession() defer sess.Close() if err := sess.Begin(); err != nil { @@ -29,6 +46,13 @@ func renameTaskErrorsToMessage(x *xorm.Engine) error { return fmt.Errorf("error on Sync2: %v", err) } + if messageExist { + // if both errors and message exist, drop message at first + if err := dropTableColumns(sess, "task", "message"); err != nil { + return err + } + } + switch { case setting.Database.UseMySQL: if _, err := sess.Exec("ALTER TABLE `task` CHANGE errors message text"); err != nil {