Skip to content

Commit

Permalink
code
Browse files Browse the repository at this point in the history
  • Loading branch information
Rachel Chen authored and Rachel Chen committed Jul 2, 2024
1 parent f8e1456 commit e2666d4
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
3 changes: 3 additions & 0 deletions snuba/web/db_query.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e2666d4

Please sign in to comment.