38 lines
705 B
Go
38 lines
705 B
Go
package middleware
|
|
|
|
import (
|
|
"game-driver/internal/common"
|
|
"game-driver/leaf"
|
|
"game-driver/pkg/logger"
|
|
)
|
|
|
|
// EmergencyStop 紧急停止中间件
|
|
func EmergencyStop(stopper common.Stopper) leaf.HandlerFunc {
|
|
return func(c *leaf.Context) {
|
|
cancel := leaf.WithCancel(c)
|
|
defer stopper.Reset()
|
|
|
|
// 结束信号通道
|
|
a := make(chan struct{})
|
|
// 发送结束信号
|
|
defer close(a)
|
|
|
|
logger.Infoln("监听停止信号")
|
|
go func() {
|
|
defer logger.Infoln("结束停止信号监听")
|
|
|
|
select {
|
|
case <-a:
|
|
case <-stopper.Done():
|
|
{
|
|
logger.Infoln("停止信号触发")
|
|
cancel()
|
|
leaf.WithValue[leaf.EndType](c, leaf.EndKey, leaf.EndStop)
|
|
}
|
|
}
|
|
}()
|
|
|
|
c.Next()
|
|
}
|
|
}
|