mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-16 02:02:13 -05:00
When we sent the service the status callback for a notification, we have all the information we need.
Which means we can remove the need to request the data from the database. In order for the PR to be backwards compatible I have added an optional parameter "encrypted_status_update". If this is not None then the new code is called. The next PR will send the encrypted data to this task. A final PR will remove the code that uses the database to get the notification and service callback api.
This commit is contained in:
@@ -6,6 +6,7 @@ from app import (
|
||||
db,
|
||||
DATETIME_FORMAT,
|
||||
notify_celery,
|
||||
encryption
|
||||
)
|
||||
from app.dao.notifications_dao import (
|
||||
get_notification_by_id,
|
||||
@@ -23,10 +24,61 @@ from app.config import QueueNames
|
||||
|
||||
@notify_celery.task(bind=True, name="send-delivery-status", max_retries=5, default_retry_delay=300)
|
||||
@statsd(namespace="tasks")
|
||||
def send_delivery_status_to_service(self, notification_id):
|
||||
def send_delivery_status_to_service(self, notification_id,
|
||||
encrypted_status_update=None
|
||||
):
|
||||
if not encrypted_status_update:
|
||||
process_update_with_notification_id(self, notification_id=notification_id)
|
||||
else:
|
||||
try:
|
||||
status_update = encryption.decrypt(encrypted_status_update)
|
||||
|
||||
data = {
|
||||
"id": str(notification_id),
|
||||
"reference": status_update['notification_client_reference'],
|
||||
"to": status_update['notification_to'],
|
||||
"status": status_update['notification_status'],
|
||||
"created_at": status_update['notification_created_at'],
|
||||
"completed_at": status_update['notification_updated_at'],
|
||||
"sent_at": status_update['notification_sent_at'],
|
||||
"notification_type": status_update['notification_type']
|
||||
}
|
||||
|
||||
response = request(
|
||||
method="POST",
|
||||
url=status_update['service_callback_api_url'],
|
||||
data=json.dumps(data),
|
||||
headers={
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': 'Bearer {}'.format(status_update['service_callback_api_bearer_token'])
|
||||
},
|
||||
timeout=60
|
||||
)
|
||||
current_app.logger.info('send_delivery_status_to_service sending {} to {}, response {}'.format(
|
||||
notification_id,
|
||||
status_update['service_callback_api_url'],
|
||||
response.status_code
|
||||
))
|
||||
response.raise_for_status()
|
||||
except RequestException as e:
|
||||
current_app.logger.warning(
|
||||
"send_delivery_status_to_service request failed for service_id: {} and url: {}. exc: {}".format(
|
||||
notification_id,
|
||||
status_update['service_callback_api_url'],
|
||||
e
|
||||
)
|
||||
)
|
||||
if not isinstance(e, HTTPError) or e.response.status_code >= 500:
|
||||
try:
|
||||
self.retry(queue=QueueNames.RETRY)
|
||||
except self.MaxRetriesExceededError:
|
||||
current_app.logger.exception(
|
||||
'Retry: send_delivery_status_to_service has retried the max num of times')
|
||||
|
||||
|
||||
def process_update_with_notification_id(self, notification_id):
|
||||
retry = False
|
||||
try:
|
||||
# TODO: do we need to do rate limit this?
|
||||
notification = get_notification_by_id(notification_id)
|
||||
service_callback_api = get_service_callback_api_for_service(service_id=notification.service_id)
|
||||
if not service_callback_api:
|
||||
@@ -41,9 +93,9 @@ def send_delivery_status_to_service(self, notification_id):
|
||||
"reference": str(notification.client_reference),
|
||||
"to": notification.to,
|
||||
"status": notification.status,
|
||||
"created_at": notification.created_at.strftime(DATETIME_FORMAT), # the time service sent the request
|
||||
"completed_at": notification.updated_at.strftime(DATETIME_FORMAT), # the last time the status was updated
|
||||
"sent_at": notification.sent_at.strftime(DATETIME_FORMAT), # the time the email was sent
|
||||
"created_at": notification.created_at.strftime(DATETIME_FORMAT),
|
||||
"completed_at": notification.updated_at.strftime(DATETIME_FORMAT),
|
||||
"sent_at": notification.sent_at.strftime(DATETIME_FORMAT),
|
||||
"notification_type": notification.notification_type
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user