6843c7efac
后端：
1.阶段 5 task 服务边界落地
- 新增 cmd/task 与 services/task/{dao,rpc,sv}，承载 task zrpc、tasks 表迁移和 task outbox 消费边界
- 新增 gateway/client/task、shared/contracts/task 和 task port，gateway /api/v1/task/* 切到 task zrpc client
- 将 task.urgency.promote.requested handler / relay / retry loop 迁入 cmd/task，单体 worker 不再消费 task outbox
- 保留单体 Agent 残留 task 查询的 publish-only 写入能力，避免迁移期 task 事件丢失
- active-scheduler task facts / due job scanner 切到 task RPC，并移除启动期 tasks 表依赖检查
- 更新阶段 5 文档，记录 task 切流点、旧实现保留、跨域 DB 依赖缩减和下一轮建议
- 补充 task rpc 示例配置
93 lines
3.1 KiB
Go
93 lines
3.1 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
outboxinfra "github.com/LoveLosita/smartflow/backend/infra/outbox"
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"github.com/go-redis/redis/v8"
|
||
"github.com/spf13/viper"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// OpenDBFromConfig 创建 task 服务自己的数据库句柄。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只迁移 tasks 表和 task 服务自己的 outbox 表；
|
||
// 2. 不迁移 active-scheduler、schedule、course 或 task-class 表；
|
||
// 3. 迁移期仍检查 active_schedule_jobs 是否存在，因为 task 写入后还会 best-effort 同步 due job。
|
||
func OpenDBFromConfig() (*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 = db.AutoMigrate(&model.Task{}); err != nil {
|
||
return nil, fmt.Errorf("auto migrate task tables failed: %w", err)
|
||
}
|
||
if err = autoMigrateTaskOutboxTable(db); err != nil {
|
||
return nil, err
|
||
}
|
||
if err = ensureRuntimeDependencyTables(db); err != nil {
|
||
return nil, err
|
||
}
|
||
return db, nil
|
||
}
|
||
|
||
// OpenRedisFromConfig 创建 task 服务自己的 Redis 句柄。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只负责初始化 task 缓存和紧急性平移去重锁所需 Redis client；
|
||
// 2. 不清理任何业务 key；
|
||
// 3. Ping 失败直接返回错误，避免缓存链路静默降级。
|
||
func OpenRedisFromConfig() (*redis.Client, error) {
|
||
client := redis.NewClient(&redis.Options{
|
||
Addr: viper.GetString("redis.host") + ":" + viper.GetString("redis.port"),
|
||
Password: viper.GetString("redis.password"),
|
||
DB: 0,
|
||
})
|
||
if _, err := client.Ping(context.Background()).Result(); err != nil {
|
||
return nil, err
|
||
}
|
||
return client, nil
|
||
}
|
||
|
||
// autoMigrateTaskOutboxTable 只迁移 task 服务自己的 outbox 物理表。
|
||
func autoMigrateTaskOutboxTable(db *gorm.DB) error {
|
||
cfg, ok := outboxinfra.ResolveServiceConfig(outboxinfra.ServiceTask)
|
||
if !ok {
|
||
return fmt.Errorf("resolve task outbox config failed")
|
||
}
|
||
if err := db.Table(cfg.TableName).AutoMigrate(&model.AgentOutboxMessage{}); err != nil {
|
||
return fmt.Errorf("auto migrate task outbox table failed for %s (%s): %w", cfg.Name, cfg.TableName, err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ensureRuntimeDependencyTables 显式检查 task 迁移期仍写入的跨域表。
|
||
//
|
||
// 说明：
|
||
// 1. active_schedule_jobs 属于 active-scheduler，自有迁移仍由 active-scheduler 管理；
|
||
// 2. 本轮为保持任务写入后 due job 同步语义，task 服务只检查存在性；
|
||
// 3. 下一轮把 due job 同步改为 active-scheduler RPC 或事件后，应从这里移除。
|
||
func ensureRuntimeDependencyTables(db *gorm.DB) error {
|
||
for _, table := range []string{"active_schedule_jobs"} {
|
||
if !db.Migrator().HasTable(table) {
|
||
return fmt.Errorf("task runtime dependency table missing: %s", table)
|
||
}
|
||
}
|
||
return nil
|
||
}
|