fix(cbrs): set request in routing decision (#7241)

fixes
https://sentry.sentry.io/issues/6678390448/?project=300688&query=&referrer=issue-stream&stream_index=1

---------

Co-authored-by: Rachel Chen <[email protected]>
Co-authored-by: Rachel Chen <[email protected]>

Author: 3 people

snuba/web/rpc/__init__.py

@@ -157,7 +157,9 @@ def execute(self, in_msg: Tin) -> Tout:
         span = scope.span
         if span is not None:
             span.description = self.config_key()
-        self.routing_context = RoutingContext(timer=self._timer, in_msg=in_msg)
+        self.routing_decision.routing_context = RoutingContext(
+            timer=self._timer, in_msg=in_msg
+        )
 
         self.__before_execute(in_msg)
         error = None
@@ -218,7 +220,7 @@ def __before_execute(self, in_msg: Tin) -> None:
         )
         self.routing_decision.strategy = selected_strategy
         self.routing_decision = selected_strategy.get_routing_decision(
-            self.routing_context
+            self.routing_decision.routing_context
         )
         self._timer.mark("rpc_start")
         self._before_execute(in_msg)

snuba/web/rpc/storage_routing/routing_strategy_selector.py

@@ -6,7 +6,9 @@
 import sentry_sdk
 from google.protobuf.message import Message as ProtobufMessage
 
+from snuba import settings
 from snuba.state import get_config
+from snuba.web.rpc.storage_routing.common import extract_message_meta
 from snuba.web.rpc.storage_routing.routing_strategies.outcomes_based import (
     OutcomesBasedRoutingStrategy,
 )
@@ -88,7 +90,8 @@ class RoutingStrategySelector:
     def get_storage_routing_config(
         self, in_msg: ProtobufMessage
     ) -> StorageRoutingConfig:
-        organization_id = str(in_msg.meta.organization_id)  # type: ignore
+        in_msg_meta = extract_message_meta(in_msg)
+        organization_id = str(in_msg_meta.organization_id)
         try:
             overrides = json.loads(
                 str(get_config(_STORAGE_ROUTING_CONFIG_OVERRIDE_KEY, "{}"))
@@ -106,7 +109,8 @@ def select_routing_strategy(
         self, routing_context: RoutingContext
     ) -> BaseRoutingStrategy:
         try:
-            combined_org_and_project_ids = f"{routing_context.in_msg.meta.organization_id}:{'.'.join(str(pid) for pid in sorted(routing_context.in_msg.meta.project_ids))}"  # type: ignore
+            in_msg_meta = extract_message_meta(routing_context.in_msg)
+            combined_org_and_project_ids = f"{in_msg_meta.organization_id}:{'.'.join(str(pid) for pid in sorted(in_msg_meta.project_ids))}"
             bucket = (
                 int(hashlib.md5(combined_org_and_project_ids.encode()).hexdigest(), 16)
                 % _NUM_BUCKETS
@@ -121,6 +125,8 @@ def select_routing_strategy(
                 if bucket < cumulative_buckets:
                     return BaseRoutingStrategy.get_from_name(strategy_name)()
         except Exception as e:
+            if settings.RAISE_ON_ROUTING_STRATEGY_FAILURES:
+                raise e
             sentry_sdk.capture_message(f"Error selecting routing strategy: {e}")
             return OutcomesBasedRoutingStrategy()