4 from oslo_serialization import jsonutils
6 from api.database.v1.handlers import TasksHandler
7 from yardstick.common import constants as consts
9 logger = logging.getLogger(__name__)
10 logger.setLevel(logging.DEBUG)
13 class TaskThread(threading.Thread):
15 def __init__(self, target, args):
16 super(TaskThread, self).__init__(target=target, args=args)
21 task_handler = TasksHandler()
22 data = {'task_id': self.args.task_id, 'status': consts.TASK_NOT_DONE}
23 task_handler.insert(data)
25 logger.info('Starting run task')
27 data = self.target(self.args)
28 except Exception as e:
29 logger.exception('Task Failed')
30 update_data = {'status': consts.TASK_FAILED, 'error': str(e)}
31 task_handler.update_attr(self.args.task_id, update_data)
33 logger.info('Task Finished')
34 logger.debug('Result: %s', data)
36 data['result'] = jsonutils.dumps(data.get('result', {}))
37 task_handler.update_attr(self.args.task_id, data)