39 lines
438 B
Go
39 lines
438 B
Go
|
package pool
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
// Pool -
|
||
|
|
type Pool struct {
|
||
|
|
queue chan bool
|
||
|
|
wg *sync.WaitGroup
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewPool -
|
||
|
|
func NewPool(size int) *Pool {
|
||
|
|
if size < 1 {
|
||
|
|
size = 1
|
||
|
|
}
|
||
|
|
p := &Pool{
|
||
|
|
queue: make(chan bool, size),
|
||
|
|
wg: &sync.WaitGroup{},
|
||
|
|
}
|
||
|
|
return p
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add -
|
||
|
|
func (p *Pool) Add() {
|
||
|
|
p.queue <- true
|
||
|
|
p.wg.Add(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Done -
|
||
|
|
func (p *Pool) Done() {
|
||
|
|
<-p.queue
|
||
|
|
p.wg.Done()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Wait -
|
||
|
|
func (p *Pool) Wait() {
|
||
|
|
p.wg.Wait()
|
||
|
|
}
|