go工作池,类似如线程池,不断的往池子里丢任务,池子开启多个线程不断的处理任务。go这里其实只是对go chan defer的简单封装。

work/main.go

package work

import "sync"

type Pool struct {
    wg   sync.WaitGroup
    task chan func()
}

func New(size int) *Pool {
    p := &Pool{
        task: make(chan func()),
    }
    p.wg.Add(size)

    for i := 0; i < size; i++ {
        go func() {
            defer p.wg.Done()
            for task := range p.task {
                task()
            }
        }()
    }

    return p
}

func (p *Pool) Run(f func()) {
    p.task <- f
}

func (p *Pool) Shutdown() {
    close(p.task)
    p.wg.Wait()
}

使用:

package main

import (
    "fmt"
    "math/rand"
    "time"

    "./work"
)

func init() {
    rand.Seed(time.Now().UnixNano())
}

func main() {
    start := time.Now().UnixNano()
    pool := work.New(20)

    for i := 0; i < 100; i++ {
        pool.Run(func() {
            time.Sleep(time.Duration(rand.Intn(1000)) * time.Millisecond)
        })
    }
    pool.Shutdown()

    end := time.Now().UnixNano()
    fmt.Println("spent", (end-start)/1e6)
}

发表评论

电子邮件地址不会被公开。 必填项已用*标注

鄂ICP备17003086号. Copyright © 2016. All Rights Reserved.