a3eaa9b2c2
后端：
1. 主动调度 graph + session bridge 收口——把 dry-run / select / preview / confirm / rerun 串成受限 graph，新增 active_schedule_sessions 缓存与聊天拦截，ready_preview 后释放回自由聊天
2. 会话与通知链路对齐——notification 统一绑定 conversation_id，action_url 指向 /assistant/{conversation_id}，会话不存在改回 404 语义，避免 wrong param type 误导排障
3. estimated_sections 写入与主动调度消费链路补齐——任务创建、quick task 与随口记入口都透传估计节数，主动调度只消费落库值
前端：
4. AssistantPanel 最小适配主动调度预览与失败态——复用主动调度卡片/微调弹窗，补历史加载失败可见提示与跨账号会话拦截
文档：
5. 更新主动调度缺口分阶段实施计划和实现方案，标记阶段 0-2 收口并同步接力状态
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package inits
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/spf13/viper"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
func autoMigrateModels(db *gorm.DB) error {
|
||
models := []any{
|
||
&model.User{},
|
||
&model.AgentChat{},
|
||
&model.ChatHistory{},
|
||
&model.AgentTimelineEvent{},
|
||
&model.Task{},
|
||
&model.TaskClass{},
|
||
&model.TaskClassItem{},
|
||
&model.ScheduleEvent{},
|
||
&model.Schedule{},
|
||
&model.ActiveScheduleJob{},
|
||
&model.ActiveScheduleTrigger{},
|
||
&model.ActiveSchedulePreview{},
|
||
&model.NotificationRecord{},
|
||
&model.UserNotificationChannel{},
|
||
&model.AgentOutboxMessage{},
|
||
&model.AgentScheduleState{},
|
||
&model.ActiveScheduleSession{},
|
||
&model.AgentStateSnapshotRecord{},
|
||
&model.MemoryItem{},
|
||
&model.MemoryJob{},
|
||
&model.MemoryAuditLog{},
|
||
&model.MemoryUserSetting{},
|
||
}
|
||
|
||
for _, m := range models {
|
||
if err := db.AutoMigrate(m); err != nil {
|
||
return fmt.Errorf("auto migrate failed for %T: %w", m, err)
|
||
}
|
||
}
|
||
if err := backfillAutoMigrateData(db); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// backfillAutoMigrateData 补齐 AutoMigrate 无法表达的条件回填。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只处理新增列上线后的兼容数据修复，不替代业务迁移系统；
|
||
// 2. 当前仅回填历史动态任务日程来源，确保旧的 type=task 记录按 task_item 解释；
|
||
// 3. 失败时直接返回错误，避免服务在 schema 半迁移状态下继续启动。
|
||
func backfillAutoMigrateData(db *gorm.DB) error {
|
||
// 1. AutoMigrate 只能新增列和默认值，不能表达"仅 type=task 时回填"。
|
||
// 2. 这里把历史任务日程显式标记为 task_item，避免后续主动调度读取 rel_id 时误判来源。
|
||
// 3. 新增 task_pool 正式落库仍必须由 apply 链路显式写 task_source_type=task_pool。
|
||
result := db.Exec(
|
||
"UPDATE schedule_events SET task_source_type = ? WHERE type = ? AND (task_source_type IS NULL OR task_source_type = '')",
|
||
"task_item",
|
||
"task",
|
||
)
|
||
if result.Error != nil {
|
||
return fmt.Errorf("backfill schedule_events.task_source_type failed: %w", result.Error)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func ConnectDB() (*gorm.DB, error) {
|
||
host := viper.GetString("database.host")
|
||
port := viper.GetString("database.port")
|
||
user := viper.GetString("database.user")
|
||
password := viper.GetString("database.password")
|
||
dbname := viper.GetString("database.dbname")
|
||
|
||
dsn := fmt.Sprintf(
|
||
"%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||
user, password, host, port, dbname,
|
||
)
|
||
|
||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
if err = autoMigrateModels(db); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
log.Println("Database connected successfully")
|
||
log.Println("Database auto migration completed")
|
||
return db, nil
|
||
}
|