x/solar
Code Issues Proposed changes

Added session start / end to celery using hooks

Browse Source
This commit is contained in:
Jedrzej Nowak 2015-11-20 13:15:33 +01:00
parent ffd388e6ec
commit 2c385c173a
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
solar/solar/orchestration/tasks.py
View File

@ -17,6 +17,7 @@ import subprocess
import time
from celery.app import task
from celery.signals import task_prerun, task_postrun
from solar.orchestration import graph
from solar.core import actions
@ -26,6 +27,7 @@ from solar.orchestration.runner import app
from solar.orchestration.traversal import traverse
from solar.orchestration import limits
from solar.orchestration import executor
from solar.dblayer import ModelMeta
__all__ = ['solar_resource', 'cmd', 'sleep',
@ -52,6 +54,14 @@ class ReportTask(task.Task):
report_task = partial(app.task, base=ReportTask, bind=True)
@task_prerun.connect
def start_solar_session(task_id, task, *args, **kwargs):
ModelMeta.session_start()
@task_postrun.connect
def end_solar_session(task_id, task, *args, **kwargs):
ModelMeta.session_end()
@report_task(name='solar_resource')
def solar_resource(ctxt, resource_name, action):