fix shutdown stuck or panic

internal/task/impl.go

@@ -12,6 +12,8 @@ func (t *Task) addCallback(about string, fn func(), waitSubTasks bool) {
 	defer t.mu.Unlock()
 	if t.callbacks == nil {
 		t.callbacks = make(map[*Callback]struct{})
+	}
+	if t.callbacksDone == nil {
 		t.callbacksDone = make(chan struct{})
 	}
 	t.callbacks[&Callback{fn, about, waitSubTasks}] = struct{}{}
@@ -28,14 +30,15 @@ func (t *Task) addChildCount() {
 }
 
 func (t *Task) subChildCount() {
-	if atomic.AddUint32(&t.children, ^uint32(0)) == 0 {
+	switch atomic.AddUint32(&t.children, ^uint32(0)) {
+	case 0:
 		close(t.childrenDone)
+	case ^uint32(0):
+		panic("negative child count")
 	}
 }
 
 func (t *Task) runCallbacks() {
-	t.mu.Lock()
-	defer t.mu.Unlock()
 	if len(t.callbacks) == 0 {
 		return
 	}

internal/task/task.go

@@ -117,9 +117,7 @@ func (t *Task) finish(reason any) {
 			Strs("callbacks", t.listCallbacks()).
 			Msg("Timeout waiting for callbacks to finish")
 	}
-	if t.finished != nil {
+	close(t.finished)
-		close(t.finished)
-	}
 	if t == root {
 		return
 	}

internal/task/utils.go

@@ -30,9 +30,12 @@ func RootTask(name string, needFinish ...bool) *Task {
 }
 
 func newRoot() *Task {
-	t := &Task{name: "root"}
+	t := &Task{
+		name:         "root",
+		childrenDone: make(chan struct{}),
+		finished:     make(chan struct{}),
+	}
 	t.ctx, t.cancel = context.WithCancelCause(context.Background())
-	t.callbacks = make(map[*Callback]struct{})
 	return t
 }