server/beego
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

task manager graceful shutdown support

Browse Source
This commit is contained in:
guoxingyong 2021-05-26 10:19:03 +08:00
parent 9327e2b026
commit 93b73ddb34
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
task/task.go
View File

@ -472,6 +472,11 @@ func ClearTask() {
globalTaskManager.ClearTask() globalTaskManager.ClearTask()
} }
// GracefulShutdown wait all task done
func GracefulShutdown() <-chan struct{} {
return globalTaskManager.GracefulShutdown()
}
// StartTask start all tasks // StartTask start all tasks
func (m *taskManager) StartTask() { func (m *taskManager) StartTask() {
m.taskLock.Lock() m.taskLock.Lock()
@ -586,7 +591,7 @@ func (m *taskManager) StopTask() {
}() }()
} }
// StopTask stop all tasks // GracefulShutdown wait all task done
func (m *taskManager) GracefulShutdown() <-chan struct{} { func (m *taskManager) GracefulShutdown() <-chan struct{} {
done := make(chan struct{}, 0) done := make(chan struct{}, 0)
go func() { go func() {