83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
|
package api
|
||
|
|
|
||
|
|
// TaskGroup is the unit of scheduling.
|
||
|
|
type TaskGroup struct {
|
||
|
|
Name string
|
||
|
|
Count int
|
||
|
|
Constraints []*Constraint
|
||
|
|
Tasks []*Task
|
||
|
|
Meta map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewTaskGroup creates a new TaskGroup.
|
||
|
|
func NewTaskGroup(name string, count int) *TaskGroup {
|
||
|
|
return &TaskGroup{
|
||
|
|
Name: name,
|
||
|
|
Count: count,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Constrain is used to add a constraint to a task group.
|
||
|
|
func (g *TaskGroup) Constrain(c *Constraint) *TaskGroup {
|
||
|
|
g.Constraints = append(g.Constraints, c)
|
||
|
|
return g
|
||
|
|
}
|
||
|
|
|
||
|
|
// AddMeta is used to add a meta k/v pair to a task group
|
||
|
|
func (g *TaskGroup) SetMeta(key, val string) *TaskGroup {
|
||
|
|
if g.Meta == nil {
|
||
|
|
g.Meta = make(map[string]string)
|
||
|
|
}
|
||
|
|
g.Meta[key] = val
|
||
|
|
return g
|
||
|
|
}
|
||
|
|
|
||
|
|
// AddTask is used to add a new task to a task group.
|
||
|
|
func (g *TaskGroup) AddTask(t *Task) *TaskGroup {
|
||
|
|
g.Tasks = append(g.Tasks, t)
|
||
|
|
return g
|
||
|
|
}
|
||
|
|
|
||
|
|
// Task is a single process in a task group.
|
||
|
|
type Task struct {
|
||
|
|
Name string
|
||
|
|
Driver string
|
||
|
|
Config map[string]string
|
||
|
|
Constraints []*Constraint
|
||
|
|
Resources *Resources
|
||
|
|
Meta map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewTask creates and initializes a new Task.
|
||
|
|
func NewTask(name, driver string) *Task {
|
||
|
|
return &Task{
|
||
|
|
Name: name,
|
||
|
|
Driver: driver,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Configure is used to configure a single k/v pair on
|
||
|
|
// the task.
|
||
|
|
func (t *Task) SetConfig(key, val string) *Task {
|
||
|
|
if t.Config == nil {
|
||
|
|
t.Config = make(map[string]string)
|
||
|
|
}
|
||
|
|
t.Config[key] = val
|
||
|
|
return t
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetMeta is used to add metadata k/v pairs to the task.
|
||
|
|
func (t *Task) SetMeta(key, val string) *Task {
|
||
|
|
if t.Meta == nil {
|
||
|
|
t.Meta = make(map[string]string)
|
||
|
|
}
|
||
|
|
t.Meta[key] = val
|
||
|
|
return t
|
||
|
|
}
|
||
|
|
|
||
|
|
// Require is used to add resource requirements to a task.
|
||
|
|
func (t *Task) Require(r *Resources) *Task {
|
||
|
|
t.Resources = r
|
||
|
|
return t
|
||
|
|
}
|