go实现协程池

package main

import (

"fmt"

"sync"

"sync/atomic"

"time"

)

type WorkerPool struct {

workers int

jobs chan func()

wg sync.WaitGroup

}

func NewWorkerPool(workers int) *WorkerPool {

pool := &WorkerPool{

workers: workers,

jobs: make(chan func()),

}

pool.startWorkers()

return pool

}

func (p *WorkerPool) startWorkers() {

for i := 0; i < p.workers; i++ {

go func() {

for job := range p.jobs {

job()

}

}()

}

}

func (p *WorkerPool) Do(job func()) {

p.wg.Add(1)

go func() {

defer p.wg.Done()

p.jobs <- job

}()

}

func (p *WorkerPool) Wait() {

p.wg.Wait()

close(p.jobs)

}

func main() {

pool := NewWorkerPool(10)

var count int32

for i := 0; i < 100; i++ {

pool.Do(func() {

time.Sleep(10 * time.Millisecond)

atomic.AddInt32(&count, 1)

})

}

pool.Wait()

fmt.Printf("Count is %d\n", count)

}

相关推荐
lly20240612 分钟前
HTML 元素:构建网页的基础
开发语言
低调小一13 分钟前
LRU缓存科普与实现(Kotlin 与 Swift)
开发语言·缓存·kotlin
爱好学习的青年人14 分钟前
一文详解Go语言字符串
开发语言·后端·golang
Never_Satisfied19 分钟前
在JavaScript / HTML中,line-height是一个CSS属性
javascript·css·html
浅川.2531 分钟前
xtuoj string
开发语言·c++·算法
望获linux34 分钟前
【实时Linux实战系列】实时系统的可观测性:Prometheus 与 Grafana 集成
大数据·linux·服务器·开发语言·网络·操作系统
用户916357440951 小时前
LeetCode热题100——15.三数之和
javascript·算法
skykun1 小时前
都2026年了还在说闭包吗?
javascript
饮水机战神1 小时前
小程序被下架后,我连夜加了个 "安全接口"
前端·javascript
加油吧zkf1 小时前
Python入门:从零开始的完整学习指南
开发语言·前端·python