forked from mirror/ants
49 lines
639 B
Go
49 lines
639 B
Go
package ants
|
|
|
|
import (
|
|
"sync/atomic"
|
|
)
|
|
|
|
type Worker struct {
|
|
pool *Pool
|
|
task chan f
|
|
}
|
|
|
|
//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) run() {
|
|
go func() {
|
|
for f := range w.task {
|
|
if f == nil {
|
|
atomic.AddInt32(&w.pool.running, -1)
|
|
return
|
|
}
|
|
f()
|
|
w.pool.putWorker(w)
|
|
w.pool.wg.Done()
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (w *Worker) stop() {
|
|
w.task <- nil
|
|
}
|
|
|
|
func (w *Worker) sendTask(task f) {
|
|
w.task <- task
|
|
}
|