美文网首页
聊聊tunny的workerWrapper

聊聊tunny的workerWrapper

作者: go4it | 来源:发表于2021-04-28 22:57 被阅读0次

本文主要研究一下tunny的workerWrapper

workerWrapper

type workerWrapper struct {
    worker        Worker
    interruptChan chan struct{}

    // reqChan is NOT owned by this type, it is used to send requests for work.
    reqChan chan<- workRequest

    // closeChan can be closed in order to cleanly shutdown this worker.
    closeChan chan struct{}

    // closedChan is closed by the run() goroutine when it exits.
    closedChan chan struct{}
}

func newWorkerWrapper(
    reqChan chan<- workRequest,
    worker Worker,
) *workerWrapper {
    w := workerWrapper{
        worker:        worker,
        interruptChan: make(chan struct{}),
        reqChan:       reqChan,
        closeChan:     make(chan struct{}),
        closedChan:    make(chan struct{}),
    }

    go w.run()

    return &w
}

workerWrapper包装了worker,定义了interruptChan、reqChan、closeChan、closedChan属性

interrupt

func (w *workerWrapper) interrupt() {
    close(w.interruptChan)
    w.worker.Interrupt()
}

interrupt方法关闭w.interruptChan,执行w.worker.Interrupt()

run

func (w *workerWrapper) run() {
    jobChan, retChan := make(chan interface{}), make(chan interface{})
    defer func() {
        w.worker.Terminate()
        close(retChan)
        close(w.closedChan)
    }()

    for {
        // NOTE: Blocking here will prevent the worker from closing down.
        w.worker.BlockUntilReady()
        select {
        case w.reqChan <- workRequest{
            jobChan:       jobChan,
            retChan:       retChan,
            interruptFunc: w.interrupt,
        }:
            select {
            case payload := <-jobChan:
                result := w.worker.Process(payload)
                select {
                case retChan <- result:
                case <-w.interruptChan:
                    w.interruptChan = make(chan struct{})
                }
            case _, _ = <-w.interruptChan:
                w.interruptChan = make(chan struct{})
            }
        case <-w.closeChan:
            return
        }
    }
}

run首先创建jobChan、retChan,然后for循环执行select读取reqChan,之后读取jobChan的payload,进行处理,然后写入到retChan

stop

func (w *workerWrapper) stop() {
    close(w.closeChan)
}

stop方法关闭w.closeChan

join

func (w *workerWrapper) join() {
    <-w.closedChan
}

join方法则等待w.closedChan

小结

tunny的workerWrapper包装了worker,定义了interruptChan、reqChan、closeChan、closedChan属性,它提供了interrupt、run、stop、join方法。

doc

相关文章

  • 聊聊tunny的workerWrapper

    序 本文主要研究一下tunny的workerWrapper workerWrapper workerWrapper...

  • 聊聊golang的tunny

    序 本文主要研究一下tunny Worker Worker接口定义了Process、BlockUntilReady...

  • tunny源代码阅读

    线程池——tunny 代码:https://github.com/Jeffail/tunny[https://gi...

  • tunny源码阅读

    前言 最近在学习Go并发,在同学强烈推荐下,阅读了tunny源码。在此记录自己的理解和感想。 tunny 基于Go...

  • Go 每日一库之 tunny

    简介 之前写过一篇文章介绍了ants这个 goroutine 池实现。当时在网上查看相关资料的时候,发现了另外一个...

  • 聊聊…聊聊?

    世界不大,一座城市里,用高楼大厦圈出来的的圈子更小了… 心再大,也会被城市里喧嚣的汽笛压抑自己 不记得有多久没有好...

  • 聊聊聊

    今天主要的时间是和阿q过的,非常开心我们有了这么一次聊天! 我觉得自己不孤单了。我俩目前拥有的感情非常相似,是比较...

  • 聊聊聊出来的感情!

    刚好回学校那天晚上,我有个比赛,以此草草结束了聊天。等忙完,我吱了一声,就直接洗洗睡了,用行动加强自己的决心。 没...

  • 无聊聊聊

  • 聊聊,聊聊选择

    今早梦到一杯豆浆15元,我给自己的孩子买了一杯50元的奶茶,对她感叹“在我们那个年代一杯奶茶才10元”孩子问我那么...

网友评论

      本文标题:聊聊tunny的workerWrapper

      本文链接:https://www.haomeiwen.com/subject/odobrltx.html