perf(sentry): Add optional referrer to query.search for granularity (#44585)

### Summary
This adds an optional referrer to split up the discover__search referrer
being passed to snuba, since many endpoints call the same query.search
function with potentially different behaviour.

Other:
- Not sure we want to keep the `_sample` behaviour for custom referrers
to split the use cases, leaving it here for now to preserve existing
behaviour.

Author: k-fish

src/sentry/api/endpoints/organization_group_index.py

@@ -175,6 +175,7 @@ def _search(
                 query_kwargs.pop("sort_by")
                 result = inbox_search(**query_kwargs)
             else:
+                query_kwargs["referrer"] = "search.group_index"
                 result = search.query(**query_kwargs)
             return result, query_kwargs
 

src/sentry/search/base.py

@@ -34,5 +34,6 @@ def query(
         date_from: Optional[datetime] = None,
         date_to: Optional[datetime] = None,
         max_hits: Optional[int] = None,
+        referrer: Optional[str] = None,
     ) -> CursorResult[Group]:
         raise NotImplementedError

src/sentry/search/snuba/backend.py

@@ -361,6 +361,7 @@ def query(
         date_from: Optional[datetime] = None,
         date_to: Optional[datetime] = None,
         max_hits: Optional[int] = None,
+        referrer: Optional[str] = None,
     ) -> CursorResult[Group]:
         search_filters = search_filters if search_filters is not None else []
 
@@ -414,6 +415,7 @@ def query(
             date_from=date_from,
             date_to=date_to,
             max_hits=max_hits,
+            referrer=referrer,
         )
 
     def _build_group_queryset(

src/sentry/search/snuba/executors.py

@@ -141,6 +141,7 @@ def query(
         date_from: Optional[datetime],
         date_to: Optional[datetime],
         max_hits: Optional[int] = None,
+        referrer: Optional[str] = None,
     ) -> CursorResult[Group]:
         """This function runs your actual query and returns the results
         We usually return a paginator object, which contains the results and the number of hits"""
@@ -297,6 +298,7 @@ def snuba_search(
         offset: int = 0,
         get_sample: bool = False,
         search_filters: Optional[Sequence[SearchFilter]] = None,
+        referrer: Optional[str] = None,
     ) -> Tuple[List[Tuple[int, Any]], int]:
         """Queries Snuba for events with associated Groups based on the input criteria.
 
@@ -315,7 +317,8 @@ def snuba_search(
                 ).values_list("name", flat=True)
             )
 
-        referrer = "search_sample" if get_sample else "search"
+        referrer = referrer or "search"
+        referrer = f"{referrer}_sample" if get_sample else referrer
 
         snuba_search_filters = [
             sf
@@ -465,6 +468,7 @@ def query(
         date_from: Optional[datetime],
         date_to: Optional[datetime],
         max_hits: Optional[int] = None,
+        referrer: Optional[str] = None,
     ) -> CursorResult[Group]:
         now = timezone.now()
         end = None
@@ -641,6 +645,7 @@ def query(
                 limit=chunk_limit,
                 offset=offset,
                 search_filters=search_filters,
+                referrer=referrer,
             )
             metrics.timing("snuba.search.num_snuba_results", len(snuba_groups))
             count = len(snuba_groups)
@@ -927,6 +932,7 @@ def query(
         date_from: Optional[datetime],
         date_to: Optional[datetime],
         max_hits: Optional[int] = None,
+        referrer: Optional[str] = None,
     ) -> CursorResult[Group]:
 
         if not validate_cdc_search_filters(search_filters):

src/sentry/snuba/referrer.py

@@ -462,6 +462,8 @@ class ReferrerBase(Enum):
     REPROCESSING2_START_GROUP_REPROCESSING = "reprocessing2.start_group_reprocessing"
     SEARCH_SAMPLE = "search_sample"
     SEARCH = "search"
+    SEARCH_GROUP_INDEX = "search.group_index"
+    SEARCH_GROUP_INDEX_SAMPLE = "search.group_index_sample"
     SERIALIZERS_GROUPSERIALIZERSNUBA__EXECUTE_ERROR_SEEN_STATS_QUERY = (
         "serializers.GroupSerializerSnuba._execute_error_seen_stats_query"
     )