woodpecker/server/queue/queue.go

package queue

import (
	"github.com/drone/drone/pkg/resource/build"
	"github.com/drone/drone/pkg/resource/commit"
	"github.com/drone/drone/pkg/resource/repo"
	"github.com/drone/drone/pkg/resource/user"

	"github.com/drone/drone/pkg/build/script"
)

// 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.Repo
	Commit *commit.Commit
	Build  *build.Build
	User   *user.User

	// Build instructions from the .drone.yml
	// file, unmarshalled.
	Script *script.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
}