mirror of
				https://github.com/go-gitea/gitea
				synced 2025-11-04 05:18:25 +00:00 
			
		
		
		
	Fix dropTableColumns sqlite implementation (#7710)
* Fix dropTableColumns sqlite implementation * use droptables and its index dropping support in v78 and v85 * golang-ci fixes * Add migration from gitea 1.3.3 for sqlite which reveals the droptables bug - thus showing this works
This commit is contained in:
		
							
								
								
									
										
											BIN
										
									
								
								integrations/migration-test/gitea-v1.3.3.sqlite3.sql.gz
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										
											BIN
										
									
								
								integrations/migration-test/gitea-v1.3.3.sqlite3.sql.gz
									
									
									
									
									
										Normal file
									
								
							
										
											Binary file not shown.
										
									
								
							@@ -327,11 +327,25 @@ func dropTableColumns(sess *xorm.Session, tableName string, columnNames ...strin
 | 
				
			|||||||
			return err
 | 
								return err
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		tableSQL := string(res[0]["sql"])
 | 
							tableSQL := string(res[0]["sql"])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Separate out the column definitions
 | 
				
			||||||
		tableSQL = tableSQL[strings.Index(tableSQL, "("):]
 | 
							tableSQL = tableSQL[strings.Index(tableSQL, "("):]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Remove the required columnNames
 | 
				
			||||||
		for _, name := range columnNames {
 | 
							for _, name := range columnNames {
 | 
				
			||||||
			tableSQL = regexp.MustCompile(regexp.QuoteMeta("`"+name+"`")+"[^`,)]*[,)]").ReplaceAllString(tableSQL, "")
 | 
								tableSQL = regexp.MustCompile(regexp.QuoteMeta("`"+name+"`")+"[^`,)]*?[,)]").ReplaceAllString(tableSQL, "")
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Ensure the query is ended properly
 | 
				
			||||||
 | 
							tableSQL = strings.TrimSpace(tableSQL)
 | 
				
			||||||
 | 
							if tableSQL[len(tableSQL)-1] != ')' {
 | 
				
			||||||
 | 
								if tableSQL[len(tableSQL)-1] == ',' {
 | 
				
			||||||
 | 
									tableSQL = tableSQL[:len(tableSQL)-1]
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								tableSQL += ")"
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Find all the columns in the table
 | 
				
			||||||
		columns := regexp.MustCompile("`([^`]*)`").FindAllString(tableSQL, -1)
 | 
							columns := regexp.MustCompile("`([^`]*)`").FindAllString(tableSQL, -1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		tableSQL = fmt.Sprintf("CREATE TABLE `new_%s_new` ", tableName) + tableSQL
 | 
							tableSQL = fmt.Sprintf("CREATE TABLE `new_%s_new` ", tableName) + tableSQL
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -5,13 +5,7 @@
 | 
				
			|||||||
package migrations
 | 
					package migrations
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"code.gitea.io/gitea/models"
 | 
					 | 
				
			||||||
	"code.gitea.io/gitea/modules/log"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"github.com/go-xorm/xorm"
 | 
						"github.com/go-xorm/xorm"
 | 
				
			||||||
	"xorm.io/core"
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func renameRepoIsBareToIsEmpty(x *xorm.Engine) error {
 | 
					func renameRepoIsBareToIsEmpty(x *xorm.Engine) error {
 | 
				
			||||||
@@ -21,73 +15,28 @@ func renameRepoIsBareToIsEmpty(x *xorm.Engine) error {
 | 
				
			|||||||
		IsEmpty bool `xorm:"INDEX"`
 | 
							IsEmpty bool `xorm:"INDEX"`
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// First remove the index
 | 
					 | 
				
			||||||
	sess := x.NewSession()
 | 
						sess := x.NewSession()
 | 
				
			||||||
	defer sess.Close()
 | 
						defer sess.Close()
 | 
				
			||||||
	if err := sess.Begin(); err != nil {
 | 
						if err := sess.Begin(); err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	var err error
 | 
					 | 
				
			||||||
	if models.DbCfg.Type == core.POSTGRES || models.DbCfg.Type == core.SQLITE {
 | 
					 | 
				
			||||||
		_, err = sess.Exec("DROP INDEX IF EXISTS IDX_repository_is_bare")
 | 
					 | 
				
			||||||
	} else if models.DbCfg.Type == core.MSSQL {
 | 
					 | 
				
			||||||
		_, err = sess.Exec(`DECLARE @ConstraintName VARCHAR(256)
 | 
					 | 
				
			||||||
		DECLARE @SQL NVARCHAR(256)
 | 
					 | 
				
			||||||
		SELECT @ConstraintName = obj.name FROM sys.columns col LEFT OUTER JOIN sys.objects obj ON obj.object_id = col.default_object_id AND obj.type = 'D' WHERE col.object_id = OBJECT_ID('repository') AND obj.name IS NOT NULL AND col.name = 'is_bare'
 | 
					 | 
				
			||||||
		SET @SQL = N'ALTER TABLE [repository] DROP CONSTRAINT [' + @ConstraintName + N']'
 | 
					 | 
				
			||||||
		EXEC sp_executesql @SQL`)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			return err
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	} else if models.DbCfg.Type == core.MYSQL {
 | 
					 | 
				
			||||||
		indexes, err := sess.QueryString(`SHOW INDEX FROM repository WHERE KEY_NAME = 'IDX_repository_is_bare'`)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			return err
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if len(indexes) >= 1 {
 | 
					 | 
				
			||||||
			_, err = sess.Exec("DROP INDEX IDX_repository_is_bare ON repository")
 | 
					 | 
				
			||||||
			if err != nil {
 | 
					 | 
				
			||||||
				return fmt.Errorf("Drop index failed: %v", err)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		_, err = sess.Exec("DROP INDEX IDX_repository_is_bare ON repository")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return fmt.Errorf("Drop index failed: %v", err)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err = sess.Commit(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err := sess.Begin(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err := sess.Sync2(new(Repository)); err != nil {
 | 
						if err := sess.Sync2(new(Repository)); err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if _, err := sess.Exec("UPDATE repository SET is_empty = is_bare;"); err != nil {
 | 
						if _, err := sess.Exec("UPDATE repository SET is_empty = is_bare;"); err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						if err := sess.Commit(); err != nil {
 | 
				
			||||||
	if models.DbCfg.Type != core.SQLITE {
 | 
					 | 
				
			||||||
		_, err = sess.Exec("ALTER TABLE repository DROP COLUMN is_bare")
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			return fmt.Errorf("Drop column failed: %v", err)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err = sess.Commit(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if models.DbCfg.Type == core.SQLITE {
 | 
						if err := sess.Begin(); err != nil {
 | 
				
			||||||
		log.Warn("TABLE repository's COLUMN is_bare should be DROP but sqlite is not supported, you could manually do that.")
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return nil
 | 
						if err := dropTableColumns(sess, "repository", "is_bare"); err != nil {
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return sess.Commit()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -8,9 +8,7 @@ import (
 | 
				
			|||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/go-xorm/xorm"
 | 
						"github.com/go-xorm/xorm"
 | 
				
			||||||
	"xorm.io/core"
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"code.gitea.io/gitea/models"
 | 
					 | 
				
			||||||
	"code.gitea.io/gitea/modules/generate"
 | 
						"code.gitea.io/gitea/modules/generate"
 | 
				
			||||||
	"code.gitea.io/gitea/modules/log"
 | 
						"code.gitea.io/gitea/modules/log"
 | 
				
			||||||
	"code.gitea.io/gitea/modules/util"
 | 
						"code.gitea.io/gitea/modules/util"
 | 
				
			||||||
@@ -37,41 +35,6 @@ func hashAppToken(x *xorm.Engine) error {
 | 
				
			|||||||
	// First remove the index
 | 
						// First remove the index
 | 
				
			||||||
	sess := x.NewSession()
 | 
						sess := x.NewSession()
 | 
				
			||||||
	defer sess.Close()
 | 
						defer sess.Close()
 | 
				
			||||||
	if err := sess.Begin(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	var err error
 | 
					 | 
				
			||||||
	if models.DbCfg.Type == core.POSTGRES || models.DbCfg.Type == core.SQLITE {
 | 
					 | 
				
			||||||
		_, err = sess.Exec("DROP INDEX IF EXISTS UQE_access_token_sha1")
 | 
					 | 
				
			||||||
	} else if models.DbCfg.Type == core.MSSQL {
 | 
					 | 
				
			||||||
		_, err = sess.Exec(`DECLARE @ConstraintName VARCHAR(256)
 | 
					 | 
				
			||||||
		DECLARE @SQL NVARCHAR(256)
 | 
					 | 
				
			||||||
		SELECT @ConstraintName = obj.name FROM sys.columns col LEFT OUTER JOIN sys.objects obj ON obj.object_id = col.default_object_id AND obj.type = 'D' WHERE col.object_id = OBJECT_ID('access_token') AND obj.name IS NOT NULL AND col.name = 'sha1'
 | 
					 | 
				
			||||||
		SET @SQL = N'ALTER TABLE [access_token] DROP CONSTRAINT [' + @ConstraintName + N']'
 | 
					 | 
				
			||||||
		EXEC sp_executesql @SQL`)
 | 
					 | 
				
			||||||
	} else if models.DbCfg.Type == core.MYSQL {
 | 
					 | 
				
			||||||
		indexes, err := sess.QueryString(`SHOW INDEX FROM access_token WHERE KEY_NAME = 'UQE_access_token_sha1'`)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			return err
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if len(indexes) >= 1 {
 | 
					 | 
				
			||||||
			_, err = sess.Exec("DROP INDEX UQE_access_token_sha1 ON access_token")
 | 
					 | 
				
			||||||
			if err != nil {
 | 
					 | 
				
			||||||
				return err
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		_, err = sess.Exec("DROP INDEX UQE_access_token_sha1 ON access_token")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return fmt.Errorf("Drop index failed: %v", err)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err = sess.Commit(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if err := sess.Begin(); err != nil {
 | 
						if err := sess.Begin(); err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
@@ -81,7 +44,7 @@ func hashAppToken(x *xorm.Engine) error {
 | 
				
			|||||||
		return fmt.Errorf("Sync2: %v", err)
 | 
							return fmt.Errorf("Sync2: %v", err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if err = sess.Commit(); err != nil {
 | 
						if err := sess.Commit(); err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user