feat: add chat cache (#152)

This commit is contained in:
Buer
2024-04-16 10:36:18 +08:00
committed by GitHub
parent bbaa4eec4b
commit 3c7c13758b
23 changed files with 557 additions and 49 deletions

37
cron/main.go Normal file
View File

@@ -0,0 +1,37 @@
package cron
import (
"one-api/common"
"one-api/model"
"time"
"github.com/go-co-op/gocron/v2"
)
func InitCron() {
scheduler, err := gocron.NewScheduler()
if err != nil {
common.SysLog("Cron scheduler error: " + err.Error())
return
}
// 添加删除cache的任务
_, err = scheduler.NewJob(
gocron.DailyJob(
1,
gocron.NewAtTimes(
gocron.NewAtTime(0, 5, 0),
)),
gocron.NewTask(func() {
model.RemoveChatCache(time.Now().Unix())
common.SysLog("删除过期缓存数据")
}),
)
if err != nil {
common.SysLog("Cron job error: " + err.Error())
return
}
scheduler.Start()
}