darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-07 07:48:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

try flask caching to get around SQLAlchemy problem with cachetools

Browse Source
This commit is contained in:
Kenneth Kehl
2025-09-26 11:09:21 -07:00
parent e0a1d15f32
commit 87094c26b9
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/job/rest.py
View File

@@ -1,3 +1,4 @@
import os
from zoneinfo import ZoneInfo from zoneinfo import ZoneInfo
import dateutil import dateutil
@@ -48,9 +49,11 @@ cache = Cache(config={"CACHE_TYPE": "SimpleCache"})
cache.init_app(app) cache.init_app(app)
register_errors(job_blueprint) register_errors(job_blueprint)
cache_timeout = os.getenv("REST_CACHE_TIMEOUT", 10)
@job_blueprint.route("/<job_id>", methods=["GET"]) @job_blueprint.route("/<job_id>", methods=["GET"])
@cache.cached(timeout=30) @cache.cached(timeout=cache_timeout)
def get_job_by_service_and_job_id(service_id, job_id): def get_job_by_service_and_job_id(service_id, job_id):
current_app.logger.info(hilite("ENTER get_job_by_service_and_job_id")) current_app.logger.info(hilite("ENTER get_job_by_service_and_job_id"))
check_suspicious_id(service_id, job_id) check_suspicious_id(service_id, job_id)
@@ -77,6 +80,7 @@ def cancel_job(service_id, job_id):
@job_blueprint.route("/<job_id>/notifications", methods=["GET"]) @job_blueprint.route("/<job_id>/notifications", methods=["GET"])
@cache.cached(timeout=cache_timeout)
def get_all_notifications_for_service_job(service_id, job_id): def get_all_notifications_for_service_job(service_id, job_id):
check_suspicious_id(service_id, job_id) check_suspicious_id(service_id, job_id)
@@ -137,6 +141,7 @@ def get_all_notifications_for_service_job(service_id, job_id):
@job_blueprint.route("/<job_id>/recent_notifications", methods=["GET"]) @job_blueprint.route("/<job_id>/recent_notifications", methods=["GET"])
@cache.cached(timeout=cache_timeout)
def get_recent_notifications_for_service_job(service_id, job_id): def get_recent_notifications_for_service_job(service_id, job_id):
current_app.logger.info(hilite("ENTER get_recent_notifications_for_service_job")) current_app.logger.info(hilite("ENTER get_recent_notifications_for_service_job"))
@@ -206,6 +211,7 @@ def get_recent_notifications_for_service_job(service_id, job_id):
@job_blueprint.route("/<job_id>/notification_count", methods=["GET"]) @job_blueprint.route("/<job_id>/notification_count", methods=["GET"])
@cache.cached(timeout=cache_timeout)
def get_notification_count_for_job_id(service_id, job_id): def get_notification_count_for_job_id(service_id, job_id):
check_suspicious_id(service_id, job_id) check_suspicious_id(service_id, job_id)
@@ -215,6 +221,7 @@ def get_notification_count_for_job_id(service_id, job_id):
@job_blueprint.route("", methods=["GET"]) @job_blueprint.route("", methods=["GET"])
@cache.cached(timeout=cache_timeout)
def get_jobs_by_service(service_id): def get_jobs_by_service(service_id):
check_suspicious_id(service_id) check_suspicious_id(service_id)
if request.args.get("limit_days"): if request.args.get("limit_days"):