def _get_process_resource_consumption(self, task_id, process_name): if task_id not in self.active_tasks: log.debug("Task %s not found in active tasks" % task_id) return ProcessSample.empty().to_dict() sample = self.active_tasks[task_id].resource_monitor.sample_by_process(process_name).to_dict() log.debug('Resource consumption (%s, %s) => %s' % (task_id, process_name, sample)) return sample
def _sample(self, task_id): if task_id not in self.active_tasks: log.debug("Task %s not found in active tasks" % task_id) sample = ProcessSample.empty().to_dict() sample['disk'] = 0 else: resource_sample = self.active_tasks[task_id].resource_monitor.sample()[1] sample = resource_sample.process_sample.to_dict() sample['disk'] = resource_sample.disk_usage log.debug("Got sample for task %s: %s" % (task_id, sample)) return sample