separate frontend repository

cronjob/checkCoreJob.go

@@ -0,0 +1,17 @@
+package cronjob
+
+import (
+	"s-ui/service"
+)
+
+type CheckCoreJob struct {
+	service.ConfigService
+}
+
+func NewCheckCoreJob() *CheckCoreJob {
+	return &CheckCoreJob{}
+}
+
+func (s *CheckCoreJob) Run() {
+	s.ConfigService.StartCore("")
+}

cronjob/cronJob.go

@@ -0,0 +1,39 @@
+package cronjob
+
+import (
+	"time"
+
+	"github.com/robfig/cron/v3"
+)
+
+type CronJob struct {
+	cron *cron.Cron
+}
+
+func NewCronJob() *CronJob {
+	return &CronJob{}
+}
+
+func (c *CronJob) Start(loc *time.Location, trafficAge int) error {
+	c.cron = cron.New(cron.WithLocation(loc), cron.WithSeconds())
+	c.cron.Start()
+
+	go func() {
+		// Start stats job
+		c.cron.AddJob("@every 10s", NewStatsJob())
+		// Start expiry job
+		c.cron.AddJob("@every 1m", NewDepleteJob())
+		// Start deleting old stats
+		c.cron.AddJob("@daily", NewDelStatsJob(trafficAge))
+		// Start core if it is not running
+		c.cron.AddJob("@every 5s", NewCheckCoreJob())
+	}()
+
+	return nil
+}
+
+func (c *CronJob) Stop() {
+	if c.cron != nil {
+		c.cron.Stop()
+	}
+}

cronjob/delStatsJob.go

@@ -0,0 +1,26 @@
+package cronjob
+
+import (
+	"s-ui/logger"
+	"s-ui/service"
+)
+
+type DelStatsJob struct {
+	service.StatsService
+	trafficAge int
+}
+
+func NewDelStatsJob(ta int) *DelStatsJob {
+	return &DelStatsJob{
+		trafficAge: ta,
+	}
+}
+
+func (s *DelStatsJob) Run() {
+	err := s.StatsService.DelOldStats(s.trafficAge)
+	if err != nil {
+		logger.Warning("Deleting old statistics failed: ", err)
+		return
+	}
+	logger.Debug("Stats older than ", s.trafficAge, " days were deleted")
+}

cronjob/depleteJob.go

@@ -0,0 +1,22 @@
+package cronjob
+
+import (
+	"s-ui/logger"
+	"s-ui/service"
+)
+
+type DepleteJob struct {
+	service.ClientService
+}
+
+func NewDepleteJob() *DepleteJob {
+	return new(DepleteJob)
+}
+
+func (s *DepleteJob) Run() {
+	err := s.ClientService.DepleteClients()
+	if err != nil {
+		logger.Warning("Disable depleted users failed: ", err)
+		return
+	}
+}

cronjob/statsJob.go

@@ -0,0 +1,22 @@
+package cronjob
+
+import (
+	"s-ui/logger"
+	"s-ui/service"
+)
+
+type StatsJob struct {
+	service.StatsService
+}
+
+func NewStatsJob() *StatsJob {
+	return &StatsJob{}
+}
+
+func (s *StatsJob) Run() {
+	err := s.StatsService.SaveStats()
+	if err != nil {
+		logger.Warning("Get stats failed: ", err)
+		return
+	}
+}