基本逻辑完成
This commit is contained in:
32
internal/middleware/stop.go
Normal file
32
internal/middleware/stop.go
Normal file
@@ -0,0 +1,32 @@
|
||||
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()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user