def get_tasks_status_query(self, instance_ids):
query = TaskQuery()
query.owner = Identity(role=self._role)
query.environment = self._env
query.jobName = self._name
query.statuses = set([ScheduleStatus.RUNNING])
query.instanceIds = set(instance_ids)
return query
def _create_query(self, instance_ids): query = TaskQuery() query.owner = Identity(role=self._job_key.role) query.environment = self._job_key.environment query.jobName = self._job_key.name query.statuses = set([ScheduleStatus.RUNNING]) query.instanceIds = instance_ids return query
def _create_query(self, instance_ids):
query = TaskQuery()
query.owner = Identity(role=self._job_key.role)
query.environment = self._job_key.environment
query.jobName = self._job_key.name
query.statuses = set([ScheduleStatus.RUNNING])
query.instanceIds = instance_ids
return query
def get_tasks_status_query(self, instance_ids): query = TaskQuery() query.owner = Identity(role=self._role) query.environment = self._env query.jobName = self._name query.statuses = set([ScheduleStatus.RUNNING]) query.instanceIds = set(instance_ids) return query
def expect_task_status(self, once=False, instances=None):
query = TaskQuery(jobKeys=[
JobKey(role=self._job_key.role, environment=self._job_key.env, name=self._job_key.name)])
if instances is not None:
query.instanceIds = frozenset([int(s) for s in instances])
if once:
self._scheduler.getTasksWithoutConfigs.assert_called_once_with(query)
else:
self._scheduler.getTasksWithoutConfigs.assert_called_with(query)
def expect_task_status(self, once=False, instances=None):
query = TaskQuery(owner=Identity(role=self._job_key.role),
environment=self._job_key.env,
jobName=self._job_key.name)
if instances is not None:
query.instanceIds = frozenset([int(s) for s in instances])
if once:
self._scheduler.getTasksWithoutConfigs.assert_called_once_with(
query)
else:
self._scheduler.getTasksWithoutConfigs.assert_called_with(query)
def expect_task_status(self, once=False, instances=None):
query = TaskQuery(
owner=Identity(role=self._job_key.role),
environment=self._job_key.env,
jobName=self._job_key.name)
if instances is not None:
query.instanceIds = frozenset([int(s) for s in instances])
if once:
self._scheduler.getTasksStatus.assert_called_once_with(query)
else:
self._scheduler.getTasksStatus.assert_called_with(query)
def _get_tasks_by_instance_id(self, instance_ids):
log.debug('Querying instance statuses.')
query = TaskQuery()
query.owner = Identity(role=self._job_key.role)
query.environment = self._job_key.environment
query.jobName = self._job_key.name
query.statuses = set([ScheduleStatus.RUNNING])
query.instanceIds = instance_ids
try:
resp = self._scheduler.getTasksStatus(query)
except IOError as e:
log.error('IO Exception during scheduler call: %s' % e)
return []
tasks = []
if resp.responseCode == ResponseCode.OK:
tasks = resp.result.scheduleStatusResult.tasks
log.debug('Response from scheduler: %s (message: %s)'
% (ResponseCode._VALUES_TO_NAMES[resp.responseCode], resp.messageDEPRECATED))
return tasks
def create_query(cls, instances): query = TaskQuery() query.instanceIds = set(instances) return query
def get_tasks_status_query(self, instance_ids): query = TaskQuery() query.jobKeys = set([self._job_key]) query.statuses = set([ScheduleStatus.RUNNING]) query.instanceIds = set(instance_ids) return query
def _create_query(self, instance_ids):
query = TaskQuery()
query.jobKeys = set([self._job_key])
query.statuses = set([ScheduleStatus.RUNNING])
query.instanceIds = instance_ids
return query
def get_tasks_status_query(self, instance_ids):
query = TaskQuery()
query.jobKeys = set([self._job_key])
query.statuses = set([ScheduleStatus.RUNNING])
query.instanceIds = set(instance_ids)
return query
def _create_query(self, instance_ids): query = TaskQuery() query.jobKeys = set([self._job_key]) query.statuses = set([ScheduleStatus.RUNNING]) query.instanceIds = instance_ids return query
