115 lines
2.9 KiB
Python
Executable File
115 lines
2.9 KiB
Python
Executable File
#
|
|
# Copyright (c) 2015-2016 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
from nfv_common import debug
|
|
from nfv_common import exceptions
|
|
from nfv_common import timers
|
|
|
|
from nfv_common.helpers import Result
|
|
|
|
DLOG = debug.debug_get_logger('nfv_common.tasks.task_work')
|
|
|
|
|
|
class TaskWork(object):
|
|
"""
|
|
Task Work
|
|
"""
|
|
_id = 0
|
|
|
|
def __init__(self, timeout, target, *args, **kwargs):
|
|
"""
|
|
Create task work
|
|
"""
|
|
self._id = TaskWork._id
|
|
self._name = target.__name__
|
|
self._task_id = None
|
|
self._target = target
|
|
self._timeout_in_secs = timeout
|
|
self._args = list(args)
|
|
self._kwargs = dict(kwargs)
|
|
self._result = None
|
|
self._ancillary_result_data = None
|
|
self._create_timestamp_ms = timers.get_monotonic_timestamp_in_ms()
|
|
|
|
DLOG.debug("TaskWork created, id=%s, name=%s, timeout_in_secs=%i."
|
|
% (self._id, self._name, self._timeout_in_secs))
|
|
TaskWork._id += 1
|
|
|
|
@property
|
|
def name(self):
|
|
"""
|
|
Returns the name of the task work
|
|
"""
|
|
return self._name
|
|
|
|
@property
|
|
def id(self):
|
|
"""
|
|
Returns the unique identifier of the task work
|
|
"""
|
|
return self._id
|
|
|
|
@property
|
|
def task_id(self):
|
|
"""
|
|
Returns the task identifier that owns this task work
|
|
"""
|
|
return self._task_id
|
|
|
|
@task_id.setter
|
|
def task_id(self, task_id):
|
|
"""
|
|
Set the task identifier that owns this task work
|
|
"""
|
|
self._task_id = task_id
|
|
|
|
@property
|
|
def create_timestamp_ms(self):
|
|
"""
|
|
Returns the creation timestamp in milliseconds
|
|
"""
|
|
return self._create_timestamp_ms
|
|
|
|
@property
|
|
def timeout_in_secs(self):
|
|
"""
|
|
Returns the maximum timeout in seconds that the task work should
|
|
take to run
|
|
"""
|
|
return self._timeout_in_secs
|
|
|
|
@property
|
|
def ancillary_result_data(self):
|
|
"""
|
|
Returns the ancillary result data for the task work
|
|
"""
|
|
return self._ancillary_result_data
|
|
|
|
@property
|
|
def result(self):
|
|
"""
|
|
Returns the result of the task work
|
|
"""
|
|
return self._result
|
|
|
|
def run(self):
|
|
"""
|
|
Runs the task work
|
|
"""
|
|
DLOG.debug("TaskWork run, id=%s, name=%s." % (self._id, self._name))
|
|
try:
|
|
result = self._target(*self._args, **self._kwargs)
|
|
if isinstance(result, Result):
|
|
self._result = result.result_data
|
|
self._ancillary_result_data = result.ancillary_data
|
|
else:
|
|
self._result = result
|
|
|
|
except Exception as e:
|
|
if isinstance(e, exceptions.PickleableException):
|
|
self._result = e
|
|
else:
|
|
self._result = Exception(e.__class__.__name__ + ": " + str(e))
|