feat(CapMan): More tenant_ids for TSDB Queries (#45274)

### Overview
- Added `organization_id` to more Snuba Requests, specifically targeting
`tsdb-modelid:4` queries as they are the biggest chunk of queries Snuba
is handling in production
- Most calls to `tsdb.get_range` handled as well, some I wasn't able to
figure out
- More context: #44788

Author: rahul-kumar-saini

src/sentry/api/endpoints/group_details.py

@@ -120,7 +120,11 @@ def _get_context_plugins(self, request: Request, group):
 
     @staticmethod
     def __group_hourly_daily_stats(group: Group, environment_ids: Sequence[int]):
-        get_range = functools.partial(tsdb.get_range, environment_ids=environment_ids)
+        get_range = functools.partial(
+            tsdb.get_range,
+            environment_ids=environment_ids,
+            tenant_ids={"organization_id": group.project.organization_id},
+        )
         model = get_issue_tsdb_group_model(group.issue_category)
         now = timezone.now()
         hourly_stats = tsdb.rollup(

src/sentry/api/endpoints/group_stats.py

@@ -19,7 +19,10 @@ def get(self, request: Request, group) -> Response:
             raise ResourceDoesNotExist
 
         data = tsdb.get_range(
-            model=tsdb.models.group, keys=[group.id], **self._parse_args(request, environment_id)
+            model=tsdb.models.group,
+            keys=[group.id],
+            **self._parse_args(request, environment_id),
+            tenant_ids={"organization_id": group.project.organization_id},
         )[group.id]
 
         return Response(data)

src/sentry/api/endpoints/integrations/sentry_apps/interaction.py

@@ -29,7 +29,10 @@ def get(self, request: Request, sentry_app) -> Response:
         """
 
         views = tsdb.get_range(
-            model=tsdb.models.sentry_app_viewed, keys=[sentry_app.id], **self._parse_args(request)
+            model=tsdb.models.sentry_app_viewed,
+            keys=[sentry_app.id],
+            **self._parse_args(request),
+            tenant_ids={"organization_id": sentry_app.owner.id},
         )[sentry_app.id]
 
         component_interactions = tsdb.get_range(
@@ -39,6 +42,7 @@ def get(self, request: Request, sentry_app) -> Response:
                 for component in sentry_app.components.all()
             ],
             **self._parse_args(request),
+            tenant_ids={"organization_id": sentry_app.owner.id},
         )
 
         return Response(

src/sentry/api/endpoints/organization_stats.py

@@ -89,7 +89,10 @@ def get(self, request: Request, organization) -> Response:
         if stat_model is None:
             raise ValueError(f"Invalid group: {group}, stat: {stat}")
         data = tsdb.get_range(
-            model=stat_model, keys=keys, **self._parse_args(request, **query_kwargs)
+            model=stat_model,
+            keys=keys,
+            **self._parse_args(request, **query_kwargs),
+            tenant_ids={"organization_id": organization.id},
         )
 
         if group == "organization":

src/sentry/api/endpoints/project_group_stats.py

@@ -37,7 +37,10 @@ def get(self, request: Request, project) -> Response:
             return Response(status=204)
 
         data = tsdb.get_range(
-            model=tsdb.models.group, keys=group_ids, **self._parse_args(request, environment_id)
+            model=tsdb.models.group,
+            keys=group_ids,
+            **self._parse_args(request, environment_id),
+            tenant_ids={"organization_id": project.organization_id},
         )
 
         return Response({str(k): v for k, v in data.items()})

src/sentry/api/endpoints/project_servicehook_stats.py

@@ -20,7 +20,12 @@ def get(self, request: Request, project, hook_id) -> Response:
 
         stats = {}
         for model, name in ((tsdb.models.servicehook_fired, "total"),):
-            result = tsdb.get_range(model=model, keys=[hook.id], **stat_args)[hook.id]
+            result = tsdb.get_range(
+                model=model,
+                keys=[hook.id],
+                **stat_args,
+                tenant_ids={"organization_id": project.organization_id},
+            )[hook.id]
             for ts, count in result:
                 stats.setdefault(int(ts), {})[name] = count
 

src/sentry/api/endpoints/project_stats.py

@@ -62,7 +62,10 @@ def get(self, request: Request, project) -> Response:
                 raise ValueError("Invalid stat: %s" % stat)
 
         data = tsdb.get_range(
-            model=stat_model, keys=[project.id], **self._parse_args(request, **query_kwargs)
+            model=stat_model,
+            keys=[project.id],
+            **self._parse_args(request, **query_kwargs),
+            tenant_ids={"organization_id": project.organization_id},
         )[project.id]
 
         return Response(data)

src/sentry/api/endpoints/team_stats.py

@@ -51,6 +51,7 @@ def get(self, request: Request, team) -> Response:
                 model=tsdb.models.project,
                 keys=[p.id for p in projects],
                 **self._parse_args(request, environment_id),
+                tenant_ids={"organization_id": team.organization_id},
             ).values()
         )
 

src/sentry/api/serializers/models/group_stream.py

@@ -153,11 +153,13 @@ def query_tsdb(self, groups: Sequence[Group], query_params, **kwargs):
         except Environment.DoesNotExist:
             stats = {g.id: tsdb.make_series(0, **query_params) for g in groups}
         else:
+            org_id = groups[0].project.organization_id if groups else None
             stats = tsdb.get_range(
                 model=tsdb.models.group,
                 keys=[g.id for g in groups],
                 environment_ids=environment and [environment.id],
                 **query_params,
+                tenant_ids={"organization_id": org_id} if org_id else None,
             )
 
         return stats
@@ -344,6 +346,7 @@ def query_tsdb(
                 generic_issue_ids.append(group.id)
 
         results = {}
+
         get_range = functools.partial(
             snuba_tsdb.get_range,
             environment_ids=environment_ids,

src/sentry/digests/notifications.py

@@ -82,7 +82,13 @@ def fetch_state(project: Project, records: Sequence[Record]) -> Mapping[str, Any
         "rules": Rule.objects.in_bulk(
             itertools.chain.from_iterable(record.value.rules for record in records)
         ),
-        "event_counts": tsdb.get_sums(tsdb.models.group, list(groups.keys()), start, end),
+        "event_counts": tsdb.get_sums(
+            tsdb.models.group,
+            list(groups.keys()),
+            start,
+            end,
+            tenant_ids={"organization_id": project.organization_id},
+        ),
         "user_counts": tsdb.get_distinct_counts_totals(
             tsdb.models.users_affected_by_group, list(groups.keys()), start, end
         ),