[DEV-5042] Optimize /filter/jira cache performance

athenianco · Sep 23, 2022 · 6c2418e · 6c2418e
1 parent 49e8765
commit 6c2418e
Show file tree

Hide file tree

Showing 4 changed files with 550 additions and 4 deletions.
diff --git a/server/athenian/api/controllers/jira_controller.py b/server/athenian/api/controllers/jira_controller.py
@@ -91,6 +91,7 @@
     JIRAUser,
     PullRequest as WebPullRequest,
 )
+from athenian.api.models.web_model_io import deserialize_models, serialize_models
 from athenian.api.request import AthenianWebRequest
 from athenian.api.response import ResponseError, model_response
 from athenian.api.tracing import sentry_span
@@ -224,8 +225,8 @@ async def filter_jira_stuff(request: AthenianWebRequest, body: dict) -> web.Resp
 @sentry_span
 @cached(
     exptime=short_term_exptime,
-    serialize=pickle.dumps,
-    deserialize=pickle.loads,
+    serialize=serialize_models,
+    deserialize=deserialize_models,
     key=lambda return_, time_from, time_to, exclude_inactive, label_filter, priorities, reporters, assignees, commenters, default_branches, release_settings, logical_settings, **_: (  # noqa
         JIRAFilterReturn.EPICS in return_,
         JIRAFilterReturn.PRIORITIES in return_,
@@ -548,8 +549,8 @@ async def _epic_flow(
 @sentry_span
 @cached(
     exptime=short_term_exptime,
-    serialize=pickle.dumps,
-    deserialize=pickle.loads,
+    serialize=serialize_models,
+    deserialize=deserialize_models,
     key=lambda return_, time_from, time_to, exclude_inactive, label_filter, priorities, reporters, assignees, commenters, default_branches, release_settings, logical_settings, **_: (  # noqa
         JIRAFilterReturn.ISSUES in return_,
         JIRAFilterReturn.ISSUE_BODIES in return_,