Golang中如何修复WaitGroup计数器为负数的问题

通常，如果我在以下代码中使用 wgAllScan.Add(1)：

go func(wg *sync.WaitGroup) {
	for res := range resultsCh {
		wgAllScan.Add(1)
		fmt.Println(wgAllScan)

		if res.StateProc == 0 {
			host, err := ParseXMLResultFile(res.FileScan)
			if err != nil {
				log.Printf("Parsing file not done: %s", err)
			}

			results = append(results, host)

			err = os.Remove(res.FileScan)
			if err != nil {
				log.Fatal("Can't delete file", res.FileScan, err)
			}
		}

那么 goroutine 会按需执行，但随后我的主函数将停止等待 goroutine 并完成其执行。

如果我在第一个循环和第二个循环中都添加 wgAllScan.Add(1)，那么该函数会运行多次而不完成，但计数器永远不会完成，并且再次发生泄漏。

我不明白该如何继续。

问题出在 WaitGroup 的使用方式上。你的代码存在几个关键问题：

1. WaitGroup 计数器管理不一致：wgAllScan.Add(1) 在循环中调用，但 wgAllScan.Done() 在另一个 goroutine 中调用，这可能导致计数器不同步
2. 并发访问共享数据：results 切片在多个 goroutine 中被并发修改，没有同步保护
3. 通道使用问题：resultsCh 通道的关闭和读取逻辑不清晰

以下是修复后的代码：

func Scan(cmdSc *pb.ScanCommandReq) []*models.Cmdrun {
    wgAllScan := &sync.WaitGroup{}
    var results []*models.Cmdrun
    var mu sync.Mutex // 添加互斥锁保护共享数据

    CheckFolder(cmdSc.TrgDst)
    go cmdTicker()
    
    ports := FPorts()
    
    // 先计算需要等待的goroutine数量
    portCount := len(ports["segFir"])
    wgAllScan.Add(portCount)
    
    for _, port := range ports["segFir"] {
        scanFileRes := fmt.Sprintf("%s/%s%ddef", 
            cmdSc.TrgDst, cmdSc.TrgDst, time.Now().UnixNano())
        
        fileXMLCreate(scanFileRes)
        
        go func(port string, file string) {
            defer wgAllScan.Done() // 确保Done一定会被调用
            runScan(newScan, file, port)
            
            // 处理结果
            if res.StateProc == 0 {
                host, err := ParseXMLResultFile(res.FileScan)
                if err != nil {
                    log.Printf("Parsing file not done: %s", err)
                    return
                }
                
                mu.Lock() // 加锁保护results
                results = append(results, host)
                mu.Unlock()
                
                err = os.Remove(res.FileScan)
                if err != nil {
                    log.Printf("Can't delete file %s: %v", res.FileScan, err)
                }
            }
        }(port, scanFileRes)
    }
    
    wgAllScan.Wait()
    return results
}

或者，如果你需要保持使用通道的模式，这里是一个使用通道的正确版本：

func Scan(cmdSc *pb.ScanCommandReq) []*models.Cmdrun {
    wgAllScan := &sync.WaitGroup{}
    var results []*models.Cmdrun
    
    CheckFolder(cmdSc.TrgDst)
    go cmdTicker()
    
    ports := FPorts()
    resultsCh := make(chan *models.Cmdrun, len(ports["segFir"])) // 缓冲通道
    
    // 启动结果收集goroutine
    go func() {
        for res := range resultsCh {
            results = append(results, res)
        }
    }()
    
    for _, port := range ports["segFir"] {
        scanFileRes := fmt.Sprintf("%s/%s%ddef", 
            cmdSc.TrgDst, cmdSc.TrgDst, time.Now().UnixNano())
        
        fileXMLCreate(scanFileRes)
        
        wgAllScan.Add(1)
        go func(port string, file string) {
            defer wgAllScan.Done()
            
            // 假设runScan返回结果
            res := runScan(newScan, file, port)
            
            if res.StateProc == 0 {
                host, err := ParseXMLResultFile(res.FileScan)
                if err != nil {
                    log.Printf("Parsing file not done: %s", err)
                    return
                }
                
                resultsCh <- host // 发送结果到通道
                
                err = os.Remove(res.FileScan)
                if err != nil {
                    log.Printf("Can't delete file %s: %v", res.FileScan, err)
                }
            }
        }(port, scanFileRes)
    }
    
    // 等待所有goroutine完成
    wgAllScan.Wait()
    close(resultsCh) // 关闭通道
    
    return results
}

关键修复点：

1. Add() 必须在启动 goroutine 之前调用，最好在循环外一次性添加
2. 使用 defer wgAllScan.Done() 确保即使 goroutine 崩溃也会调用 Done()
3. 保护共享数据：使用互斥锁或通道来安全地访问 results 切片
4. 避免重复使用 WaitGroup：每次函数调用都应该创建新的 WaitGroup 实例

计数器为负数的根本原因是 Done() 被调用的次数超过了 Add() 的次数，通常是由于 goroutine 异常退出或逻辑错误导致的。