From a3aafb14d4c31e71093a0236972488dd315b1510 Mon Sep 17 00:00:00 2001
From: Riccardo Busetti <riccardob36@gmail.com>
Date: Thu, 7 Dec 2023 08:13:08 +0100
Subject: [PATCH] feat(on-demand): Add support for transaction.source in event
 getter (#2820)

---
 CHANGELOG.md                             |  1 +
 relay-event-schema/src/protocol/event.rs | 21 +++++++++++++++++++--
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 002ad8e293..4d4928a812 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,6 +11,7 @@
 - Add Mixed JS/Android Profiles events processing. ([#2706](https://github.com/getsentry/relay/pull/2706))
 - Allow to ingest measurements on a span. ([#2792](https://github.com/getsentry/relay/pull/2792))
 - Extract size metrics for all resource spans when permitted. ([#2805](https://github.com/getsentry/relay/pull/2805))
+- Allow access to more fields in dynamic sampling and metric extraction. ([#2820](https://github.com/getsentry/relay/pull/2820))
 
 **Bug Fixes**:
 
diff --git a/relay-event-schema/src/protocol/event.rs b/relay-event-schema/src/protocol/event.rs
index eb7e216990..b048ed966f 100644
--- a/relay-event-schema/src/protocol/event.rs
+++ b/relay-event-schema/src/protocol/event.rs
@@ -660,6 +660,13 @@ impl Getter for Event {
             "user.geo.subdivision" => self.user.value()?.geo.value()?.subdivision.as_str()?.into(),
             "request.method" => self.request.value()?.method.as_str()?.into(),
             "request.url" => self.request.value()?.url.as_str()?.into(),
+            "transaction.source" => self
+                .transaction_info
+                .value()?
+                .source
+                .value()?
+                .as_str()
+                .into(),
             "sdk.name" => self.client_sdk.value()?.name.as_str()?.into(),
             "sdk.version" => self.client_sdk.value()?.version.as_str()?.into(),
 
@@ -794,7 +801,9 @@ mod tests {
     use uuid::uuid;
 
     use super::*;
-    use crate::protocol::{Headers, IpAddr, JsonLenientString, PairList, TagEntry};
+    use crate::protocol::{
+        Headers, IpAddr, JsonLenientString, PairList, TagEntry, TransactionSource,
+    };
 
     #[test]
     fn test_event_roundtrip() {
@@ -1096,6 +1105,10 @@ mod tests {
                 ..Default::default()
             }),
             transaction: Annotated::new("some-transaction".into()),
+            transaction_info: Annotated::new(TransactionInfo {
+                source: Annotated::new(TransactionSource::Route),
+                ..Default::default()
+            }),
             tags: {
                 let items = vec![Annotated::new(TagEntry(
                     Annotated::new("custom".to_string()),
@@ -1214,7 +1227,11 @@ mod tests {
         assert_eq!(
             Some(Val::Uuid(uuid!("abadcade-feed-dead-beef-8addadfeedaa"))),
             event.get_value("event.contexts.profile.profile_id")
-        )
+        );
+        assert_eq!(
+            Some(Val::String("route")),
+            event.get_value("event.transaction.source")
+        );
     }
 
     #[test]