60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package queue
|
|
|
|
import (
|
|
"github.com/drone/drone/pkg/build/docker"
|
|
"github.com/drone/drone/pkg/build/script"
|
|
. "github.com/drone/drone/pkg/model"
|
|
"runtime"
|
|
"time"
|
|
)
|
|
|
|
// 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
|
|
|
|
// Build instructions from the .drone.yml
|
|
// file, unmarshalled.
|
|
Script *script.Build
|
|
}
|
|
|
|
var defaultQueue = Start(runtime.NumCPU(), newRunner(docker.DefaultClient, 300*time.Second)) // TEMPORARY; INJECT PLEASE
|
|
|
|
var Add = defaultQueue.Add // TEMPORARY; INJECT PLEASE
|
|
|
|
func Start(workers int, runner Runner) *Queue {
|
|
// get the number of CPUs. Since builds
|
|
// tend to be CPU-intensive we should only
|
|
// execute 1 build per CPU.
|
|
// must be at least 1
|
|
// if ncpu < 1 {
|
|
// ncpu = 1
|
|
// }
|
|
|
|
tasks := make(chan *BuildTask)
|
|
|
|
queue := &Queue{tasks: tasks}
|
|
|
|
// spawn a worker for each CPU
|
|
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
|
|
}
|