Golang并发编程：使用Go WaitGroup实现任务调度器- 技术经验 -卓越飞翔博客

Golang并发编程：使用Go WaitGroup实现任务调度器

引言
在Golang中，实现并发编程可以大大提升程序的性能和效率。而任务调度器是并发编程中非常重要的一部分，它可以用于调度并发任务的执行顺序和同步任务的完成。本文将介绍如何使用Golang中的WaitGroup来实现一个简单的任务调度器，并给出具体的代码示例。
WaitGroup简介
WaitGroup是Golang包sync中的一个重要类型，它提供了并发任务的同步和等待功能。WaitGroup有三个主要的方法：Add、Done和Wait。

Add(n int)：添加 n 个并发任务到WaitGroup中。
Done()：标记一个任务完成。
Wait()：等待所有任务完成，并阻塞当前的goroutine。

WaitGroup内部维护了一个计数器，用来记录未完成的任务数量。当调用Add方法时，计数器加上指定的值；当调用Done方法时，计数器减1；当调用Wait方法时，如果计数器大于0，当前goroutine会被阻塞，直到计数器归零。

任务调度器实现
下面是一个使用WaitGroup实现任务调度器的示例代码：

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Printf("Worker %d starting
", id)
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done
", id)
}

func main() {
    var wg sync.WaitGroup
    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }
    wg.Wait()
    fmt.Println("All workers done")
}

在这个示例中，我们使用一个for循环启动5个worker goroutine，并通过调用wg.Add(1)来将任务数量添加到WaitGroup中。在worker函数中，使用defer关键字来确保在函数返回前调用wg.Done()，来减少计数器的值。最后，使用wg.Wait()来等待所有任务完成。

运行结果
运行这个示例代码，你会看到如下的输出结果：

Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 4 starting
Worker 5 starting
Worker 3 done
Worker 1 done
Worker 2 done
Worker 5 done
Worker 4 done
All workers done

可以看到，所有的worker goroutine被并发执行，但是输出的顺序并不一定是按照启动顺序来的，这是因为goroutine的调度是由Go运行时进行的。

总结
使用WaitGroup可以方便地实现并发任务的调度和同步。通过调用Add方法添加任务数量、调用Done方法标记任务完成、调用Wait方法来等待任务完成，可以实现任务的同步和顺序控制。在实际应用中，任务调度器可以用于处理并发任务、限制并发数量等场景，帮助我们更好地利用Golang的并发能力。

通过本文的介绍和示例代码，相信你对如何使用Golang的WaitGroup实现任务调度器有了更深的理解。希望本文对你学习和使用Golang的并发编程有所帮助！

相关推荐