diff --git a/job.go b/job.go index b553520..698ecfa 100644 --- a/job.go +++ b/job.go @@ -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() } diff --git a/task.go b/task.go index e4037c3..a6e9f5a 100644 --- a/task.go +++ b/task.go @@ -287,6 +287,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 {