task manager graceful shutdown support

This commit is contained in:
guoxingyong 2021-05-26 10:19:03 +08:00
parent 9327e2b026
commit 93b73ddb34

View File

@ -472,6 +472,11 @@ func ClearTask() {
globalTaskManager.ClearTask()
}
// GracefulShutdown wait all task done
func GracefulShutdown() <-chan struct{} {
return globalTaskManager.GracefulShutdown()
}
// StartTask start all tasks
func (m *taskManager) StartTask() {
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{} {
done := make(chan struct{}, 0)
go func() {