feat: add track event api

Author: gangb-tech

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 edition = "2021"
 name = "feature-probe-server-sdk"
-version = "1.2.12"
+version = "2.0.1"
 license = "Apache-2.0"
 authors = ["[email protected]"]
 description = "FeatureProbe Server Side SDK for Rust"
@@ -44,8 +44,8 @@ thiserror = "1.0"
 tracing = "0.1"
 url = "2"
 
-feature-probe-event-std = { optional = true, version = "1.1.2", package = "feature-probe-event" }
-feature-probe-event-tokio = { optional = true, version = "1.1.2", features = [
+feature-probe-event-std = { optional = true, version = "1.1.3", package = "feature-probe-event" }
+feature-probe-event-tokio = { optional = true, version = "1.1.3", features = [
   "use_tokio",
 ], default-features = false, package = "feature-probe-event" }
 
@@ -70,5 +70,5 @@ criterion = "0.4"
 rusty-hook = "^0.11.2"
 tokio = { version = "1", features = ["full"] }
 tracing-subscriber = "0.3"
-feature-probe-server = { version = "1.3.11", features = ["realtime"] }
+feature-probe-server = { version = "2.0.1", features = ["realtime"] }
 

Cargo.toml

@@ -184,6 +184,10 @@ impl Toggle {
         self.default_serve.select_variation_value(&eval_param)
     }
 
+    pub fn track_access_events(&self) -> bool {
+        self.track_access_events.unwrap_or(false)
+    }
+
     pub fn eval_detail(
         &self,
         user: &FPUser,

src/evaluate.rs

@@ -1,12 +1,3 @@
-use parking_lot::RwLock;
-use serde_json::Value;
-#[cfg(all(feature = "use_tokio", feature = "realtime"))]
-use socketio_rs::Client;
-use std::collections::HashMap;
-use std::fmt::Debug;
-use std::sync::Arc;
-use tracing::trace;
-
 use crate::sync::SyncType;
 use crate::{
     config::Config,
@@ -18,18 +9,33 @@ use crate::{FPDetail, SdkAuthorization, Toggle};
 #[cfg(feature = "event")]
 use feature_probe_event_std::event::AccessEvent;
 #[cfg(feature = "event")]
+use feature_probe_event_std::event::CustomEvent;
+#[cfg(feature = "event")]
+use feature_probe_event_std::event::Event;
+#[cfg(feature = "event")]
 use feature_probe_event_std::recorder::unix_timestamp;
 #[cfg(feature = "event")]
 use feature_probe_event_std::recorder::EventRecorder;
 #[cfg(feature = "event_tokio")]
 use feature_probe_event_tokio::event::AccessEvent;
 #[cfg(feature = "event_tokio")]
+use feature_probe_event_tokio::event::CustomEvent;
+#[cfg(feature = "event_tokio")]
+use feature_probe_event_tokio::event::Event;
+#[cfg(feature = "event_tokio")]
 use feature_probe_event_tokio::recorder::unix_timestamp;
 #[cfg(feature = "event_tokio")]
 use feature_probe_event_tokio::recorder::EventRecorder;
 #[cfg(all(feature = "use_tokio", feature = "realtime"))]
 use futures_util::FutureExt;
-
+use parking_lot::RwLock;
+use serde_json::Value;
+#[cfg(all(feature = "use_tokio", feature = "realtime"))]
+use socketio_rs::Client;
+use std::collections::HashMap;
+use std::fmt::Debug;
+use std::sync::Arc;
+use tracing::{trace, warn};
 #[cfg(all(feature = "use_tokio", feature = "realtime"))]
 type SocketCallback = std::pin::Pin<Box<dyn futures_util::Future<Output = ()> + Send>>;
 
@@ -122,6 +128,24 @@ impl FeatureProbe {
         self.generic_detail(toggle, user, default, Some)
     }
 
+    pub fn track(&self, event_name: &str, user: &FPUser, value: Option<f64>) {
+        let recorder = match self.event_recorder.as_ref() {
+            None => {
+                warn!("Event Recorder no ready.");
+                return;
+            }
+            Some(recorder) => recorder,
+        };
+        let event = CustomEvent {
+            kind: "custom".to_string(),
+            time: unix_timestamp(),
+            user: user.key(),
+            name: event_name.to_string(),
+            value,
+        };
+        recorder.record_event(Event::CustomEvent(event));
+    }
+
     pub fn new_with(server_key: String, repo: Repository) -> Self {
         Self {
             config: Config {
@@ -198,26 +222,39 @@ impl FeatureProbe {
             .toggles
             .get(toggle)
             .map(|toggle| toggle.eval_detail(user, &repo.segments));
-
+        let track_access_events = match repo.toggles.get(toggle) {
+            Some(toggle) => toggle.track_access_events(),
+            None => false,
+        };
         #[cfg(any(feature = "event", feature = "event_tokio"))]
-        self.record_detail(toggle, &detail);
-
+        self.record_detail(toggle, user, track_access_events, &detail);
         detail
     }
 
     #[cfg(any(feature = "event", feature = "event_tokio"))]
-    fn record_detail(&self, toggle: &str, detail: &Option<EvalDetail<Value>>) -> Option<()> {
+    fn record_detail(
+        &self,
+        toggle: &str,
+        user: &FPUser,
+        track_access_events: bool,
+        detail: &Option<EvalDetail<Value>>,
+    ) -> Option<()> {
         let recorder = self.event_recorder.as_ref()?;
         let detail = detail.as_ref()?;
         let value = detail.value.as_ref()?;
-        recorder.record_access(AccessEvent {
+        let event = AccessEvent {
+            kind: "access".to_string(),
             time: unix_timestamp(),
             key: toggle.to_owned(),
+            user: user.key(),
             value: value.clone(),
-            index: detail.variation_index,
+            variation_index: detail.variation_index.unwrap(),
             version: detail.version,
-            reason: detail.reason.clone(),
-        });
+            rule_index: detail.rule_index,
+            reason: Some(detail.reason.to_string()),
+            track_access_events,
+        };
+        recorder.record_event(Event::AccessEvent(event));
         None
     }
 

src/feature_probe.rs

@@ -89,6 +89,7 @@ async fn setup_server(api_port: u16, server_port: u16, realtime_port: u16, realt
     let events_url: Url = format!("http://0.0.0.0:{}/api/events", api_port)
         .parse()
         .unwrap();
+    let analysis_url = None;
     let refresh_interval = Duration::from_secs(1);
     let config = ServerConfig {
         toggles_url,
@@ -97,6 +98,7 @@ async fn setup_server(api_port: u16, server_port: u16, realtime_port: u16, realt
         realtime_path,
         refresh_interval,
         keys_url: None,
+        analysis_url: None,
         events_url: events_url.clone(),
         client_sdk_key: Some(client_sdk_key.clone()),
         server_sdk_key: Some(server_sdk_key.clone()),
@@ -108,6 +110,7 @@ async fn setup_server(api_port: u16, server_port: u16, realtime_port: u16, realt
     let feature_probe_server = FpHttpHandler {
         repo: repo.clone(),
         events_url,
+        analysis_url,
         events_timeout: Duration::from_secs(1),
         http_client: Default::default(),
     };