feat(telemetry): add telemetry data point when checking license (#18371)

Signed-off-by: tabVersion <[email protected]>

src/license/Cargo.toml

@@ -15,7 +15,10 @@ ignored = ["workspace-hack"]
 normal = ["workspace-hack"]
 
 [dependencies]
+jsonbb = { workspace = true }
 jsonwebtoken = "9"
+risingwave_pb = { workspace = true }
+risingwave_telemetry_event = { workspace = true }
 serde = { version = "1", features = ["derive"] }
 thiserror = "1"
 thiserror-ext = { workspace = true }

src/license/src/feature.rs

@@ -14,7 +14,7 @@
 
 use thiserror::Error;
 
-use super::{License, LicenseKeyError, LicenseManager, Tier};
+use super::{report_telemetry, License, LicenseKeyError, LicenseManager, Tier};
 
 /// Define all features that are available based on the tier of the license.
 ///
@@ -84,6 +84,14 @@ macro_rules! def_feature {
                     )*
                 }
             }
+
+            fn get_feature_name(&self) -> &'static str {
+                match &self {
+                    $(
+                        Self::$name => stringify!($name),
+                    )*
+                }
+            }
         }
     };
 }
@@ -113,7 +121,7 @@ pub enum FeatureNotAvailable {
 impl Feature {
     /// Check whether the feature is available based on the current license.
     pub fn check_available(self) -> Result<(), FeatureNotAvailable> {
-        match LicenseManager::get().license() {
+        let check_res = match LicenseManager::get().license() {
             Ok(license) => {
                 if license.tier >= self.min_tier() {
                     Ok(())
@@ -136,6 +144,10 @@ impl Feature {
                     })
                 }
             }
-        }
+        };
+
+        report_telemetry(&self, self.get_feature_name(), check_res.is_ok());
+
+        check_res
     }
 }

src/license/src/lib.rs

@@ -20,3 +20,26 @@ mod manager;
 pub use feature::*;
 pub use key::*;
 pub use manager::*;
+use risingwave_pb::telemetry::PbTelemetryEventStage;
+use risingwave_telemetry_event::report_event_common;
+
+pub(crate) fn report_telemetry(feature: &Feature, feature_name: &str, success_flag: bool) {
+    if matches!(feature, Feature::TestPaid) {
+        let mut attr_builder = jsonbb::Builder::<Vec<u8>>::new();
+        attr_builder.begin_object();
+        attr_builder.add_string("success");
+        attr_builder.add_value(jsonbb::ValueRef::Bool(success_flag));
+        attr_builder.end_object();
+        let attr = attr_builder.finish();
+
+        report_event_common(
+            PbTelemetryEventStage::Unspecified,
+            feature_name,
+            0,
+            None,
+            None,
+            Some(attr),
+            "paywall".to_string(),
+        );
+    }
+}