From bdc533d2bc615df614e24fb4055bf68349e6cf88 Mon Sep 17 00:00:00 2001
From: Ellie Huxtable <ellie@elliehuxtable.com>
Date: Tue, 23 Jan 2024 20:01:20 +0000
Subject: feat: add registered and deleted metrics (#1622)

---
 atuin-server/src/handlers/user.rs | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'atuin-server')

diff --git a/atuin-server/src/handlers/user.rs b/atuin-server/src/handlers/user.rs
index 4f2ad891f..535c740f1 100644
--- a/atuin-server/src/handlers/user.rs
+++ b/atuin-server/src/handlers/user.rs
@@ -11,6 +11,7 @@ use axum::{
     Json,
 };
 use http::StatusCode;
+use metrics::counter;
 use rand::rngs::OsRng;
 use tracing::{debug, error, info, instrument};
 use uuid::Uuid;
@@ -142,6 +143,8 @@ pub async fn register<DB: Database>(
         .await;
     }
 
+    counter!("atuin_users_registered", 1);
+
     match db.add_session(&new_session).await {
         Ok(_) => Ok(Json(RegisterResponse { session: token })),
         Err(e) => {
@@ -166,6 +169,9 @@ pub async fn delete<DB: Database>(
         return Err(ErrorResponse::reply("failed to delete user")
             .with_status(StatusCode::INTERNAL_SERVER_ERROR));
     };
+
+    counter!("atuin_users_deleted", 1);
+
     Ok(Json(DeleteUserResponse {}))
 }
 
-- 
cgit v1.2.3