mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
43 lines
775 B
Go
43 lines
775 B
Go
package vtbquotation
|
|
|
|
import (
|
|
"github.com/fumiama/cron"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/FloatTech/zbputils/process"
|
|
|
|
"github.com/FloatTech/ZeroBot-Plugin/plugin_vtb_quotation/model"
|
|
)
|
|
|
|
func init() {
|
|
go func() {
|
|
process.SleepAbout1sTo2s()
|
|
log.Println("[vtb/cron] 开启vtb数据库日常更新")
|
|
vtbDaily()
|
|
}()
|
|
}
|
|
|
|
func vtbDaily() {
|
|
c := cron.New()
|
|
_, err := c.AddFunc("0 4 * * *", func() { vtbData() })
|
|
if err != nil {
|
|
log.Errorln("定时任务有错误:", err)
|
|
} else {
|
|
log.Println("开启vtb数据库定时任务")
|
|
c.Start()
|
|
}
|
|
}
|
|
|
|
func vtbData() {
|
|
db := model.Initialize(dbfile)
|
|
if db != nil {
|
|
for _, v := range db.GetVtbList() {
|
|
db.StoreVtb(v)
|
|
}
|
|
err := db.Close()
|
|
if err != nil {
|
|
log.Errorln("[vtb/cron]", err)
|
|
}
|
|
}
|
|
}
|