def stop_job(self):
request = task.StopRequest(jobId=peloton.JobID(value=self.job_id))
self.client.task_svc.Stop(
request,
metadata=self.client.jobmgr_metadata,
timeout=default_timeout,
)
def stop_task(self, job_id, instance_id):
"""
param job_id: id of the job
param instance_id: instance id of the task to stop
type job_id: str
type instance_id: int
rtype: task.StopResponse
"""
rng = task.InstanceRange(to=instance_id + 1)
setattr(rng, "from", instance_id)
request = task.StopRequest(
jobId=peloton.JobID(value=job_id), ranges=[rng]
)
try:
print_okblue("Stopping task %d of Job %s" % (instance_id, job_id))
resp = self.client.task_svc.Stop(
request,
metadata=self.client.jobmgr_metadata,
timeout=default_timeout,
)
return resp
except Exception as e:
print_fail("Exception calling Stop Tasks :%s" % str(e))
raise
def stop(self, ranges=None):
"""
Stops a job or certain tasks based on the ranges
:param ranges: the instance ranges to stop
:return: task stop response from the API
"""
request = task.StopRequest(jobId=peloton.JobID(value=self.job_id),
ranges=ranges)
response = self.client.task_svc.Stop(
request,
metadata=self.client.jobmgr_metadata,
timeout=self.config.rpc_timeout_sec,
)
log.info("stopping tasks in job {0} with ranges {1}".format(
self.job_id, ranges))
return response
def stop_job(self, job_id):
"""
param job_id: id of the job
type job_id: str
rtype: job.StopResponse
"""
request = task.StopRequest(jobId=peloton.JobID(value=job_id))
try:
print_okblue("Killing all tasks of Job %s" % job_id)
resp = self.client.task_svc.Stop(
request,
metadata=self.client.jobmgr_metadata,
timeout=default_timeout,
)
return resp
except Exception as e:
print_fail("Exception calling List Tasks :%s" % str(e))
raise
