golang 实现workqueue

最新推荐文章于 2025-07-28 14:00:29 发布

原创最新推荐文章于 2025-07-28 14:00:29 发布 · 932 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang

golang 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何利用Go的goroutine和channel轻松构建一个工作队列，以并行执行任务。通过创建一个channel存储任务，多个goroutine从channel中取出任务执行，对比Java展现了Go语言在并发处理上的便利性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

强大的gorouting 和channle可以很轻松的实现一个 workqueue 来并行的执行任务，思路：可以将将要执行的任务放到一channel，然后创建多个go routing 去chnanel 取任务执行, 可以看下是不是要比java方便好多。

package main


import (
    "fmt"
    "sync"
)


type DoWorkPieceFunc func(piece int)


// Parallelize is a very simple framework that allow for parallelizing
// N independent pieces of work.
func Parallelize(workers, pieces int, doWorkPiece DoWorkPieceFunc) {
    toProcess := make(chan int, pieces)
    for i := 0; i < pieces; i++ {
        toProcess <- i
    }
    close(toProcess)


    wg := sync.WaitGroup{}
    wg.Add(workers)
    for i := 0; i < workers; i++ {
        go func(i int) {
            defer wg.Done()
            fmt.Printf("worker %v is working\n", i)
            for piece := range toProcess {
                fmt.Printf("to be done no: %v\n", len(toProcess))
                doWorkPiece(piece)
            }
        }(i)
    }
    wg.Wait()
}
func main() {
    processNode := func(i int) {
        fmt.Printf("process i:%v\n", i)
    }
    Parallelize(16, 100, processNode)
}
~