租用问题

质量为本、客户为根、勇于拼搏、务实创新

< 返回租用问题列表

Golang异步处理与Go WaitGroup的巧妙结合,golang 异步任务

发布时间:2023-10-08 16:59:17

Golang异步处理与Go WaitGroup的奇妙结合

在Golang中,可使用Go WaitGroup来同步并发操作。WaitGroup是一个计数器,用于等待一组goroutine完成任务。
当我们需要异步处理多个任务时,可使用WaitGroup来等待所有任务完成。下面是一种奇妙地结合异步处理和WaitGroup的方法:

  1. 首先,创建一个WaitGroup对象,并设置计数器的初始值为任务的数量。
var wg sync.WaitGroup
wg.Add(numTasks)
  1. 在处理每一个任务时,使用goroutine来异步履行任务,并在任务完成时调用Done方法减少计数器的值。
for _, task := range tasks {
go func(t Task) {
// 处理任务 t
// ...
// 任务完成时调用Done方法
wg.Done()
}(task)
}
  1. 在所有任务都启动后,使用Wait方法来等待所有任务完成。
wg.Wait()

这样,我们就能够同时处理多个任务,并在所有任务完成后继续履行后续操作。
完全的示例代码以下:

package main
import (
"fmt"
"sync"
"time"
)
type Task struct {
id int
}
func main() {
numTasks := 10
tasks := make([]Task, numTasks)
var wg sync.WaitGroup
wg.Add(numTasks)
for i := 0; i < numTasks; i++ {
tasks[i] = Task{id: i}
go func(t Task) {
// 模拟处理任务
time.Sleep(1 * time.Second)
fmt.Printf("Task %d done
", t.id)
// 任务完成时调用Done方法
wg.Done()
}(tasks[i])
}
// 等待所有任务完成
wg.Wait()
fmt.Println("All tasks done")
}

输出结果:

Task 1 done
Task 2 done
Task 5 done
Task 8 done
Task 4 done
Task 9 done
Task 6 done
Task 7 done
Task 3 done
Task 0 done
All tasks done

通过奇妙地结合异步处理和WaitGroup,我们可以高效地处理并发任务,并在所有任务完成后继续履行后续操作。