71 lines
1.1 KiB
Go
71 lines
1.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"game-driver/internal/common"
|
|
"game-driver/leaf"
|
|
"go.uber.org/zap"
|
|
"sync"
|
|
)
|
|
|
|
func Pause(ctrl *common.CtrlWait) leaf.HandlerFunc {
|
|
return func(c *leaf.Context) {
|
|
var cancel context.CancelFunc
|
|
|
|
// 获取锚点
|
|
holdPoint := c.Hold()
|
|
|
|
// 等待组
|
|
var wait sync.WaitGroup
|
|
defer wait.Wait()
|
|
|
|
// 结束信号通道
|
|
a := make(chan struct{})
|
|
// 发送结束信号
|
|
defer close(a)
|
|
|
|
run := true
|
|
|
|
// 保存原始的 Context
|
|
originalCtx := c.Context
|
|
|
|
wait.Add(1)
|
|
go func() {
|
|
defer wait.Done()
|
|
zap.S().Infoln("待机控制器")
|
|
|
|
for {
|
|
select {
|
|
case <-originalCtx.Done():
|
|
zap.S().Infoln("待机控制器监听结束")
|
|
return
|
|
case <-ctrl.R:
|
|
{
|
|
zap.S().Infoln("待机控制器 Resume 触发")
|
|
c.Context = originalCtx
|
|
run = true
|
|
}
|
|
case <-ctrl.P:
|
|
{
|
|
zap.S().Infoln("待机控制器 Pause 触发")
|
|
run = false
|
|
cancel()
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-originalCtx.Done():
|
|
return
|
|
default:
|
|
if run {
|
|
cancel = leaf.WithCancel(c)
|
|
c.Resume(holdPoint)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|