优化逻辑

internal/routes/standby_ctrl/cron.go

@@ -0,0 +1,94 @@
+package standby_ctrl
+
+import (
+	"context"
+	"github.com/go-pkgz/cronrange"
+	"go.uber.org/zap"
+	"sync"
+	"time"
+)
+
+// Cron 时间控制器
+func Cron(rootRules []cronrange.Rule, cron string, play func(c context.Context) error) func(c context.Context) error {
+	// 设定默认时间规则
+	if cron == "" {
+		cron = "* * * *"
+	}
+
+	rules, err := cronrange.Parse(cron)
+	if err != nil {
+		zap.S().Errorln("解析时间规则异常: ", err)
+		return nil
+	}
+
+	return func(c context.Context) error {
+		a := make(chan bool)
+		defer close(a)
+
+		zap.S().Infoln("待机时间控制器")
+		defer zap.S().Infoln("待机时间控制器结束")
+
+		// 等待组
+		var waitGroup sync.WaitGroup
+		defer waitGroup.Wait()
+
+		ticker := time.NewTicker(time.Second)
+		defer ticker.Stop()
+
+		waitGroup.Add(1)
+		go func() {
+			defer waitGroup.Done()
+			for {
+				select {
+				case <-c.Done():
+					return
+				case <-ticker.C:
+					if cronrange.Match(rules, time.Now()) && cronrange.Match(rootRules, time.Now()) {
+						a <- true
+					} else {
+						a <- false
+					}
+				}
+			}
+		}()
+
+		var cancel context.CancelFunc
+		var m sync.Mutex
+		for {
+			select {
+			case <-c.Done():
+				if cancel != nil {
+					cancel()
+					cancel = nil
+				}
+				return nil
+			case r := <-a:
+				if r {
+					if ok := m.TryLock(); ok {
+						ctx, cc := context.WithCancel(context.TODO())
+						cancel = cc
+						waitGroup.Add(1)
+						go func() {
+							defer waitGroup.Done()
+							defer m.Unlock()
+							defer func() { cancel = nil }()
+
+							err := play(ctx)
+							if err != nil {
+								zap.S().Errorln("执行动作异常: ", err)
+								select {
+								case <-ctx.Done():
+									return
+								case <-time.After(time.Minute):
+								}
+							}
+						}()
+					}
+				} else if cancel != nil {
+					cancel()
+					cancel = nil
+				}
+			}
+		}
+	}
+}