Files
goodBaby/handler/checker/checker.go
T

59 lines
1.4 KiB
Go
Raw Normal View History

2026-04-06 10:10:31 +08:00
package checker
import (
"log"
"time"
"github.com/ssdomei232/goodBaby/handler/db"
"github.com/ssdomei232/goodBaby/handler/reminder"
"github.com/ssdomei232/goodBaby/handler/runner"
"github.com/ssdomei232/goodBaby/model"
)
2026-07-26 22:19:41 +08:00
// CheckTimers 定时检查所有 timers 是否需要被触发
2026-04-06 10:10:31 +08:00
//
2026-07-26 22:19:41 +08:00
// 同时向即将到期的 timers 对应的用户发送提醒
2026-04-06 10:10:31 +08:00
func CheckTimers() {
gormDB, err := db.GetGormDB()
if err != nil {
log.Printf("failed to get gorm db: %v", err)
return
}
2026-07-26 22:19:41 +08:00
// 1. 获取所有启用中的 timers
2026-04-06 10:10:31 +08:00
var timers []model.Timer
2026-07-26 22:19:41 +08:00
if err := gormDB.Where("enabled = ?", true).Find(&timers).Error; err != nil {
2026-04-06 10:10:31 +08:00
log.Printf("failed to get timers: %v", err)
return
}
2026-07-26 22:19:41 +08:00
currentTime := time.Now().Unix()
2026-04-06 10:10:31 +08:00
2026-07-26 22:19:41 +08:00
// 2. 检查每个 timer 是否需要被提醒或触发
for i := range timers {
timer := timers[i]
2026-04-06 10:10:31 +08:00
2026-07-26 22:19:41 +08:00
// 已触发过的 timer 在用户重新签到之前不再处理
if timer.Triggered {
continue
2026-04-06 10:10:31 +08:00
}
2026-07-26 22:19:41 +08:00
nextSignTime := timer.NextSignTime()
remindTime := timer.RemindAt()
2026-04-06 10:10:31 +08:00
if currentTime >= nextSignTime {
runner.Runner(&timer)
2026-07-26 22:19:41 +08:00
continue
}
// 每个签到周期只提醒一次
if currentTime >= remindTime && timer.LastRemind < remindTime {
reminder.Reminder(&timer)
if err := gormDB.Model(&model.Timer{}).Where("id = ?", timer.ID).
Update("last_remind", currentTime).Error; err != nil {
log.Printf("更新 Timer(ID: %d) 提醒时间失败: %v", timer.ID, err)
}
2026-04-06 10:10:31 +08:00
}
}
}