[Breaking Change] feat: Implement multiple sending of events (#479)

* implement sending events

* mod exmples

* change example file names

Author: ibuibu

agent/src/controllers/public/send_event.rs

@@ -17,26 +17,32 @@ pub struct MessageContainer {
     occurred_at: u64,
 }
 
-pub async fn handler(Json(json): Json<MessageContainer>) -> Result<StatusCode, AgentErrorCode> {
-    if json.key.is_empty() {
-        Err(AgentErrorCode::SendEventNoKey)?
-    }
-    if json.detail.is_empty() {
-        Err(AgentErrorCode::SendEventNoDetail)?
-    }
+pub async fn handler(
+    Json(json): Json<Vec<MessageContainer>>,
+) -> Result<StatusCode, AgentErrorCode> {
+    let events = json
+        .iter()
+        .map(|m| {
+            if m.key.is_empty() {
+                return Err(AgentErrorCode::SendEventNoKey);
+            }
+            if m.detail.is_empty() {
+                return Err(AgentErrorCode::SendEventNoDetail);
+            }
 
-    let occurred_at =
-        milliseconds_to_time(json.occurred_at).ok_or(AgentErrorCode::SendEventInvalidOccurredAt)?;
+            let occurred_at = milliseconds_to_time(m.occurred_at)
+                .ok_or(AgentErrorCode::SendEventInvalidOccurredAt)?;
 
-    let usecase = EventUsecase::new();
-    match usecase
-        .save(EventStoreRequest {
-            key: json.key,
-            detail: json.detail,
-            occurred_at,
+            Ok(EventStoreRequest {
+                key: m.key.clone(),
+                detail: m.detail.clone(),
+                occurred_at,
+            })
         })
-        .await
-    {
+        .collect::<Result<Vec<EventStoreRequest>, AgentErrorCode>>()?;
+
+    let usecase = EventUsecase::new();
+    match usecase.save(events).await {
         Ok(_) => {
             log::info!("save event");
             Ok(StatusCode::NO_CONTENT)

agent/src/repository/event_repository.rs

@@ -10,5 +10,5 @@ pub struct EventStoreRequest {
 
 #[trait_variant::make(Send)]
 pub trait EventStoreRepository {
-    async fn save(&self, request: EventStoreRequest) -> anyhow::Result<()>;
+    async fn save(&self, request: Vec<EventStoreRequest>) -> anyhow::Result<()>;
 }

agent/src/services/studio.rs

@@ -464,7 +464,7 @@ impl MetricStoreRepository for Studio {
 }
 
 impl EventStoreRepository for Studio {
-    async fn save(&self, request: EventStoreRequest) -> anyhow::Result<()> {
+    async fn save(&self, request: Vec<EventStoreRequest>) -> anyhow::Result<()> {
         self.relay_to_studio("/v1/events", request).await
     }
 }

agent/src/usecase/event_usecase.rs

@@ -19,7 +19,7 @@ impl EventUsecase<Studio> {
 }
 
 impl<R: EventStoreRepository> EventUsecase<R> {
-    pub async fn save(&self, request: EventStoreRequest) -> anyhow::Result<()> {
+    pub async fn save(&self, request: Vec<EventStoreRequest>) -> anyhow::Result<()> {
         match self.repository.save(request).await {
             Ok(_) => {
                 log::info!("save event");
@@ -43,7 +43,7 @@ mod tests {
     pub struct MockEventStoreRepository {}
 
     impl EventStoreRepository for MockEventStoreRepository {
-        async fn save(&self, _: EventStoreRequest) -> anyhow::Result<()> {
+        async fn save(&self, _: Vec<EventStoreRequest>) -> anyhow::Result<()> {
             Ok(())
         }
     }
@@ -54,11 +54,11 @@ mod tests {
             repository: MockEventStoreRepository {},
         };
         let _ = usecase
-            .save(EventStoreRequest {
+            .save(vec![EventStoreRequest {
                 key: "test".to_string(),
                 detail: "test".to_string(),
                 occurred_at: chrono::Utc::now(),
-            })
+            }])
             .await
             .map_err(|e| panic!("{:?}", e));
     }

examples/nodejs/src/generateCustomMetric.js renamed to examples/nodejs/src/generateCustomMetrics.js

@@ -1,11 +1,11 @@
 import { post } from "./lib/sock.js";
 
 (async () => {
-  const payload = {
-    key: "test-key",
-    detail: "test-detail",
-    occurred_at: Date.now(),
-  };
+  const payload = Array.from({ length: 10 }, (_, i) => ({
+    key: `test-key${i}`,
+    detail: `test-detail${i}`,
+    occurred_at: Date.now() + i,
+  }));
   const response = await post("/events", payload);
 
   console.log("The response is as follows.\n");

examples/nodejs/src/generateEvent.js renamed to examples/nodejs/src/generateEvents.js

@@ -6,13 +6,18 @@
 else:
     from sock import post
 
+events = [
+    {
+        "key": "test-key" + str(x + 10),
+        "detail": "test-detail" + str(x),
+        "occurred_at": int(time.time() * 1000 + x),
+    }
+    for x in range(10)
+]
+
 json_response = post(
     "/events",
-    {
-        "key": "test-key",
-        "detail": "test-detail",
-        "occurred_at": int(time.time() * 1000),
-    },
+    events,
 )
 
 print("The response is as follows.\n")