fix: wrap new TA impl in try except for cache rollups (#361)

Author: joseph-sentry

apps/worker/tasks/cache_test_rollups.py

@@ -2,6 +2,7 @@
 from typing import Literal
 
 import polars as pl
+import sentry_sdk
 from django.db import connections
 from redis.exceptions import LockError
 
@@ -125,10 +126,15 @@ def run_impl(
                 f"rollups:{repo_id}:{branch}", timeout=300, blocking_timeout=2
             ):
                 if impl_type == "new" or impl_type == "both":
-                    cache_rollups(repo_id, branch)
-                    cache_rollups(repo_id, None)
-                    if impl_type == "new":
-                        return {"success": True}
+                    try:
+                        cache_rollups(repo_id, branch)
+                        cache_rollups(repo_id, None)
+                        if impl_type == "new":
+                            return {"success": True}
+                    except Exception as e:
+                        sentry_sdk.capture_exception(e)
+                        if impl_type == "new":
+                            return {"success": False}
 
                 self.run_impl_within_lock(repo_id, branch)
 

apps/worker/tasks/process_flakes.py

@@ -1,6 +1,7 @@
 import logging
 from typing import Any, Literal, cast
 
+import sentry_sdk
 from redis import Redis
 from redis.exceptions import LockError
 from sqlalchemy.orm import Session
@@ -69,9 +70,14 @@ def run_impl(
         log.info("Received process flakes task")
 
         if impl_type == "new" or impl_type == "both":
-            process_flakes_for_repo(repo_id)
-            if impl_type == "new":
-                return {"successful": True}
+            try:
+                process_flakes_for_repo(repo_id)
+                if impl_type == "new":
+                    return {"successful": True}
+            except Exception as e:
+                sentry_sdk.capture_exception(e)
+                if impl_type == "new":
+                    return {"successful": False}
 
         redis_client = get_redis_connection()
         lock_name = LOCK_NAME.format(repo_id)