go实现go程池https://www.jianshu.com/p/508f5d3b2f59

https://www.cnblogs.com/williamjie/p/9267741.html 牛的很,pool
package main

import (
   "fmt"
   "time"
)

/* 有关Task任务相关定义及操作 */
//定义任务Task类型,每一个任务Task都可以抽象成一个函数
type Task struct {
   f func() error //一个无参的函数类型
}

//通过NewTask来创建一个Task
func NewTask(f func() error) *Task {
   t := Task{
      f: f,
   }

   return &t
}

//执行Task任务的方法
func (t *Task) Execute() {
   t.f() //调用任务所绑定的函数
}

/* 有关协程池的定义及操作 */
//定义池类型
type Pool struct {
   //对外接收Task的入口
   EntryChannel chan *Task

   //协程池最大worker数量,限定Goroutine的个数
   worker_num int

   //协程池内部的任务就绪队列
   JobsChannel chan *Task
}

//创建一个协程池
func NewPool(cap int) *Pool {
   p := Pool{
      EntryChannel: make(chan *Task),
      worker_num:   cap,
      JobsChannel:  make(chan *Task),
   }

   return &p
}

//协程池创建一个worker并且开始工作
func (p *Pool) worker(work_ID int) {
   //worker不断的从JobsChannel内部任务队列中拿任务
   for task := range p.JobsChannel {
      //如果拿到任务,则执行task任务
      task.Execute()
      fmt.Println("worker ID ", work_ID, " 执行完毕任务")
   }
}

//让协程池Pool开始工作
func (p *Pool) Run() {
   //1,首先根据协程池的worker数量限定,开启固定数量的Worker,
   //  每一个Worker用一个Goroutine承载
   for i := 0; i < p.worker_num; i++ {
      go p.worker(i)
   }

   //2, 从EntryChannel协程池入口取外界传递过来的任务
   //   并且将任务送进JobsChannel中
   for task := range p.EntryChannel {
      p.JobsChannel <- task
   }

   //3, 执行完毕需要关闭JobsChannel
   close(p.JobsChannel)

   //4, 执行完毕需要关闭EntryChannel
   close(p.EntryChannel)
}

//主函数
func main() {
   //创建一个Task
   t := NewTask(func() error {
      fmt.Println(time.Now())
      return nil
   })

   //创建一个协程池,最大开启3个协程worker
   p := NewPool(3)

   //开一个协程 不断的向 Pool 输送打印一条时间的task任务
   go func() {
      for {
         p.EntryChannel <- t
      }
   }()

   //启动协程池p
   p.Run()

}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值