code (#6088)

Co-authored-by: Rachel Chen <[email protected]>
getsentry · Jul 3, 2024 · 2cd5cc1 · 2cd5cc1
1 parent 1efe9ae
commit 2cd5cc1
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/snuba/web/db_query.py b/snuba/web/db_query.py
@@ -869,18 +869,19 @@ def _apply_allocation_policies_quota(
                 allowance = allocation_policy.get_quota_allowance(
                     attribution_info.tenant_ids, query_id
                 )
-                num_threads = min(num_threads, allowance.max_threads)
                 can_run &= allowance.can_run
                 quota_allowances[allocation_policy.config_key()] = allowance
                 span.set_data(
                     "quota_allowance",
                     quota_allowances[allocation_policy.config_key()],
                 )
                 if allowance.is_throttled:
-                    throttle_quota_and_policy = _QuotaAndPolicy(
-                        quota_allowance=allowance,
-                        policy_name=allocation_policy.config_key(),
-                    )
+                    if allowance.max_threads < num_threads:
+                        throttle_quota_and_policy = _QuotaAndPolicy(
+                            quota_allowance=allowance,
+                            policy_name=allocation_policy.config_key(),
+                        )
+                num_threads = min(num_threads, allowance.max_threads)
                 if not can_run:
                     rejection_quota_and_policy = _QuotaAndPolicy(
                         quota_allowance=allowance,