```
fix(database): 修复SQLite迁移中自增主键的定义问题 - 修复了自增列同时标记为主键时重复添加PRIMARY KEY约束的问题 - 确保自增主键列只添加一次PRIMARY KEY约束 - 优化了复合主键约束的添加逻辑,避免与自增约束冲突 - 修正了自增主键的类型定义为INTEGER以符合SQLite规范 ```main v1.0.2025
parent
33faa6f722
commit
7e54a96454
|
|
@ -52,14 +52,16 @@ func (m *Migration) CreateTable(ctx context.Context, table string, columns map[s
|
|||
var primaryKeys []string
|
||||
for name, def := range columns {
|
||||
colDef := m.buildColumnDefinition(name, def)
|
||||
if def.PrimaryKey {
|
||||
// Only add to primaryKeys if it's not an auto-increment column
|
||||
// (auto-increment columns already have PRIMARY KEY in their definition)
|
||||
if def.PrimaryKey && !def.AutoIncrement {
|
||||
primaryKeys = append(primaryKeys, database.QuoteIdentifier(name))
|
||||
}
|
||||
colDefs = append(colDefs, " "+colDef)
|
||||
}
|
||||
|
||||
// Add primary key constraint if needed
|
||||
if len(primaryKeys) > 0 {
|
||||
// Add composite primary key constraint if needed (for non-auto-increment keys)
|
||||
if len(primaryKeys) > 1 {
|
||||
colDefs = append(colDefs, fmt.Sprintf(" PRIMARY KEY (%s)", strings.Join(primaryKeys, ", ")))
|
||||
}
|
||||
|
||||
|
|
@ -74,18 +76,16 @@ func (m *Migration) buildColumnDefinition(name string, def *database.ColumnDefin
|
|||
var parts []string
|
||||
parts = append(parts, database.QuoteIdentifier(name))
|
||||
|
||||
// Handle SQLite-specific types
|
||||
dbType := def.Type
|
||||
if def.AutoIncrement && def.PrimaryKey {
|
||||
if dbType == "INT" || dbType == "INTEGER" {
|
||||
dbType = "INTEGER"
|
||||
}
|
||||
}
|
||||
parts = append(parts, dbType)
|
||||
|
||||
// Handle SQLite-specific types for auto-increment primary key
|
||||
if def.PrimaryKey && def.AutoIncrement {
|
||||
parts = append(parts, "PRIMARY KEY AUTOINCREMENT")
|
||||
} else {
|
||||
// SQLite requires INTEGER type for AUTOINCREMENT
|
||||
parts = append(parts, "INTEGER PRIMARY KEY AUTOINCREMENT")
|
||||
return strings.Join(parts, " ")
|
||||
}
|
||||
|
||||
// Regular column definition
|
||||
parts = append(parts, def.Type)
|
||||
|
||||
if !def.Null {
|
||||
parts = append(parts, "NOT NULL")
|
||||
}
|
||||
|
|
@ -98,7 +98,6 @@ func (m *Migration) buildColumnDefinition(name string, def *database.ColumnDefin
|
|||
defaultValue := formatDefaultValue(def.Default)
|
||||
parts = append(parts, fmt.Sprintf("DEFAULT %s", defaultValue))
|
||||
}
|
||||
}
|
||||
|
||||
return strings.Join(parts, " ")
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue