127 lines
3.9 KiB
Go
127 lines
3.9 KiB
Go
package runner
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/ssdomei232/goodBaby/handler/db"
|
|
"github.com/ssdomei232/goodBaby/handler/logstore"
|
|
"github.com/ssdomei232/goodBaby/internal/retry"
|
|
"github.com/ssdomei232/goodBaby/model"
|
|
)
|
|
|
|
// Runner 执行某个 Timer 下所有启用的规则。
|
|
//
|
|
// 触发前先把 Timer 标记为已触发，避免规则执行期间(可能长达数小时)
|
|
// 被下一轮检查重复触发；用户重新签到后 Triggered 会被重置。
|
|
func Runner(timer *model.Timer) {
|
|
gormDB, err := db.GetGormDB()
|
|
if err != nil {
|
|
log.Printf("failed to get gorm db: %v", err)
|
|
return
|
|
}
|
|
|
|
// 1. 获取所有需要执行的 Rule
|
|
var rules []model.Rule
|
|
if err := gormDB.Where("timer_id = ? AND enabled = ?", timer.ID, true).Find(&rules).Error; err != nil {
|
|
log.Printf("failed to get rules: %v", err)
|
|
return
|
|
}
|
|
|
|
// 2. 标记已触发
|
|
now := time.Now().Unix()
|
|
if err := gormDB.Model(&model.Timer{}).Where("id = ?", timer.ID).
|
|
Updates(map[string]any{"triggered": true, "last_trigger": now}).Error; err != nil {
|
|
log.Printf("标记 Timer(ID: %d) 已触发失败: %v", timer.ID, err)
|
|
return
|
|
}
|
|
|
|
log.Printf("Timer %s (ID: %d) 已到期，开始执行 %d 条规则", timer.Name, timer.ID, len(rules))
|
|
|
|
// 3. 并发执行每个 Rule
|
|
var wg sync.WaitGroup
|
|
for i := range rules {
|
|
rule := rules[i]
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
ExecuteRule(&rule, model.TriggerTimer)
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
// NewTask 组装执行一条规则所需的输入：规则本体 + 从数据库取出的关联账号。
|
|
//
|
|
// 数据库访问只发生在这一层，driver 拿到的 task 已经是自包含的。
|
|
func NewTask(rule *model.Rule) (*model.RuleTask, error) {
|
|
account, err := db.GetAccount(rule.AccountID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &model.RuleTask{Rule: rule, Account: account}, nil
|
|
}
|
|
|
|
// ExecuteRule 执行单条规则并记录执行日志
|
|
func ExecuteRule(rule *model.Rule, trigger string) error {
|
|
ctx, cancel := retry.ExecutionContext()
|
|
defer cancel()
|
|
return ExecuteRuleWithContext(ctx, rule, trigger)
|
|
}
|
|
|
|
// ExecuteRuleWithContext 使用给定 context 执行规则，手动测试时可传入较短的超时
|
|
func ExecuteRuleWithContext(ctx context.Context, rule *model.Rule, trigger string) error {
|
|
task, err := NewTask(rule)
|
|
return execTask(ctx, task, err, &model.ExecutionLog{
|
|
UID: rule.UID,
|
|
RuleID: rule.ID,
|
|
RuleName: rule.Name,
|
|
RuleType: rule.Type,
|
|
TimerID: rule.TimerID,
|
|
Trigger: trigger,
|
|
})
|
|
}
|
|
|
|
// ExecuteGatewayRuleWithContext 执行一条消息网关规则，日志里带上网关信息。
|
|
//
|
|
// 与定时器规则走同一条执行链路，区别只有触发来源与日志字段。
|
|
func ExecuteGatewayRuleWithContext(ctx context.Context, rule *model.GatewayRule, trigger string) error {
|
|
task, err := NewTask(rule.AsRule())
|
|
return execTask(ctx, task, err, &model.ExecutionLog{
|
|
UID: rule.UID,
|
|
RuleName: rule.Name,
|
|
RuleType: rule.Type,
|
|
GatewayID: rule.GatewayID,
|
|
GatewayRuleID: rule.ID,
|
|
Trigger: trigger,
|
|
})
|
|
}
|
|
|
|
// execTask 执行规则并写入执行日志。
|
|
//
|
|
// taskErr 是组装输入时的错误(例如账号已被删除)：这时不会走到 driver，
|
|
// 但同样要留下一条失败日志，方便用户在 WebUI 上看到原因。
|
|
func execTask(ctx context.Context, task *model.RuleTask, taskErr error, entry *model.ExecutionLog) error {
|
|
if taskErr != nil {
|
|
return record(entry, taskErr)
|
|
}
|
|
|
|
log.Printf("执行规则 [%s] %s (触发来源: %s)", entry.RuleType, entry.RuleName, entry.Trigger)
|
|
return record(entry, GetGlobalExecutorRegistry().Execute(ctx, task))
|
|
}
|
|
|
|
// record 写入执行日志并返回原始错误
|
|
func record(entry *model.ExecutionLog, err error) error {
|
|
entry.Success = err == nil
|
|
entry.Message = "执行成功"
|
|
if err != nil {
|
|
entry.Message = err.Error()
|
|
log.Printf("执行规则失败 [%s] %s: %v", entry.RuleType, entry.RuleName, err)
|
|
}
|
|
|
|
logstore.Record(entry)
|
|
return err
|
|
}
|