api/utils/thread.py

   1 import threading
   2 import logging
   3
   4 from oslo_serialization import jsonutils
   5
   6 from api.database.v1.handlers import TasksHandler
   7 from yardstick.common import constants as consts
   8
   9 logger = logging.getLogger(__name__)
  10 logger.setLevel(logging.DEBUG)
  11
  12
  13 class TaskThread(threading.Thread):
  14
  15     def __init__(self, target, args):
  16         super(TaskThread, self).__init__(target=target, args=args)
  17         self.target = target
  18         self.args = args
  19
  20     def run(self):
  21         task_handler = TasksHandler()
  22         data = {'task_id': self.args.task_id, 'status': consts.TASK_NOT_DONE}
  23         task_handler.insert(data)
  24
  25         logger.info('Starting run task')
  26         try:
  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)
  32         else:
  33             logger.info('Task Finished')
  34             logger.debug('Result: %s', data)
  35
  36             data['result'] = jsonutils.dumps(data.get('result', {}))
  37             task_handler.update_attr(self.args.task_id, data)