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