Adding support for PartitionPolicy.

This commit is contained in:
Robert Allen 2018-12-20 16:38:06 -06:00 committed by Renan DelValle
parent 461b23400c
commit c553f67d4e
4 changed files with 77 additions and 3 deletions

8
job.go
View file

@ -202,3 +202,11 @@ func (j *AuroraJob) ThermosExecutor(thermos ThermosExecutor) *AuroraJob {
func (j *AuroraJob) BuildThermosPayload() error {
return j.task.BuildThermosPayload()
}
func (j *AuroraJob) PartitionPolicy(reschedule bool, delay int64) *AuroraJob {
j.task.PartitionPolicy(aurora.PartitionPolicy{
Reschedule: reschedule,
DelaySecs: &delay,
})
return j
}