fix(canvas): 画布版本 - 合并 PR #528 防止旧内容覆盖并支持历史恢复
将画布 revision CAS 与历史快照接到 schema v23,保留当前顶栏积分弹窗和已有菜单入口。 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -135,10 +135,14 @@ func primaryKeyColumn[T any](db *gorm.DB) (string, error) {
|
||||
if err := statement.Parse(new(T)); err != nil {
|
||||
return "", err
|
||||
}
|
||||
if len(statement.Schema.PrimaryFields) != 1 {
|
||||
return "", fmt.Errorf("表 %s 必须有且只有一个主键", statement.Schema.Table)
|
||||
if len(statement.Schema.PrimaryFields) == 0 {
|
||||
return "", fmt.Errorf("表 %s 必须有主键", statement.Schema.Table)
|
||||
}
|
||||
return statement.Schema.PrimaryFields[0].DBName, nil
|
||||
columns := make([]string, 0, len(statement.Schema.PrimaryFields))
|
||||
for _, field := range statement.Schema.PrimaryFields {
|
||||
columns = append(columns, field.DBName)
|
||||
}
|
||||
return strings.Join(columns, ", "), nil
|
||||
}
|
||||
|
||||
var timeType = reflect.TypeOf(time.Time{})
|
||||
@@ -279,6 +283,8 @@ func migrations() []tableMigration {
|
||||
migrateTable[model.WorkflowStepTask]("workflow_step_tasks"),
|
||||
migrateTable[model.ProductionTaskLink]("production_task_links"),
|
||||
migrateTable[model.CanvasProject]("canvas_projects"),
|
||||
migrateTable[model.CanvasSnapshot]("canvas_snapshots"),
|
||||
migrateTable[model.CanvasSnapshotResource]("canvas_snapshot_resources"),
|
||||
migrateTable[model.CanvasShare]("canvas_shares"),
|
||||
migrateTable[model.PromptTemplate]("prompt_templates"),
|
||||
migrateTable[model.UserPromptCustomization]("user_prompt_customizations"),
|
||||
|
||||
@@ -5,7 +5,9 @@ import (
|
||||
"testing"
|
||||
|
||||
"infinite-canvas/backend/internal/database"
|
||||
"infinite-canvas/backend/internal/model"
|
||||
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/schema"
|
||||
)
|
||||
|
||||
@@ -34,3 +36,30 @@ func TestMigrationListCoversSchemaModels(t *testing.T) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestCanvasHistoryCompositeKeyMigration(t *testing.T) {
|
||||
source, err := database.Open(database.Config{Driver: "sqlite", DSN: "file:history-migration-source?mode=memory&cache=shared"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
target, err := database.Open(database.Config{Driver: "sqlite", DSN: "file:history-migration-target?mode=memory&cache=shared"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
for _, db := range []*gorm.DB{source, target} {
|
||||
if err := db.AutoMigrate(&model.Resource{}, &model.CanvasSnapshotResource{}); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := db.Create(&model.Resource{ID: "resource"}).Error; err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
refs := []model.CanvasSnapshotResource{{SnapshotID: "b", ResourceID: "resource"}, {SnapshotID: "a", ResourceID: "resource"}}
|
||||
if err := source.Create(&refs).Error; err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
count, err := migrateTable[model.CanvasSnapshotResource]("canvas_snapshot_resources").run(source, target, true)
|
||||
if err != nil || count != 2 {
|
||||
t.Fatalf("history refs not copied/verified: %d %v", count, err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user