feat: add some metrics related to Atuin as an app (#1399)

This commit is contained in:
Ellie Huxtable 2023-11-18 18:07:23 +00:00 committed by GitHub
parent 15d214e237
commit 7575a83fa7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 0 deletions

View File

@ -6,6 +6,7 @@ use axum::{
Json,
};
use http::StatusCode;
use metrics::counter;
use time::{Month, UtcOffset};
use tracing::{debug, error, instrument};
@ -65,6 +66,8 @@ pub async fn list<DB: Database>(
if req.sync_ts.unix_timestamp_nanos() < 0 || req.history_ts.unix_timestamp_nanos() < 0 {
error!("client asked for history from < epoch 0");
counter!("atuin_history_epoch_before_zero", 1);
return Err(
ErrorResponse::reply("asked for history from before epoch 0")
.with_status(StatusCode::BAD_REQUEST),
@ -93,6 +96,8 @@ pub async fn list<DB: Database>(
user.id
);
counter!("atuin_history_returned", history.len() as u64);
Ok(Json(SyncHistoryResponse { history }))
}
@ -127,6 +132,7 @@ pub async fn add<DB: Database>(
let State(AppState { database, settings }) = state;
debug!("request to add {} history items", req.len());
counter!("atuin_history_uploaded", req.len() as u64);
let mut history: Vec<NewHistory> = req
.into_iter()
@ -146,6 +152,8 @@ pub async fn add<DB: Database>(
// Don't return an error here. We want to insert as much of the
// history list as we can, so log the error and continue going.
if !keep {
counter!("atuin_history_too_long", 1);
tracing::warn!(
"history too long, got length {}, max {}",
h.data.len(),

View File

@ -1,5 +1,6 @@
use axum::{extract::Query, extract::State, Json};
use http::StatusCode;
use metrics::counter;
use serde::Deserialize;
use tracing::{error, instrument};
@ -23,11 +24,15 @@ pub async fn post<DB: Database>(
"request to add records"
);
counter!("atuin_record_uploaded", records.len() as u64);
let too_big = records
.iter()
.any(|r| r.data.data.len() >= settings.max_record_size || settings.max_record_size == 0);
if too_big {
counter!("atuin_record_too_large", 1);
return Err(
ErrorResponse::reply("could not add records; record too large")
.with_status(StatusCode::BAD_REQUEST),