修复投影仪控制

internal/routes/standby_ctrl/pause.go

@@ -0,0 +1,66 @@
+package standby_ctrl
+
+import (
+	"context"
+	"game-driver/internal/common"
+	"go.uber.org/zap"
+	"sync"
+)
+
+func Pause(ps *common.PauseSub, isPause bool, play func(c context.Context) error) func(c context.Context) error {
+	return func(c context.Context) error {
+		var cancel context.CancelFunc
+		run := true
+
+		if isPause {
+			p := make(chan int8)
+			defer close(p)
+
+			ps.Add(p)
+			defer ps.Remove(p)
+
+			zap.S().Infoln("待机控制器")
+			defer zap.S().Infoln("待机控制器结束")
+
+			// 等待组
+			var wait sync.WaitGroup
+			defer wait.Wait()
+
+			wait.Add(1)
+			go func() {
+				defer wait.Done()
+				for {
+					select {
+					case <-c.Done():
+						return
+					case v := <-p:
+						if v == 1 {
+							zap.S().Infoln("待机控制器 Pause 触发")
+							run = false
+							cancel()
+						} else {
+							zap.S().Infoln("待机控制器 Resume 触发")
+							run = true
+						}
+					}
+				}
+			}()
+		}
+
+		for {
+			select {
+			case <-c.Done():
+				return nil
+			default:
+				if run {
+					nc, cc := context.WithCancel(c)
+					cancel = cc
+					err := play(nc)
+					if err != nil {
+						zap.S().Infoln("执行后续操作异常: ", err)
+					}
+				}
+			}
+		}
+	}
+}

internal/routes/standby_ctrl/time.go

@@ -0,0 +1,91 @@
+package standby_ctrl
+
+import (
+	"context"
+	"github.com/go-pkgz/cronrange"
+	"go.uber.org/zap"
+	"sync"
+	"time"
+)
+
+// Time 时间控制器
+func Time(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)
+
+		// 等待组
+		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
+				}
+			}
+		}
+	}
+}