feat(billing): Uptime DataCategory (#4363)

getsentry · Dec 11, 2024 · 21aad1e · 21aad1e
1 parent d4b0730
commit 21aad1e
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -21,6 +21,7 @@
 
 **Internal**:
 
+- Adds the UPTIME DataCategory ([#4363](https://github.com/getsentry/relay/pull/4363))
 - Promote some `span.data` fields to the top level. ([#4298](https://github.com/getsentry/relay/pull/4298))
 - Remove metrics summaries. ([#4278](https://github.com/getsentry/relay/pull/4278), [#4279](https://github.com/getsentry/relay/pull/4279), [#4296](https://github.com/getsentry/relay/pull/4296))
 - Use async `redis` for `projectconfig`. ([#4284](https://github.com/getsentry/relay/pull/4284))

diff --git a/py/sentry_relay/consts.py b/py/sentry_relay/consts.py
@@ -30,6 +30,7 @@ class DataCategory(IntEnum):
     PROFILE_CHUNK = 18
     METRIC_SECOND = 19
     REPLAY_VIDEO = 20
+    UPTIME = 21
     UNKNOWN = -1
     # end generated
 

diff --git a/relay-base-schema/src/data_category.rs b/relay-base-schema/src/data_category.rs
@@ -88,6 +88,8 @@ pub enum DataCategory {
     ///
     /// This is the data category for Session Replays produced via a video recording.
     ReplayVideo = 20,
+    /// This is the data category for Uptime monitors.
+    Uptime = 21,
     //
     // IMPORTANT: After adding a new entry to DataCategory, go to the `relay-cabi` subfolder and run
     // `make header` to regenerate the C-binding. This allows using the data category from Python.
@@ -125,6 +127,7 @@ impl DataCategory {
             "profile_chunk" => Self::ProfileChunk,
             "metric_second" => Self::MetricSecond,
             "replay_video" => Self::ReplayVideo,
+            "uptime" => Self::Uptime,
             _ => Self::Unknown,
         }
     }
@@ -154,6 +157,7 @@ impl DataCategory {
             Self::ProfileChunk => "profile_chunk",
             Self::MetricSecond => "metric_second",
             Self::ReplayVideo => "replay_video",
+            Self::Uptime => "uptime",
             Self::Unknown => "unknown",
         }
     }

diff --git a/relay-cabi/include/relay.h b/relay-cabi/include/relay.h
@@ -134,6 +134,11 @@ enum RelayDataCategory {
    * This is the data category for Session Replays produced via a video recording.
    */
   RELAY_DATA_CATEGORY_REPLAY_VIDEO = 20,
+  /**
+   * Uptime Monitor
+   *
+   */
+  RELAY_DATA_CATEGORY_UPTIME = 21,
   /**
    * Any other data category not known by this Relay.
    */

diff --git a/relay-quotas/src/quota.rs b/relay-quotas/src/quota.rs
@@ -197,6 +197,7 @@ impl CategoryUnit {
             | DataCategory::MetricBucket
             | DataCategory::UserReportV2
             | DataCategory::ProfileChunk
+            | DataCategory::Uptime
             | DataCategory::MetricSecond => Some(Self::Count),
             DataCategory::Attachment => Some(Self::Bytes),
             DataCategory::Session => Some(Self::Batched),