mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-11-16 13:13:41 +08:00
✨ feat: add chat cache (#152)
This commit is contained in:
37
cron/main.go
Normal file
37
cron/main.go
Normal 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()
|
||||
}
|
||||
Reference in New Issue
Block a user