mirror of
https://github.com/woodchen-ink/aimodels-prices.git
synced 2025-07-18 05:32:00 +08:00
wood chen
b11cf228e3
- 修改 InitCronJobs 函数中的定时任务执行时间，OpenRouter价格获取任务和其他厂商价格更新任务分别调整为每天06:04和06:30执行。 - 在 UpdateOtherPrices 函数中新增清除倍率缓存的逻辑，确保价格更新后缓存数据的准确性。 - 优化 GetPriceRates 函数，移除不必要的字段查询，提升查询效率。
74 lines
2.0 KiB
Go
74 lines
2.0 KiB
Go
package cron
|
||
|
||
import (
|
||
"log"
|
||
"time"
|
||
|
||
"github.com/robfig/cron/v3"
|
||
|
||
openrouter_api "aimodels-prices/cron/openrouter-api"
|
||
)
|
||
|
||
var cronScheduler *cron.Cron
|
||
|
||
// InitCronJobs 初始化并启动所有定时任务
|
||
func InitCronJobs() {
|
||
log.Println("初始化定时任务...")
|
||
|
||
// 创建一个新的cron调度器，使用秒级精度
|
||
cronScheduler = cron.New(cron.WithSeconds())
|
||
|
||
// 注册OpenRouter价格获取任务
|
||
// 每24小时执行一次
|
||
_, err := cronScheduler.AddFunc("4 6 * * *", func() {
|
||
if err := openrouter_api.FetchAndSavePrices(); err != nil {
|
||
log.Printf("OpenRouter价格获取任务执行失败: %v", err)
|
||
}
|
||
})
|
||
|
||
if err != nil {
|
||
log.Printf("注册OpenRouter价格获取任务失败: %v", err)
|
||
}
|
||
|
||
// 注册其他厂商价格更新任务
|
||
// 每24小时执行一次，错开时间避免同时执行
|
||
_, err = cronScheduler.AddFunc("30 6 * * *", func() {
|
||
if err := openrouter_api.UpdateOtherPrices(); err != nil {
|
||
log.Printf("其他厂商价格更新任务执行失败: %v", err)
|
||
}
|
||
})
|
||
|
||
if err != nil {
|
||
log.Printf("注册其他厂商价格更新任务失败: %v", err)
|
||
}
|
||
|
||
// 启动定时任务
|
||
cronScheduler.Start()
|
||
log.Println("定时任务已启动")
|
||
|
||
// 立即执行一次价格获取任务
|
||
go func() {
|
||
// 等待几秒钟，确保应用程序和数据库已完全初始化
|
||
time.Sleep(5 * time.Second)
|
||
log.Println("立即执行OpenRouter价格获取任务...")
|
||
if err := openrouter_api.FetchAndSavePrices(); err != nil {
|
||
log.Printf("初始OpenRouter价格获取任务执行失败: %v", err)
|
||
}
|
||
|
||
// 等待几秒后执行其他厂商价格更新任务
|
||
time.Sleep(3 * time.Second)
|
||
log.Println("立即执行其他厂商价格更新任务...")
|
||
if err := openrouter_api.UpdateOtherPrices(); err != nil {
|
||
log.Printf("初始其他厂商价格更新任务执行失败: %v", err)
|
||
}
|
||
}()
|
||
}
|
||
|
||
// StopCronJobs 停止所有定时任务
|
||
func StopCronJobs() {
|
||
if cronScheduler != nil {
|
||
cronScheduler.Stop()
|
||
log.Println("定时任务已停止")
|
||
}
|
||
}
|