当前位置： > > > > 如何等待缓冲通道（信号量）为空？

如何等待缓冲通道（信号量）为空？

来源：stackoverflow
2024-04-26 10:54:38
0浏览
收藏

今天米云给大家带来了《如何等待缓冲通道（信号量）为空？》，其中涉及到的知识点包括等等，无论你是小白还是老手，都适合看一看哦~有好的建议也欢迎大家在评论留言，若是看完有所收获，也希望大家能多多点赞支持呀！一起加油学习~

问题内容

我有一个整数切片，它们是同时操作的：

ints := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}

我使用缓冲通道作为信号量，以获得同时运行的 go 例程的上限：

sem := make(chan struct{}, 2)

for _, i := range ints {
  // acquire semaphore
  sem <- struct{}{}

  // start long running go routine
  go func(id int, sem chan struct{}) {
    // do something

    // release semaphore
    <- sem
  }(i, sem)
}

上面的代码在到达最后一个或最后两个整数之前工作得很好，因为程序在最后的 go 例程完成之前结束。

问题：如何等待缓冲通道耗尽？

解决方案

您不能以这种方式使用信号量（在本例中为通道）。当您处理值和调度更多 goroutine 时，无法保证它不会为空。在这种情况下，这不是一个问题，因为您是同步分派工作，但因为没有无竞争的方法来检查通道的长度，所以没有基元来等待通道的长度达到 0。

使用 sync.waitgroup 等待所有 goroutine 完成

sem := make(chan struct{}, 2)

var wg sync.waitgroup

for _, i := range ints {
    wg.add(1)
    // acquire semaphore
    sem <- struct{}{}
    // start long running go routine
    go func(id int) {
        defer wg.done()
        // do something
        // release semaphore
        <-sem
    }(i)
}

wg.wait()

使用“工作池”来处理您的数据。它比为 each int 运行 goroutine、为其中的变量分配内存等等更便宜……

ints := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}

ch := make(chan int)

var wg sync.WaitGroup

// run worker pool
for i := 2; i > 0; i-- {
    wg.Add(1)

    go func() {
        defer wg.Done()

        for id := range ch {
            // do something
            fmt.Println(id)
        }
    }()
}

// send ints to workers
for _, i := range ints {
    ch <- i
}

close(ch)

wg.Wait()

理论要掌握，实操不能落！以上关于《如何等待缓冲通道（信号量）为空？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注米云公众号吧！