Skip to content

Instantly share code, notes, and snippets.

@jbrodriguez
Created July 17, 2015 10:04
Show Gist options
  • Save jbrodriguez/a2e52696459af4ef8a9b to your computer and use it in GitHub Desktop.
Save jbrodriguez/a2e52696459af4ef8a9b to your computer and use it in GitHub Desktop.
package pool
type Task interface {
Execute(id int)
}
type Pool struct {
tasks chan Task
kill chan bool
}
func NewPool(size, queue int) *Pool {
pool := &Pool{
tasks: make(chan Task, queue),
kill: make(chan bool),
}
for i := 0; i < size; i++ {
go pool.worker(i)
}
return pool
}
func (p *Pool) worker(id int) {
for {
select {
case task, ok := <-p.tasks:
if !ok {
return
}
task.Execute(id)
case <-p.kill:
// should really kill it
return
}
}
}
func (p *Pool) Close() {
close(p.tasks)
}
func (p *Pool) Exec(task Task) {
p.tasks <- task
}
@erazmis
Copy link

erazmis commented Jul 25, 2021

kill must be the Context.

@jbrodriguez
Copy link
Author

Thanks ! I don't think Context was available when I scribbled this one, but it definitely looks like it would be of good use here

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment