待机配置

internal/middleware/bgm.go

@@ -1,63 +1,19 @@
 package middleware
 
 import (
-	"bytes"
+	"game-driver/internal/common"
 	"game-driver/internal/schema"
 	"game-driver/leaf"
 	"game-driver/pkg/audio"
-	"io"
-	"log"
-	"net/http"
-	"net/url"
-	"os"
-	"strings"
 )
 
-func open(u string) io.ReadCloser {
-	p, _ := strings.CutPrefix(u, "file://")
-	f, e := os.Open(p)
-	if e != nil {
-		log.Printf("BGM 文件 [%v] 打开错误: %v\n", u, e)
-		return nil
-	}
-	return f
-}
-
-func get(u string) io.ReadCloser {
-	resp, e := http.Get(u)
-	if e != nil {
-		log.Printf("BGM 文件 [%v] 下载失败: %v\n", u, e)
-		return nil
-	}
-	return resp.Body
-}
-
-func toBuffer(b io.ReadCloser) io.ReadCloser {
-	data := &bytes.Buffer{}
-	_, _ = data.ReadFrom(b)
-	defer b.Close()
-	return io.NopCloser(data)
-}
-
 func PlayBgm() leaf.HandlerFunc {
 	return func(c *leaf.Context) {
 		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
-		u, err := url.Parse(pm.BGM)
-		if err != nil {
-			log.Println("BGM URL 解析错误: ", err)
-		} else {
-			var bgm io.ReadCloser
-			if u.Scheme == "file" {
-				bgm = open(u.String())
-			} else if u.Scheme == "http" || u.Scheme == "https" {
-				bgm = get(u.String())
-			} else {
-				log.Printf("不支持的BGM文件协议: %v\n", u.String())
-			}
-			// 获取到了资源进行播放
-			if bgm != nil {
-				go audio.PlayMP3(c, toBuffer(bgm))
-			}
+
+		bgm := common.LinkAudio(pm.BGM)
+		if bgm != nil {
+			go audio.PlayMP3(c, bgm)
 		}
 
 		c.Next()

internal/middleware/json.go

@@ -12,14 +12,14 @@ type JSONKey string
 const PayloadJSONKey JSONKey = "payload_json"
 
 // PayloadJSON 解析报文
-func PayloadJSON() leaf.HandlerFunc {
+func PayloadJSON[T schema.JsonModel]() leaf.HandlerFunc {
 	return func(c *leaf.Context) {
-		pm := &schema.PlayModal{}
+		pm := new(T)
 		err := json.Unmarshal(c.Payload, pm)
 		if err != nil {
 			log.Panicf("报文解析错误: %v\n", err)
 		}
-		leaf.WithValue[*schema.PlayModal](c, PayloadJSONKey, pm)
+		leaf.WithValue[*T](c, PayloadJSONKey, pm)
 		c.Next()
 	}
 }

internal/middleware/stop.go

@@ -6,10 +6,10 @@ import (
 )
 
 // EmergencyStop 紧急停止中间件
-func EmergencyStop() leaf.HandlerFunc {
+func EmergencyStop(stopper common.Stopper) leaf.HandlerFunc {
 	return func(c *leaf.Context) {
 		cancel := leaf.WithCancel(c)
-		defer common.GlobalStopper.Reset()
+		defer stopper.Reset()
 
 		// 结束信号通道
 		a := make(chan struct{})
@@ -19,7 +19,7 @@ func EmergencyStop() leaf.HandlerFunc {
 		go func() {
 			select {
 			case <-a:
-			case <-common.GlobalStopper.Done():
+			case <-stopper.Done():
 				{
 					cancel()
 					leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndStop)

internal/middleware/ticker.go

@@ -0,0 +1,61 @@
+package middleware
+
+import (
+	"game-driver/internal/schema"
+	"game-driver/leaf"
+	"game-driver/pkg/tts"
+	"sync"
+	"time"
+)
+
+func TickerAction(t *tts.AliTTS) leaf.HandlerFunc {
+	return func(c *leaf.Context) {
+		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
+
+		// 构建打印和语音播报的时间映射
+		printMap := make(map[int]schema.PrintModal, len(pm.Print))
+		for _, p := range pm.Print {
+			printMap[p.Time] = p
+		}
+		ttsMap := make(map[int]schema.TTSTimer, len(pm.TTS.Timer))
+		for _, t := range pm.TTS.Timer {
+			ttsMap[t.Time] = t
+		}
+
+		// 等待组
+		var wait sync.WaitGroup
+		defer wait.Wait()
+
+		// 结束信号通道
+		a := make(chan struct{})
+		// 发送结束信号
+		defer close(a)
+
+		go func() {
+			start := time.Now()
+			// 等待结束
+			wait.Add(1)
+			defer wait.Done()
+			// 定时器
+			ticker := time.NewTicker(time.Second)
+			defer ticker.Stop()
+			for over := false; !over; {
+				select {
+				case <-a:
+					over = true
+				case m := <-ticker.C:
+					{
+						s := int(m.Sub(start).Seconds())
+						if _, ok := printMap[s]; ok {
+							//TODO: 屏幕打印
+						}
+						if to, ok := ttsMap[s]; ok {
+							t.Sound(to.Value)
+						}
+					}
+				}
+			}
+		}()
+		c.Next()
+	}
+}

internal/middleware/timer.go

@@ -3,26 +3,15 @@ package middleware
 import (
 	"game-driver/internal/schema"
 	"game-driver/leaf"
-	"game-driver/pkg/tts"
 	"sync"
 	"time"
 )
 
-// TimerAction 定时器中间件，用于定时触发屏幕打印和语音播报。 t 是语音播报实例
-func TimerAction(t *tts.AliTTS, maxTimeout int) leaf.HandlerFunc {
+// TimeoutOver 定时器中间件，用于定时触发屏幕打印和语音播报。 t 是语音播报实例
+func TimeoutOver(maxTimeout int) leaf.HandlerFunc {
 	return func(c *leaf.Context) {
 		pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
 
-		// 构建打印和语音播报的时间映射
-		printMap := make(map[int]schema.PrintModal, len(pm.Print))
-		for _, p := range pm.Print {
-			printMap[p.Time] = p
-		}
-		ttsMap := make(map[int]schema.TTSTimer, len(pm.TTS.Timer))
-		for _, t := range pm.TTS.Timer {
-			ttsMap[t.Time] = t
-		}
-
 		// 定时器
 		var timer *time.Timer
 		if pm.Timeout != 0 {
@@ -43,34 +32,16 @@ func TimerAction(t *tts.AliTTS, maxTimeout int) leaf.HandlerFunc {
 
 		cancel := leaf.WithCancel(c)
 		go func() {
-			start := time.Now()
 			// 等待结束
 			wait.Add(1)
 			defer wait.Done()
-			// 定时器
-			ticker := time.NewTicker(time.Second)
-			defer ticker.Stop()
 			// 结束标志
-			for over := false; !over; {
-				select {
-				case <-a:
-					over = true
-				case <-timer.C: // 定时器结束
-					{
-						cancel()
-						over = true
-						leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndTimer)
-					}
-				case m := <-ticker.C:
-					{
-						s := int(m.Sub(start).Seconds())
-						if _, ok := printMap[s]; ok {
-							//TODO: 屏幕打印
-						}
-						if to, ok := ttsMap[s]; ok {
-							t.Sound(to.Value)
-						}
-					}
+			select {
+			case <-a:
+			case <-timer.C: // 定时器结束
+				{
+					cancel()
+					leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndTimer)
 				}
 			}
 		}()