46 lines
806 B
Go
46 lines
806 B
Go
package middleware
|
|
|
|
import (
|
|
"game-driver/internal/common"
|
|
"game-driver/leaf"
|
|
"go.uber.org/zap"
|
|
"sync"
|
|
)
|
|
|
|
// EmergencyStop 紧急停止中间件
|
|
func EmergencyStop(stopper common.Stopper) leaf.HandlerFunc {
|
|
return func(c *leaf.Context) {
|
|
cancel := leaf.WithCancel(c)
|
|
defer stopper.Reset()
|
|
|
|
zap.S().Infoln("监听停止信号")
|
|
defer zap.S().Infoln("结束停止信号监听")
|
|
|
|
// 等待组
|
|
var wait sync.WaitGroup
|
|
defer wait.Wait()
|
|
|
|
// 结束信号通道
|
|
a := make(chan struct{})
|
|
// 发送结束信号
|
|
defer close(a)
|
|
|
|
wait.Add(1)
|
|
go func() {
|
|
defer wait.Done()
|
|
|
|
select {
|
|
case <-a:
|
|
case <-stopper.Done():
|
|
{
|
|
zap.S().Infoln("停止信号触发")
|
|
cancel()
|
|
leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndStop)
|
|
}
|
|
}
|
|
}()
|
|
|
|
c.Next()
|
|
}
|
|
}
|