fix wg.Wait() bug

yhy0 · Apr 15, 2024 · 3fc391e · 3fc391e
1 parent 9de06eb
commit 3fc391e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/task/scan.go b/pkg/task/scan.go
@@ -139,7 +139,8 @@ func (t *Task) DoneWg(host string) {
 
 func (t *Task) WaitWg(host string) {
     t.WgLock.Lock() // 保护对ScanTask映射的访问
-    defer t.WgLock.Unlock()
+    wg := t.ScanTask[host].Wg
+    t.WgLock.Unlock() // 解锁，以便其他goroutine可以操作WaitGroup
 
-    t.ScanTask[host].Wg.Wait()
+    wg.Wait() // 现在可以安全地等待，而不会持有锁
 }