summaryrefslogtreecommitdiffstats
path: root/api/utils/thread.py
blob: 2106548f5d9c81f801f2812cfd440634df18bee5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import threading
import logging

from oslo_serialization import jsonutils

from api.database.v1.handlers import TasksHandler
from yardstick.common import constants as consts

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)


class TaskThread(threading.Thread):

    def __init__(self, target, args):
        super(TaskThread, self).__init__(target=target, args=args)
        self.target = target
        self.args = args

    def run(self):
        task_handler = TasksHandler()
        data = {'task_id': self.args.task_id, 'status': consts.TASK_NOT_DONE}
        task_handler.insert(data)

        logger.info('Starting run task')
        try:
            data = self.target(self.args)
        except Exception as e:
            logger.exception('Task Failed')
            update_data = {'status': consts.TASK_FAILED, 'error': str(e)}
            task_handler.update_attr(self.args.task_id, update_data)
        else:
            logger.info('Task Finished')
            logger.debug('Result: %s', data)

            data['result'] = jsonutils.dumps(data.get('result', {}))
            task_handler.update_attr(self.args.task_id, data)