fix(issues): Replace 'Internal' Error message on Postgres query timeout errors

Handle Postgres statement timeout errors on issue search which previously displayed 'Internal Error' but
will now show more appropriate message explaining the query took too long and to possibly revise it

Closes: #80166

Author: yuvmen

src/sentry/api/utils.py

@@ -9,8 +9,10 @@
 from datetime import timedelta
 from typing import Any, Literal, overload
 
+import psycopg2.errorcodes
 import sentry_sdk
 from django.conf import settings
+from django.db.utils import OperationalError
 from django.http import HttpRequest
 from django.utils import timezone
 from rest_framework.exceptions import APIException, ParseError, Throttled
@@ -423,6 +425,14 @@ def handle_query_errors() -> Generator[None]:
         else:
             sentry_sdk.capture_exception(error)
         raise APIException(detail=message)
+    except OperationalError as error:
+        if hasattr(error, "pgcode") and error.pgcode == psycopg2.errorcodes.QUERY_CANCELED:
+            sentry_sdk.set_tag("query.error_reason", "Postgres statement timeout")
+            raise Throttled(
+                detail="Query timeout. Please try with a smaller date range or fewer conditions."
+            )
+        # Let other OperationalErrors propagate as normal
+        raise
 
 
 def update_snuba_params_with_timestamp(

tests/sentry/api/test_utils.py

@@ -2,7 +2,9 @@
 import unittest
 from unittest.mock import MagicMock, patch
 
+import psycopg2
 import pytest
+from django.db import OperationalError
 from django.utils import timezone
 from rest_framework.exceptions import APIException
 from sentry_sdk import Scope
@@ -168,13 +170,12 @@ class FooBarError(Exception):
     pass
 
 
-class HandleQueryErrorsTest:
+class HandleQueryErrorsTest(APITestCase):
     @patch("sentry.api.utils.ParseError")
     def test_handle_query_errors(self, mock_parse_error):
         exceptions = [
             DatasetSelectionError,
             IncompatibleMetricsQuery,
-            InvalidParams,
             InvalidSearchQuery,
             QueryConnectionFailed,
             QueryExecutionError,
@@ -198,6 +199,35 @@ def test_handle_query_errors(self, mock_parse_error):
             except Exception as e:
                 assert isinstance(e, (FooBarError, APIException))
 
+    @patch("sentry.api.utils.Throttled")
+    def test_handle_postgres_timeout(self, mock_throttled):
+        class TimeoutError(OperationalError):
+            pgcode = psycopg2.errorcodes.QUERY_CANCELED
+
+        mock_throttled.return_value = FooBarError()
+
+        try:
+            with handle_query_errors():
+                raise TimeoutError()
+        except Exception as e:
+            assert isinstance(e, FooBarError)
+            mock_throttled.assert_called_once_with(
+                detail="Query timeout. Please try with a smaller date range or fewer conditions."
+            )
+
+    @patch("sentry.api.utils.ParseError")
+    def test_handle_other_operational_error(self, mock_parse_error):
+        class OtherError(OperationalError):
+            # No pgcode attribute
+            pass
+
+        try:
+            with handle_query_errors():
+                raise OtherError()
+        except Exception as e:
+            assert isinstance(e, OtherError)  # Should propagate original error
+            mock_parse_error.assert_not_called()
+
 
 class ClampDateRangeTest(unittest.TestCase):
     def test_no_clamp_if_range_under_max(self):