ridv/gorealis
1
0
Fork 0
Code Issues Pull requests Projects Releases 23 Packages Wiki Activity Actions

Adding a separate function to add dedicated attributes. (#80)

Browse source 
Dedicated wrapper for "dedicated" constraints
This commit is contained in:
Renan DelValle 2018-10-11 09:43:35 -07:00 • committed by GitHub
parent e0f33ab60e
commit 231793df71
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

21
job.go
View file

@ -37,6 +37,15 @@ type Job interface {
AddNamedPorts(names ...string) Job AddNamedPorts(names ...string) Job
AddLimitConstraint(name string, limit int32) Job AddLimitConstraint(name string, limit int32) Job
AddValueConstraint(name string, negated bool, values ...string) Job AddValueConstraint(name string, negated bool, values ...string) Job
// From Aurora Docs:
// dedicated attribute. Aurora treats this specially, and only allows matching jobs
// to run on these machines, and will only schedule matching jobs on these machines.
// When a job is created, the scheduler requires that the $role component matches
// the role field in the job configuration, and will reject the job creation otherwise.
// A wildcard (*) may be used for the role portion of the dedicated attribute, which
// will allow any owner to elect for a job to run on the host(s)
AddDedicatedConstraint(role, name string) Job
AddURIs(extract bool, cache bool, values ...string) Job AddURIs(extract bool, cache bool, values ...string) Job
JobKey() *aurora.JobKey JobKey() *aurora.JobKey
JobConfig() *aurora.JobConfiguration JobConfig() *aurora.JobConfiguration
@ -61,11 +70,11 @@ func NewJob() Job {
taskConfig := aurora.NewTaskConfig() taskConfig := aurora.NewTaskConfig()
jobKey := aurora.NewJobKey() jobKey := aurora.NewJobKey()
//Job Config // Job Config
jobConfig.Key = jobKey jobConfig.Key = jobKey
jobConfig.TaskConfig = taskConfig jobConfig.TaskConfig = taskConfig
//Task Config // Task Config
taskConfig.Job = jobKey taskConfig.Job = jobKey
taskConfig.Container = aurora.NewContainer() taskConfig.Container = aurora.NewContainer()
taskConfig.Container.Mesos = aurora.NewMesosContainer() taskConfig.Container.Mesos = aurora.NewMesosContainer()
@ -73,7 +82,7 @@ func NewJob() Job {
taskConfig.Metadata = make(map[*aurora.Metadata]bool) taskConfig.Metadata = make(map[*aurora.Metadata]bool)
taskConfig.Constraints = make(map[*aurora.Constraint]bool) taskConfig.Constraints = make(map[*aurora.Constraint]bool)
//Resources // Resources
numCpus := aurora.NewResource() numCpus := aurora.NewResource()
ramMb := aurora.NewResource() ramMb := aurora.NewResource()
diskMb := aurora.NewResource() diskMb := aurora.NewResource()
@ -297,6 +306,12 @@ func (j *AuroraJob) AddLimitConstraint(name string, limit int32) Job {
return j return j
} }
func (j *AuroraJob) AddDedicatedConstraint(role, name string) Job {
j.AddValueConstraint("dedicated", false, role+"/"+name)
return j
}
// Set a container to run for the job configuration to run. // Set a container to run for the job configuration to run.
func (j *AuroraJob) Container(container Container) Job { func (j *AuroraJob) Container(container Container) Job {
j.jobConfig.TaskConfig.Container = container.Build() j.jobConfig.TaskConfig.Container = container.Build()