Adds priority for aurora-scheduler (#13)

Adds priority for task config
This commit is contained in:
lenhattan86 2021-09-16 16:29:25 -07:00 committed by GitHub
parent 983bf44b9f
commit 425faf28b8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 19 additions and 1 deletions

5
job.go
View file

@ -156,6 +156,11 @@ func (j *AuroraJob) IsService(isService bool) *AuroraJob {
return j
}
func (j *AuroraJob) Priority(priority int32) *AuroraJob {
j.task.Priority(priority)
return j
}
func (j *AuroraJob) TaskConfig() *aurora.TaskConfig {
return j.task.TaskConfig()
}

View file

@ -221,6 +221,11 @@ func (j *JobUpdate) IsService(isService bool) *JobUpdate {
return j
}
func (j *JobUpdate) Priority(priority int32) *JobUpdate {
j.task.Priority(priority)
return j
}
func (j *JobUpdate) TaskConfig() *aurora.TaskConfig {
return j.task.TaskConfig()
}

View file

@ -78,7 +78,8 @@ func TaskFromThrift(config *aurora.TaskConfig) *AuroraTask {
Role(config.Job.Role).
Name(config.Job.Name).
MaxFailure(config.MaxTaskFailures).
IsService(config.IsService)
IsService(config.IsService).
Priority(config.Priority)
if config.Tier != nil {
newTask.Tier(*config.Tier)
@ -287,6 +288,12 @@ func (t *AuroraTask) IsService(isService bool) *AuroraTask {
return t
}
//set priority for preemption or priority-queueing
func (t *AuroraTask) Priority(priority int32) *AuroraTask {
t.task.Priority = priority
return t
}
// Add a list of URIs with the same extract and cache configuration. Scheduler must have
// --enable_mesos_fetcher flag enabled. Currently there is no duplicate detection.
func (t *AuroraTask) AddURIs(extract bool, cache bool, values ...string) *AuroraTask {

View file

@ -34,6 +34,7 @@ func TestAuroraTask_Clone(t *testing.T) {
RAM(643).
Disk(1000).
IsService(true).
Priority(1).
AddPorts(10).
Tier("preferred").
MaxFailure(23).