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-06 15:44:35 +08:00
parent 8e2bf7f59b
commit ab0678aa3b
14 changed files with 317 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
internal/common/link_audio.go
View File

@@ -1,54 +0,0 @@
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
}

38
internal/middleware/bgm.go
View File

@@ -1,19 +1,51 @@
package middleware
import (
"game-driver/internal/common"
"game-driver/internal/schema"
"game-driver/leaf"
"game-driver/pkg/audio"
"github.com/gopxl/beep/v2/speaker"
"log"
"sync"
)
func PlayBgm() leaf.HandlerFunc {
return func(c *leaf.Context) {
pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
bgm := common.LinkAudio(pm.BGM)
bgm := audio.LinkAudio(pm.BGM)
if bgm != nil {
go audio.PlayMP3(c, bgm)
// 等待组
var wait sync.WaitGroup
defer wait.Wait()
// 结束信号通道
a := make(chan struct{})
// 发送结束信号
defer close(a)
go func() {
// 等待结束
wait.Add(1)
defer wait.Done()
ctrl, closer := audio.PlayBgmMP3(bgm)
defer closer()
if ctrl == nil {
log.Println("播放背景音乐失败")
return
}
select {
case <-a:
{
speaker.Lock()
ctrl.Streamer = nil
speaker.Unlock()
return
}
}
}()
}
c.Next()

8
internal/middleware/sound_start.go
View File

@@ -7,17 +7,17 @@ import (
)
// SoundStart 开始词播报
func SoundStart(t *tts.AliTTS) leaf.HandlerFunc {
func SoundStart() leaf.HandlerFunc {
return func(c *leaf.Context) {
pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
t.Sound(pm.TTS.Start)
tts.DefaultTTS.Sound(pm.TTS.Start)
defer func() {
switch leaf.Value[leaf.EndType](c, leaf.EndKey) {
case leaf.EndTimer:
t.Sound(pm.TTS.End)
tts.DefaultTTS.Sound(pm.TTS.End)
case leaf.EndStop:
t.Sound(pm.TTS.Stop)
tts.DefaultTTS.Sound(pm.TTS.Stop)
}
}()

4
internal/middleware/ticker.go
View File

@@ -8,7 +8,7 @@ import (
"time"
)
func TickerAction(t *tts.AliTTS) leaf.HandlerFunc {
func TickerAction() leaf.HandlerFunc {
return func(c *leaf.Context) {
pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
@@ -50,7 +50,7 @@ func TickerAction(t *tts.AliTTS) leaf.HandlerFunc {
//TODO: 屏幕打印
}
if to, ok := ttsMap[s]; ok {
t.Sound(to.Value)
tts.DefaultTTS.Sound(to.Value)
}
}
}

93
internal/routes/background.go
View File

@@ -1,93 +0,0 @@
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()
}
}
}
}
}
}

179
internal/routes/wait.go Normal file
View File

@@ -0,0 +1,179 @@
package routes
import (
"game-driver/internal/middleware"
"game-driver/internal/schema"
"game-driver/leaf"
"game-driver/pkg/audio"
"game-driver/pkg/relay"
"game-driver/pkg/tts"
"github.com/gopxl/beep/v2/speaker"
"log"
"sync"
"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 WaitAction(c *leaf.Context) {
payload := leaf.Value[*schema.WaitModel](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):
// 等待组
var wait sync.WaitGroup
defer wait.Wait()
for _, item := range payload.Items {
switch item.Type {
case schema.WaitAudio:
// 执行音乐播放
go func() {
wait.Add(1)
defer wait.Done()
audioAction(c, item, payload.TimeModel)
}()
case schema.WaitTTS:
// 执行TTS播放
go func() {
wait.Add(1)
defer wait.Done()
ttsAction(c, item, payload.TimeModel)
}()
case schema.WaitRelay:
// 执行继电器供电
go func() {
wait.Add(1)
defer wait.Done()
relayAction(c, item, payload.TimeModel)
}()
case schema.WaitVideo:
case schema.WaitWeb:
default:
log.Printf("不支持的类型: %d\n", item.Type)
}
}
}
}
func audioAction(c *leaf.Context, item schema.WaitItemModel, root schema.TimeModel) {
if item.Start != 0 && time.Unix(item.Start, 0).Before(time.Unix(root.Start, 0)) {
log.Println("开始时间小于根任务开始时间")
return
}
if item.End != 0 {
cancel := leaf.WithDeadline(c, time.Unix(item.End, 0))
defer cancel()
}
select {
case <-c.Done():
case <-timerAction(item.Start):
{
log.Println("开始执行后台任务")
data := audio.LinkAudio(item.Data)
ctrl, closer := audio.PlayBgmMP3(data)
defer closer()
if ctrl == nil {
log.Println("播放背景音乐失败")
return
}
select {
case <-c.Done():
{
speaker.Lock()
ctrl.Streamer = nil
speaker.Unlock()
}
}
}
}
}
func ttsAction(c *leaf.Context, item schema.WaitItemModel, root schema.TimeModel) {
if item.Start != 0 && time.Unix(item.Start, 0).Before(time.Unix(root.Start, 0)) {
log.Println("开始时间小于根任务开始时间")
return
}
if item.End != 0 {
cancel := leaf.WithDeadline(c, time.Unix(item.End, 0))
defer cancel()
}
reader, err := tts.DefaultTTS.Get(item.Data)
if err != nil {
log.Println("语音合成异常: ", err)
return
}
select {
case <-c.Done():
case <-timerAction(item.Start):
{
for {
audio.PlayWav(c, reader)
select {
case <-c.Done():
return
case <-time.After(time.Duration(item.Interval) * time.Second):
}
}
}
}
}
func relayAction(c *leaf.Context, item schema.WaitItemModel, root schema.TimeModel) {
if item.Start != 0 && time.Unix(item.Start, 0).Before(time.Unix(root.Start, 0)) {
log.Println("开始时间小于根任务开始时间")
return
}
if item.End != 0 {
cancel := leaf.WithDeadline(c, time.Unix(item.End, 0))
defer cancel()
}
device, err := relay.New(item.Data, nil)
if err != nil {
log.Println("继电器初始化异常: ", err)
return
}
defer device.Close()
select {
case <-c.Done():
case <-timerAction(item.Start):
{
device.On(1)
<-c.Done()
device.Off(1)
}
}
}

29
internal/schema/bg.go
View File

@@ -1,29 +0,0 @@
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
}

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

@@ -0,0 +1,29 @@
package schema
type WaitType int
const (
WaitAudio WaitType = iota
WaitVideo
WaitWeb
WaitTTS
WaitRelay
)
type TimeModel struct {
Start int64
End int64
}
type WaitItemModel struct {
TimeModel
Type WaitType
Data string
Interval int64
}
type WaitModel struct {
JsonModel
TimeModel
Items []WaitItemModel
}

14
internal/server.go
View File

@@ -96,7 +96,7 @@ func Run() {
}
// 构建语音合成对象
t := tts.New(ctx, config.C.Aliyun)
tts.DefaultTTS = tts.New(ctx, config.C.Aliyun)
// 构建继电器对象
//r, err := relay.New(config.C.Relay, func(msg string) {
@@ -117,10 +117,10 @@ func Run() {
middleware.PayloadJSON[schema.PlayModal](),
middleware.DeviceLock(device),
middleware.EmergencyStop(common.GlobalStopper),
middleware.SoundStart(t),
middleware.SoundStart(),
middleware.RelayMaster(nil),
middleware.TimeoutOver(config.C.Game.MaxTimeout),
middleware.TickerAction(t),
middleware.TickerAction(),
middleware.PlayBgm(),
func(c *leaf.Context) {
log.Println("接收到启动消息: ", string(c.Payload))
@@ -130,11 +130,11 @@ func Run() {
}
},
)
// 处理后台线程报文
router.RegisterHandler(topicPrefix+"bg",
middleware.PayloadJSON[schema.BackgroundModel](),
// 处理待机线程报文
router.RegisterHandler(topicPrefix+"wait",
middleware.PayloadJSON[schema.WaitModel](),
middleware.EmergencyStop(common.GlobalBgStopper),
routes.BackgroundAction,
routes.WaitAction,
)
// 处理指令
router.RegisterHandler(topicPrefix+"command", routes.Command(device))