ants/worker.go

package ants

import (
	"sync/atomic"
	"container/list"
	"sync"
)

type Worker struct {
	pool *Pool
	task chan f
	exit chan sig
}

func (w *Worker) run() {
	go func() {
		for {
			select {
			case f := <-w.task:
				f()
				w.pool.putWorker(w)
				w.pool.wg.Done()
			case <-w.exit:
				atomic.AddInt32(&w.pool.running, -1)
				return
			}
		}
	}()
}

func (w *Worker) stop() {
	w.exit <- sig{}
}

func (w *Worker) sendTask(task f) {
	w.task <- task
}

//--------------------------------------------------------------------------------

type ConcurrentQueue struct {
	queue *list.List
	m     sync.Mutex
}

func NewConcurrentQueue() *ConcurrentQueue {
	q := new(ConcurrentQueue)
	q.queue = list.New()
	return q
}

func (q *ConcurrentQueue) push(v interface{}) {
	defer q.m.Unlock()
	q.m.Lock()
	q.queue.PushFront(v)
}

func (q *ConcurrentQueue) pop() interface{} {
	defer q.m.Unlock()
	q.m.Lock()
	if elem := q.queue.Back(); elem != nil {
		return q.queue.Remove(elem)
	}
	return nil
}