woodpecker/server/queue/queue.go

50 lines
1,000 B
Go
Raw Normal View History

2014-02-07 10:10:01 +00:00
package queue
import (
2014-06-04 21:44:05 +00:00
"github.com/drone/drone/server/resource/build"
"github.com/drone/drone/server/resource/commit"
"github.com/drone/drone/server/resource/repo"
"github.com/drone/drone/server/resource/user"
2014-06-04 21:25:38 +00:00
2014-06-04 21:44:05 +00:00
"github.com/drone/drone/shared/build/script"
2014-02-07 10:10:01 +00:00
)
// A Queue dispatches tasks to workers.
type Queue struct {
tasks chan<- *BuildTask
2014-02-07 10:10:01 +00:00
}
// BuildTasks represents a build that is pending
// execution.
type BuildTask struct {
2014-06-04 21:25:38 +00:00
Repo *repo.Repo
Commit *commit.Commit
Build *build.Build
User *user.User
// Build instructions from the .drone.yml
// file, unmarshalled.
Script *script.Build
2014-02-07 10:10:01 +00:00
}
// Start N workers with the given build runner.
func Start(workers int, runner BuildRunner) *Queue {
tasks := make(chan *BuildTask)
queue := &Queue{tasks: tasks}
2014-02-07 10:10:01 +00:00
for i := 0; i < workers; i++ {
worker := worker{
runner: runner,
}
go worker.work(tasks)
2014-02-07 10:10:01 +00:00
}
return queue
2014-02-07 10:10:01 +00:00
}
// Add adds the task to the build queue.
func (q *Queue) Add(task *BuildTask) {
q.tasks <- task
2014-02-07 10:10:01 +00:00
}