package dao import ( "fmt" "github.com/LoveLosita/smartflow/backend/services/runtime/model" mysqlinfra "github.com/LoveLosita/smartflow/backend/shared/infra/mysql" redisinfra "github.com/LoveLosita/smartflow/backend/shared/infra/redis" "github.com/go-redis/redis/v8" "gorm.io/gorm" ) // OpenDBFromConfig 创建 schedule 服务自己的数据库句柄。 // // 职责边界: // 1. 只迁移 schedule_events / schedules 这两个正式日程写模型; // 2. 不迁移 task、task-class、course 或 active-scheduler 的表,避免 schedule 服务越权管理其它领域; // 3. 迁移期仍检查 task/task-class 依赖表是否存在,方便启动阶段暴露部署顺序问题。 func OpenDBFromConfig() (*gorm.DB, error) { db, err := mysqlinfra.OpenDBFromConfig() if err != nil { return nil, err } if err = db.AutoMigrate(&model.ScheduleEvent{}, &model.Schedule{}); err != nil { return nil, fmt.Errorf("auto migrate schedule tables failed: %w", err) } if err = ensureRuntimeDependencyTables(db); err != nil { return nil, err } return db, nil } // OpenRedisFromConfig 创建 schedule 服务自己的 Redis 句柄。 // // 职责边界: // 1. 只负责初始化 schedule 读缓存所需 Redis client; // 2. 不创建、不清理任何业务 key; // 3. Ping 失败直接返回错误,避免读缓存链路静默降级成难排查的启动问题。 func OpenRedisFromConfig() (*redis.Client, error) { return redisinfra.OpenRedisFromConfig() } // ensureRuntimeDependencyTables 显式检查迁移期仍需读取或锁定的外部表。 // // 说明: // 1. task_classes / task_items 支撑智能粗排、任务块撤销和补做目标校验; // 2. tasks 支撑 active-scheduler confirm 的 task_pool 锁定; // 3. 下一轮拆 task / task-class 后,应把这些依赖改为 RPC 或 read model,再从这里移除。 func ensureRuntimeDependencyTables(db *gorm.DB) error { for _, table := range []string{"task_classes", "task_items", "tasks"} { if !db.Migrator().HasTable(table) { return fmt.Errorf("schedule runtime dependency table missing: %s", table) } } return nil }