You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

51 lines
1.7 KiB

from typing import Dict
import detectors
from analytic_unit_worker import AnalyticUnitWorker
analytic_unit_id = str
analytic_workers: Dict[analytic_unit_id, AnalyticUnitWorker] = Dict()
def get_detector(self, analytic_unit_type) -> detectors.Detector:
if analytic_unit_type == 'GENERAL':
detector = detectors.GeneralDetector()
else:
detector = detectors.PatternDetector(analytic_unit_type)
return detector
def ensure_worker(analytic_unit_id, analytic_unit_type) -> AnalyticUnitWorker:
if analytic_unit_id in analytic_workers:
# TODO: check that type is the same
return analytic_workers[analytic_unit_id]
detector = get_detector(analytic_unit_type)
worker = AnalyticUnitWorker(analytic_unit_id, detector)
analytic_workers[analytic_unit_id] = worker
return worker
async def handle_analytic_task(task):
try:
worker = ensure_worker(task['analyticUnitId'], task['type'])
payload = task['payload']
if type == "PREDICT":
result_payload = await worker.do_predict(analytic_unit_id, payload)
elif type == "LEARN":
result_payload = await worker.do_learn(analytic_unit_id, payload)
else:
raise ValueError('Unknown task type "%s"' % type)
except Exception as e:
#traceback.extract_stack()
error_text = traceback.format_exc()
logger.error("do_task Exception: '%s'" % error_text)
# TODO: move result to a class which renders to json for messaging to analytics
result = {
'status': "FAILED",
'error': str(e)
}
return {
'status': 'SUCCESS',
'payload': result_payload
}