40 lines
674 B
Go
40 lines
674 B
Go
package queue
|
|
|
|
import (
|
|
. "github.com/drone/drone/pkg/model"
|
|
)
|
|
|
|
// A Queue dispatches tasks to workers.
|
|
type Queue struct {
|
|
tasks chan<- *BuildTask
|
|
}
|
|
|
|
// BuildTasks represents a build that is pending
|
|
// execution.
|
|
type BuildTask struct {
|
|
Repo *Repo
|
|
Commit *Commit
|
|
Build *Build
|
|
}
|
|
|
|
// Start N workers with the given build runner.
|
|
func Start(workers int, runner BuildRunner) *Queue {
|
|
tasks := make(chan *BuildTask)
|
|
|
|
queue := &Queue{tasks: tasks}
|
|
|
|
for i := 0; i < workers; i++ {
|
|
worker := worker{
|
|
runner: runner,
|
|
}
|
|
|
|
go worker.work(tasks)
|
|
}
|
|
|
|
return queue
|
|
}
|
|
|
|
// Add adds the task to the build queue.
|
|
func (q *Queue) Add(task *BuildTask) {
|
|
q.tasks <- task
|
|
}
|