fix: fix migrator recognizing nextval('"table_id_seq"'::regclass) (#304)

ifrvn · web-flow · commit 84224ecab0a0 · 2025-03-26T11:05:09.000+08:00
diff --git a/migrator.go b/migrator.go
@@ -495,8 +495,8 @@ func (m Migrator) ColumnTypes(value interface{}) (columnTypes []gorm.ColumnType,
 				column.LengthValue = typeLenValue
 			}
 
-			if (strings.HasPrefix(column.DefaultValueValue.String, "nextval('") &&
-				strings.HasSuffix(column.DefaultValueValue.String, "seq'::regclass)")) || (identityIncrement.Valid && identityIncrement.String != "") {
+			autoIncrementValuePattern := regexp.MustCompile(`^nextval\('"?[^']+seq"?'::regclass\)$`)
+			if autoIncrementValuePattern.MatchString(column.DefaultValueValue.String) || (identityIncrement.Valid && identityIncrement.String != "") {
 				column.AutoIncrementValue = sql.NullBool{Bool: true, Valid: true}
 				column.DefaultValueValue = sql.NullString{}
 			}

Original file line number	Diff line number	Diff line change
`@@ -495,8 +495,8 @@ func (m Migrator) ColumnTypes(value interface{}) (columnTypes []gorm.ColumnType,`
`495`	`495`	`column.LengthValue = typeLenValue`
`496`	`496`	`}`
`497`	`497`
`498`		`- if (strings.HasPrefix(column.DefaultValueValue.String, "nextval('") &&`
`499`		`- strings.HasSuffix(column.DefaultValueValue.String, "seq'::regclass)")) \|\| (identityIncrement.Valid && identityIncrement.String != "") {`
	`498`	+ autoIncrementValuePattern := regexp.MustCompile(`^nextval\('"?[^']+seq"?'::regclass\)$`)
	`499`	`+ if autoIncrementValuePattern.MatchString(column.DefaultValueValue.String) \|\| (identityIncrement.Valid && identityIncrement.String != "") {`
`500`	`500`	`column.AutoIncrementValue = sql.NullBool{Bool: true, Valid: true}`
`501`	`501`	`column.DefaultValueValue = sql.NullString{}`
`502`	`502`	`}`