Подготовить HY2XS к production-сборке

apps/middleware/cron.go

@@ -0,0 +1,33 @@
+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
+}