feat: usage & billing changes (#1033)

src/handler/http/request/search/mod.rs

@@ -26,8 +26,10 @@ use crate::common::json;
 use crate::infra::config::CONFIG;
 use crate::infra::{errors, metrics};
 use crate::meta::http::HttpResponse as MetaHttpResponse;
+use crate::meta::usage::{RequestStats, UsageType};
 use crate::meta::{self, StreamType};
 use crate::service::search as SearchService;
+use crate::service::usage::report_usage_stats;
 
 /** SearchStreamData*/
 #[utoipa::path(
@@ -154,6 +156,22 @@ pub async fn search(
                 ])
                 .inc();
             res.set_local_took(start.elapsed().as_millis() as usize, took_wait);
+
+            let req_stats = RequestStats {
+                records: res.hits.len() as u64,
+                response_time: time,
+                size: res.scan_size as f64,
+                request_body: Some(req.query.sql),
+            };
+            let num_fn = req.query.query_fn.is_some() as u16;
+            report_usage_stats(
+                req_stats,
+                &org_id,
+                StreamType::Logs,
+                UsageType::Search,
+                num_fn,
+            )
+            .await;
             Ok(HttpResponse::Ok().json(res))
         }
         Err(err) => {
@@ -427,6 +445,22 @@ pub async fn around(
         ])
         .inc();
 
+    let req_stats = RequestStats {
+        records: resp.hits.len() as u64,
+        response_time: time,
+        size: resp.scan_size as f64,
+        request_body: Some(req.query.sql),
+    };
+    let num_fn = req.query.query_fn.is_some() as u16;
+    report_usage_stats(
+        req_stats,
+        &org_id,
+        StreamType::Logs,
+        UsageType::SearchAround,
+        num_fn,
+    )
+    .await;
+
     Ok(HttpResponse::Ok().json(resp))
 }
 
@@ -610,6 +644,22 @@ pub async fn values(
         ])
         .inc();
 
+    let req_stats = RequestStats {
+        records: resp.hits.len() as u64,
+        response_time: time,
+        size: resp.scan_size as f64,
+        request_body: Some(req.query.sql),
+    };
+    let num_fn = req.query.query_fn.is_some() as u16;
+    report_usage_stats(
+        req_stats,
+        &org_id,
+        StreamType::Logs,
+        UsageType::SearchTopNValues,
+        num_fn,
+    )
+    .await;
+
     Ok(HttpResponse::Ok().json(resp))
 }