博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
go 协程池使用提示
阅读量:4099 次
发布时间:2019-05-25

本文共 2176 字,大约阅读时间需要 7 分钟。

main.go

package mainimport (   "fmt"   conc "github.com/micro/examples/Test/Concurrence"   "runtime"   //"runtime"   "time")//定义一个实现Job接口的数据type Score struct {   Num int}//定义对数据的处理func (s *Score) Do() {   fmt.Println("num:", s.Num)   //time.Sleep(1 * 1 * time.Second)}func main() {   start := time.Now()   //some func or operation   num := 100 *100*20   // debug.SetMaxThreads(num + 1000) //设置最大线程数   // 注册工作池,传入任务   // 参数1 worker并发个数   p := conc.NewWorkerPool(num)   p.Run()   //写入一亿条数据   datanum := 100  * 100 //* 100  // * 100   go func() {      for i := 1; i <= datanum; i++ {         sc := &Score{Num: i}         p.JobQueue <- sc //数据传进去会被自动执行Do()方法,具体对数据的处理自己在Do()方法中定义      }   }()   cost := time.Since(start)   fmt.Printf("cost=[%s]",cost)   //循环打印输出当前进程的Goroutine 个数   for {      fmt.Println("runtime.NumGoroutine() :", runtime.NumGoroutine())      time.Sleep(2 * time.Second)   }}

Concurrence.go

package Concurrenceimport "fmt"// --------------------------- Job ---------------------type Job interface {   Do()}// --------------------------- Worker ---------------------type Worker struct {   JobQueue chan Job}func NewWorker() Worker {   return Worker{JobQueue: make(chan Job)}}func (w Worker) Run(wq chan chan Job) {   go func() {      for {         wq <- w.JobQueue         select {         case job := <-w.JobQueue:            //fmt.Println("job.do....")            job.Do()         }      }   }()}// --------------------------- WorkerPool ---------------------type WorkerPool struct {   workerlen   int   JobQueue    chan Job   WorkerQueue chan chan Job}func NewWorkerPool(workerlen int) *WorkerPool {   return &WorkerPool{      workerlen:   workerlen,      JobQueue:    make(chan Job),      WorkerQueue: make(chan chan Job, workerlen),   }}func (wp *WorkerPool) Run() {   fmt.Println("初始化worker")   //初始化worker   for i := 0; i < wp.workerlen; i++ {      worker := NewWorker()      worker.Run(wp.WorkerQueue)   }   // 循环获取可用的worker,往worker中写job   go func() {      for {         select {         case job := <-wp.JobQueue:            fmt.Println("循环获取可用的worker,往worker中写job")            worker := <-wp.WorkerQueue            worker <- job         }      }   }()}

 

转载地址:http://ikwsi.baihongyu.com/

你可能感兴趣的文章
Android中的Binder(二)
查看>>
Framework之View的工作原理(一)
查看>>
Web应用架构
查看>>
设计模式之策略模式
查看>>
深究Java中的RMI底层原理
查看>>
用idea创建一个maven web项目
查看>>
Kafka
查看>>
9.1 为我们的角色划分权限
查看>>
维吉尼亚之加解密及破解
查看>>
DES加解密
查看>>
TCP/IP协议三次握手与四次握手流程解析
查看>>
PHP 扩展开发 : 编写一个hello world !
查看>>
inet_ntoa、 inet_aton、inet_addr
查看>>
用模板写单链表
查看>>
用模板写单链表
查看>>
链表各类操作详解
查看>>
C++实现 简单 单链表
查看>>
数据结构之单链表——C++模板类实现
查看>>
Linux的SOCKET编程 简单演示
查看>>
正则匹配函数
查看>>