api/utils/thread.py

   1 import threading
   2 import os
   3 import logging
   4
   5 from oslo_serialization import jsonutils
   6
   7 from yardstick.common import constants as consts
   8
   9 LOG = logging.getLogger(__name__)
  10 LOG.setLevel(logging.DEBUG)
  11
  12
  13 class TaskThread(threading.Thread):
  14
  15     def __init__(self, target, args, handler):
  16         super(TaskThread, self).__init__(target=target, args=args)
  17         self.target = target
  18         self.args = args
  19         self.handler = handler
  20
  21     def run(self):
  22         if self.handler.__class__.__name__.lower().startswith('v2'):
  23             self.handler.update_attr(self.args.task_id, {'status': consts.TASK_NOT_DONE})
  24         else:
  25             update_data = {'task_id': self.args.task_id, 'status': consts.TASK_NOT_DONE}
  26             self.handler.insert(update_data)
  27
  28         LOG.info('Starting run task')
  29         try:
  30             data = self.target(self.args)
  31         except Exception as e:
  32             LOG.exception('Task Failed')
  33             update_data = {'status': consts.TASK_FAILED, 'error': str(e)}
  34             self.handler.update_attr(self.args.task_id, update_data)
  35         else:
  36             LOG.info('Task Finished')
  37             LOG.debug('Result: %s', data)
  38
  39             if self.handler.__class__.__name__.lower().startswith('v2'):
  40                 new_data = {'status': consts.TASK_DONE, 'result': jsonutils.dumps(data['result'])}
  41                 self.handler.update_attr(self.args.task_id, new_data)
  42                 os.remove(self.args.inputfile[0])
  43             else:
  44                 data['result'] = jsonutils.dumps(data.get('result', {}))
  45                 self.handler.update_attr(self.args.task_id, data)