code (#6081)

Co-authored-by: Rachel Chen <rachelchen@PL6VFX9HP4.attlocal.net>
getsentry · Jul 2, 2024 · 8a00dca · 8a00dca
1 parent 63836d1
commit 8a00dca
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/snuba/query/allocation_policies/bytes_scanned_window_policy.py b/snuba/query/allocation_policies/bytes_scanned_window_policy.py
@@ -216,6 +216,10 @@ def _get_quota_allowance(
  granted_quota = granted_quotas[0]
  is_throttled = False
  if granted_quota.granted <= 0:
+ self.metrics.increment(
+ "bytes_scanned_window_policy_throttle",
+ tags={"referrer": str(referrer)},
+ )
  is_throttled = True
  explanation[
  "reason"

diff --git a/snuba/web/db_query.py b/snuba/web/db_query.py
@@ -901,6 +901,9 @@ def _apply_allocation_policies_quota(
  _add_quota_info(summary, _THROTTLED_BY, throttle_quota_and_policy)
  stats["quota_allowance"]["summary"] = summary
 
+ if throttle_quota_and_policy is not None:
+ metrics.increment("db_query_throttled_query")
+
  if not can_run:
  raise AllocationPolicyViolations.from_args(stats["quota_allowance"])
  # Before allocation policies were a thing, the query pipeline would apply