feat: add registered and deleted metrics (#1622)

This commit is contained in:
Ellie Huxtable 2024-01-23 20:01:20 +00:00 committed by GitHub
parent 067bda66e2
commit bdc533d2bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -11,6 +11,7 @@ use axum::{
Json, Json,
}; };
use http::StatusCode; use http::StatusCode;
use metrics::counter;
use rand::rngs::OsRng; use rand::rngs::OsRng;
use tracing::{debug, error, info, instrument}; use tracing::{debug, error, info, instrument};
use uuid::Uuid; use uuid::Uuid;
@ -142,6 +143,8 @@ pub async fn register<DB: Database>(
.await; .await;
} }
counter!("atuin_users_registered", 1);
match db.add_session(&new_session).await { match db.add_session(&new_session).await {
Ok(_) => Ok(Json(RegisterResponse { session: token })), Ok(_) => Ok(Json(RegisterResponse { session: token })),
Err(e) => { Err(e) => {
@ -166,6 +169,9 @@ pub async fn delete<DB: Database>(
return Err(ErrorResponse::reply("failed to delete user") return Err(ErrorResponse::reply("failed to delete user")
.with_status(StatusCode::INTERNAL_SERVER_ERROR)); .with_status(StatusCode::INTERNAL_SERVER_ERROR));
}; };
counter!("atuin_users_deleted", 1);
Ok(Json(DeleteUserResponse {})) Ok(Json(DeleteUserResponse {}))
} }