135 lines
2.3 KiB
Go
135 lines
2.3 KiB
Go
package jobspec2
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/api"
|
|
"github.com/hashicorp/nomad/helper/pointer"
|
|
)
|
|
|
|
func normalizeJob(jc *jobConfig) {
|
|
j := jc.Job
|
|
if j.Name == nil {
|
|
j.Name = &jc.JobID
|
|
}
|
|
if j.ID == nil {
|
|
j.ID = &jc.JobID
|
|
}
|
|
|
|
if j.Periodic != nil && j.Periodic.Spec != nil {
|
|
v := "cron"
|
|
j.Periodic.SpecType = &v
|
|
}
|
|
|
|
normalizeVault(jc.Vault)
|
|
|
|
if len(jc.Tasks) != 0 {
|
|
alone := make([]*api.TaskGroup, 0, len(jc.Tasks))
|
|
for _, t := range jc.Tasks {
|
|
alone = append(alone, &api.TaskGroup{
|
|
Name: &t.Name,
|
|
Tasks: []*api.Task{t},
|
|
})
|
|
}
|
|
alone = append(alone, j.TaskGroups...)
|
|
j.TaskGroups = alone
|
|
}
|
|
|
|
for _, tg := range j.TaskGroups {
|
|
normalizeNetworkPorts(tg.Networks)
|
|
for _, t := range tg.Tasks {
|
|
if t.Resources != nil {
|
|
normalizeNetworkPorts(t.Resources.Networks)
|
|
}
|
|
|
|
normalizeTemplates(t.Templates)
|
|
|
|
// normalize Vault
|
|
normalizeVault(t.Vault)
|
|
|
|
if t.Vault == nil {
|
|
t.Vault = jc.Vault
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func normalizeVault(v *api.Vault) {
|
|
if v == nil {
|
|
return
|
|
}
|
|
|
|
if v.Env == nil {
|
|
v.Env = pointer.Of(true)
|
|
}
|
|
if v.ChangeMode == nil {
|
|
v.ChangeMode = pointer.Of("restart")
|
|
}
|
|
}
|
|
|
|
func normalizeNetworkPorts(networks []*api.NetworkResource) {
|
|
if networks == nil {
|
|
return
|
|
}
|
|
for _, n := range networks {
|
|
if len(n.DynamicPorts) == 0 {
|
|
continue
|
|
}
|
|
|
|
dynamic := make([]api.Port, 0, len(n.DynamicPorts))
|
|
var reserved []api.Port
|
|
|
|
for _, p := range n.DynamicPorts {
|
|
if p.Value > 0 {
|
|
reserved = append(reserved, p)
|
|
} else {
|
|
dynamic = append(dynamic, p)
|
|
}
|
|
}
|
|
if len(dynamic) == 0 {
|
|
dynamic = nil
|
|
}
|
|
|
|
n.DynamicPorts = dynamic
|
|
n.ReservedPorts = reserved
|
|
}
|
|
|
|
}
|
|
|
|
func normalizeTemplates(templates []*api.Template) {
|
|
if len(templates) == 0 {
|
|
return
|
|
}
|
|
|
|
for _, t := range templates {
|
|
if t.ChangeMode == nil {
|
|
t.ChangeMode = pointer.Of("restart")
|
|
}
|
|
if t.Perms == nil {
|
|
t.Perms = pointer.Of("0644")
|
|
}
|
|
if t.Splay == nil {
|
|
t.Splay = pointer.Of(5 * time.Second)
|
|
}
|
|
normalizeChangeScript(t.ChangeScript)
|
|
}
|
|
}
|
|
|
|
func normalizeChangeScript(ch *api.ChangeScript) {
|
|
if ch == nil {
|
|
return
|
|
}
|
|
|
|
if ch.Args == nil {
|
|
ch.Args = []string{}
|
|
}
|
|
|
|
if ch.Timeout == nil {
|
|
ch.Timeout = pointer.Of(5 * time.Second)
|
|
}
|
|
|
|
if ch.FailOnError == nil {
|
|
ch.FailOnError = pointer.Of(false)
|
|
}
|
|
}
|