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 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