mapleafgo/game-driver
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

待机配置

Browse Source
This commit is contained in:
mapleafgo
2024-11-05 18:39:03 +08:00
parent ccfe0d311b
commit 8e2bf7f59b
15 changed files with 384 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
internal/common/link_audio.go Normal file
View File

@@ -0,0 +1,54 @@
package common
import (
"bytes"
"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("音频文件 [%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("音频文件 [%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 LinkAudio(link string) (bgm io.ReadCloser) {
u, err := url.Parse(link)
if err != nil {
log.Println("音频 URL 解析错误: ", err)
} else {
if u.Scheme == "file" {
bgm = open(u.String())
} else if u.Scheme == "http" || u.Scheme == "https" {
bgm = get(u.String())
} else {
log.Printf("不支持的音频文件协议: %v\n", u.String())
}
bgm = toBuffer(bgm)
}
return
}

3
internal/common/stopper.go
View File

@@ -56,3 +56,6 @@ func (g *simpleStopper) Done() <-chan struct{} {
// GlobalStopper 全局停止器
var GlobalStopper Stopper = &simpleStopper{}
// GlobalBgStopper 全局后台停止器
var GlobalBgStopper Stopper = &simpleStopper{}

54
internal/middleware/bgm.go
View File

@@ -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()

6
internal/middleware/json.go
View File

@@ -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()
}
}

6
internal/middleware/stop.go
View File

@@ -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)

61
internal/middleware/ticker.go Normal file
View File

@@ -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()
}
}

45
internal/middleware/timer.go
View File

@@ -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)
}
}
}()

93
internal/routes/background.go Normal file
View File

@@ -0,0 +1,93 @@
package routes
import (
"game-driver/internal/common"
"game-driver/internal/middleware"
"game-driver/internal/schema"
"game-driver/leaf"
"game-driver/pkg/audio"
"github.com/gopxl/beep/v2"
"github.com/gopxl/beep/v2/mp3"
"github.com/gopxl/beep/v2/speaker"
"log"
"time"
)
func timerAction(timestamp int64) <-chan struct{} {
a := make(chan struct{})
go func() {
if timestamp == 0 {
close(a)
} else {
<-time.After(time.Until(time.Unix(timestamp, 0)))
close(a)
}
}()
return a
}
func BackgroundAction(c *leaf.Context) {
payload := leaf.Value[*schema.BackgroundModel](c, middleware.PayloadJSONKey)
if payload.Start != 0 && payload.End != 0 && time.Unix(payload.Start, 0).After(time.Unix(payload.End, 0)) {
log.Println("开始时间大于结束时间")
return
}
if payload.End != 0 {
cancel := leaf.WithDeadline(c, time.Unix(payload.End, 0))
defer cancel()
}
select {
case <-c.Done():
case <-timerAction(payload.Start):
go audioAction(c, payload.Items[0], payload.TimeModel)
}
}
func audioAction(c *leaf.Context, item schema.BackgroundItemModel, root schema.TimeModel) {
if item.Start != 0 && time.Unix(item.Start, 0).Before(time.Unix(root.Start, 0)) {
log.Println("开始时间小于根任务开始时间")
return
}
select {
case <-c.Done():
case <-timerAction(item.Start):
{
log.Println("开始执行后台任务")
data := common.LinkAudio(item.Data)
streamer, format, err := mp3.Decode(data)
if err != nil {
return
}
defer streamer.Close()
s := beep.Resample(4, format.SampleRate, audio.DefaultSampleRate, streamer)
ctrl := &beep.Ctrl{Streamer: s}
done := make(chan struct{})
speaker.Play(beep.Seq(ctrl, beep.Callback(func() {
close(done)
})))
for {
select {
case <-done:
return
case <-c.Done():
{
speaker.Lock()
ctrl.Streamer = nil
speaker.Unlock()
}
}
}
}
}
}

2
internal/routes/command.go
View File

@@ -12,6 +12,8 @@ func Command(d *common.Device) leaf.HandlerFunc {
switch cmd {
case "stop":
common.GlobalStopper.Stop()
case "stop-bg":
common.GlobalBgStopper.Stop()
case "status":
d.PublishStatus()
default:

29
internal/schema/bg.go Normal file
View File

@@ -0,0 +1,29 @@
package schema
type ItemType int
const (
TYPE_AUDIO ItemType = iota
TYPE_VIDEO
TYPE_WEB
TYPE_TTS
TYPE_RELAY
)
type TimeModel struct {
Start int64
End int64
}
type BackgroundItemModel struct {
TimeModel
Interval int64
Type ItemType
Data string
}
type BackgroundModel struct {
JsonModel
TimeModel
Items []BackgroundItemModel
}

4
internal/schema/common.go Normal file
View File

@@ -0,0 +1,4 @@
package schema
type JsonModel interface {
}

1
internal/schema/play.go
View File

@@ -19,6 +19,7 @@ type PrintModal struct {
}
type PlayModal struct {
JsonModel
Timeout int `json:"timeout"`
Power bool `json:"power"`
BGM string `json:"bgm"`

18
internal/server.go
View File

@@ -7,6 +7,7 @@ import (
"game-driver/internal/common"
"game-driver/internal/middleware"
"game-driver/internal/routes"
"game-driver/internal/schema"
"game-driver/leaf"
"game-driver/pkg/tts"
"github.com/eclipse/paho.golang/autopaho"
@@ -113,23 +114,28 @@ func Run() {
// 处理启动报文
router.RegisterHandler(topicPrefix+"play",
middleware.PayloadJSON(),
middleware.PayloadJSON[schema.PlayModal](),
middleware.DeviceLock(device),
middleware.EmergencyStop(),
middleware.EmergencyStop(common.GlobalStopper),
middleware.SoundStart(t),
middleware.RelayMaster(nil),
middleware.TimerAction(t, config.C.Game.MaxTimeout),
middleware.TimeoutOver(config.C.Game.MaxTimeout),
middleware.TickerAction(t),
middleware.PlayBgm(),
func(c *leaf.Context) {
log.Println("接收到启动消息: ", string(c.Payload))
select {
case <-c.Done():
log.Println("程序已关闭")
case <-time.After(10 * time.Second):
log.Println("10s 结束")
log.Println("执行结束")
}
},
)
// 处理后台线程报文
router.RegisterHandler(topicPrefix+"bg",
middleware.PayloadJSON[schema.BackgroundModel](),
middleware.EmergencyStop(common.GlobalBgStopper),
routes.BackgroundAction,
)
// 处理指令
router.RegisterHandler(topicPrefix+"command", routes.Command(device))