gf/g/os/groutine/groutine_job.go

// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.

package groutine

// 开始任务
func (j *PoolJob) start() {
    go func() {
        for {
            if f := <- j.job; f != nil {
                // 执行任务
                f()
                // 清空任务(GC可回收f对应资源)
                j.job = nil
                // 执行完毕后添加到空闲队列
                j.pool.addJob(j)
            } else {
                break
            }
        }
    }()
}

// 关闭当前任务
func (j *PoolJob) stop() {
    j.setJob(nil)
}

// 设置当前任务的执行函数
func (j *PoolJob) setJob(f func()) {
    j.job <- f
}