35 lines
814 B
Go
35 lines
814 B
Go
package middleware
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/robfig/cron/v3"
|
|
"github.com/sirupsen/logrus"
|
|
"hy2xs-admin/dao"
|
|
"hy2xs-admin/model/constant"
|
|
"hy2xs-admin/service"
|
|
"time"
|
|
)
|
|
|
|
func InitCron() error {
|
|
loc := time.Now().Location()
|
|
c := cron.New(cron.WithLocation(loc))
|
|
_, err := c.AddFunc("@every 30s", service.CronHandleAccount)
|
|
if err != nil {
|
|
logrus.Errorf("cron add func CronHandleAccount err: %v", err)
|
|
return errors.New("cron add func CronHandleAccount err")
|
|
}
|
|
resetTrafficCron, err := dao.GetConfig("key = ?", constant.ResetTrafficCron)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if *resetTrafficCron.Value != "" {
|
|
_, err := c.AddFunc(*resetTrafficCron.Value, service.CronResetTraffic)
|
|
if err != nil {
|
|
logrus.Errorf("cron add func CronResetTraffic err: %v", err)
|
|
}
|
|
}
|
|
c.Start()
|
|
return nil
|
|
}
|
|
|