fix(database): 修复SQLite迁移中自增主键的定义问题

- 修复了自增列同时标记为主键时重复添加PRIMARY KEY约束的问题
- 确保自增主键列只添加一次PRIMARY KEY约束
- 优化了复合主键约束的添加逻辑,避免与自增约束冲突
- 修正了自增主键的类型定义为INTEGER以符合SQLite规范
```
main v1.0.2025
black 2026-04-14 09:54:48 +08:00
parent 33faa6f722
commit 7e54a96454
1 changed files with 23 additions and 24 deletions

View File

@ -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, " ")
}