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 prune_tasks(args, options): if len(args) == 0: die('Must specify at least cluster.') cluster = args[0] t = TaskQuery() if options.states: t.statuses = set(map(ScheduleStatus._NAMES_TO_VALUES.get, options.states.split(','))) if options.role: t.role = options.role if options.environment: t.environment = options.environment if options.limit: t.limit = options.limit api = make_admin_client_with_options(cluster) rsp = api.prune_tasks(t) if rsp.responseCode != ResponseCode.OK: die('Failed to prune tasks: %s' % combine_messages(rsp)) else: print("Tasks pruned.")
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 prune_tasks(args, options): if len(args) == 0: die('Must specify at least cluster.') cluster = args[0] t = TaskQuery() if options.states: t.statuses = set( map(ScheduleStatus._NAMES_TO_VALUES.get, options.states.split(','))) if options.role: t.role = options.role if options.environment: t.environment = options.environment if options.limit: t.limit = options.limit api = make_admin_client_with_options(cluster) rsp = api.prune_tasks(t) if rsp.responseCode != ResponseCode.OK: die('Failed to prune tasks: %s' % combine_messages(rsp)) else: print("Tasks pruned.")