darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-23 08:51:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added in tasks for created initial job stats and for updating following an outcome.

Browse Source
This commit is contained in:
Martyn Inglis
2017-05-09 15:24:05 +01:00
parent 2cbf8d29fd
commit b4bf332ea0
2 changed files with 94 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
app/celery/statistics_tasks.py
View File

@@ -1,8 +1,36 @@
from sqlalchemy.exc import SQLAlchemyError
from app import notify_celery
from flask import current_app
from app.statsd_decorators import statsd
from app.dao.statistics_dao import create_or_update_job_sending_statistics, update_job_stats_outcome_count
@notify_celery.task(bind=True, name='record_initial_job_statistics')
@notify_celery.task(bind=True, name='record_initial_job_statistics', max_retries=20, default_retry_delay=300)
@statsd(namespace="tasks")
def record_initial_job_statistics(notification):
pass
def record_initial_job_statistics(self, notification):
try:
create_or_update_job_sending_statistics(notification)
except SQLAlchemyError as e:
current_app.logger.exception(e)
self.retry(queue="retry")
except self.MaxRetriesExceededError:
current_app.logger.error(
"RETRY FAILED: task record_initial_job_statistics failed for notification {}".format(notification.id)
)
@notify_celery.task(bind=True, name='record_outcome_job_statistics', max_retries=20, default_retry_delay=300)
@statsd(namespace="tasks")
def record_outcome_job_statistics(self, notification):
try:
updated_count = update_job_stats_outcome_count(notification)
if updated_count == 0:
self.retry(queue="retry")
except SQLAlchemyError as e:
current_app.logger.exception(e)
self.retry(queue="retry")
except self.MaxRetriesExceededError:
current_app.logger.error(
"RETRY FAILED: task update_job_stats_outcome_count failed for notification {}".format(notification.id)
)