mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 09:20:22 +08:00
fix(ci): apply rustfmt to usage metadata handlers
This commit is contained in:
@@ -5,12 +5,13 @@ use crate::handlers::admin::request::{AdminAppState, AdminRequestContext};
|
|||||||
use crate::handlers::admin::shared::query_param_value;
|
use crate::handlers::admin::shared::query_param_value;
|
||||||
use crate::GatewayError;
|
use crate::GatewayError;
|
||||||
use aether_admin::observability::usage::{
|
use aether_admin::observability::usage::{
|
||||||
admin_usage_bad_request_response, admin_usage_client_family, admin_usage_data_unavailable_response,
|
admin_usage_bad_request_response, admin_usage_client_family,
|
||||||
admin_usage_has_fallback, admin_usage_is_failed, admin_usage_matches_search,
|
admin_usage_data_unavailable_response, admin_usage_has_fallback, admin_usage_is_failed,
|
||||||
admin_usage_matches_username, admin_usage_parse_ids, admin_usage_parse_limit,
|
admin_usage_matches_search, admin_usage_matches_username, admin_usage_parse_ids,
|
||||||
admin_usage_parse_offset, admin_usage_provider_key_name, admin_usage_record_json,
|
admin_usage_parse_limit, admin_usage_parse_offset, admin_usage_provider_key_name,
|
||||||
build_admin_usage_active_requests_response, build_admin_usage_records_response,
|
admin_usage_record_json, build_admin_usage_active_requests_response,
|
||||||
build_admin_usage_summary_stats_response_from_summary, ADMIN_USAGE_DATA_UNAVAILABLE_DETAIL,
|
build_admin_usage_records_response, build_admin_usage_summary_stats_response_from_summary,
|
||||||
|
ADMIN_USAGE_DATA_UNAVAILABLE_DETAIL,
|
||||||
};
|
};
|
||||||
use aether_data::repository::users::StoredUserSummary;
|
use aether_data::repository::users::StoredUserSummary;
|
||||||
use aether_data_contracts::repository::{
|
use aether_data_contracts::repository::{
|
||||||
@@ -649,7 +650,9 @@ pub(super) async fn maybe_build_local_admin_usage_summary_response(
|
|||||||
let active_client_family_filter = client_family_filter
|
let active_client_family_filter = client_family_filter
|
||||||
.as_deref()
|
.as_deref()
|
||||||
.filter(|value| !value.trim().is_empty());
|
.filter(|value| !value.trim().is_empty());
|
||||||
let (usage, total) = if attempt_status_filter.is_some() || active_client_family_filter.is_some() {
|
let (usage, total) = if attempt_status_filter.is_some()
|
||||||
|
|| active_client_family_filter.is_some()
|
||||||
|
{
|
||||||
let mut usage = state.list_usage_audits(&base_query).await?;
|
let mut usage = state.list_usage_audits(&base_query).await?;
|
||||||
let user_ids: Vec<String> = usage
|
let user_ids: Vec<String> = usage
|
||||||
.iter()
|
.iter()
|
||||||
@@ -679,12 +682,14 @@ pub(super) async fn maybe_build_local_admin_usage_summary_response(
|
|||||||
active_username_filter,
|
active_username_filter,
|
||||||
&users_by_id,
|
&users_by_id,
|
||||||
state.has_auth_user_data_reader(),
|
state.has_auth_user_data_reader(),
|
||||||
) && attempt_status_filter.is_none_or(|attempt_status| admin_usage_matches_attempt_status(
|
) && attempt_status_filter.is_none_or(|attempt_status| {
|
||||||
item,
|
admin_usage_matches_attempt_status(
|
||||||
attempt_status,
|
item,
|
||||||
&attempt_flags_by_usage_id,
|
attempt_status,
|
||||||
request_candidate_reader_available,
|
&attempt_flags_by_usage_id,
|
||||||
)) && admin_usage_matches_client_family(item, active_client_family_filter)
|
request_candidate_reader_available,
|
||||||
|
)
|
||||||
|
}) && admin_usage_matches_client_family(item, active_client_family_filter)
|
||||||
});
|
});
|
||||||
sort_usage_newest_first(&mut usage);
|
sort_usage_newest_first(&mut usage);
|
||||||
let total = usage.len();
|
let total = usage.len();
|
||||||
|
|||||||
@@ -344,7 +344,11 @@ fn users_me_usage_client_family(item: &StoredRequestUsageAudit) -> Option<&str>
|
|||||||
.and_then(serde_json::Value::as_object)
|
.and_then(serde_json::Value::as_object)
|
||||||
.and_then(|affinity| affinity.get("client_family"))
|
.and_then(|affinity| affinity.get("client_family"))
|
||||||
.and_then(serde_json::Value::as_str)
|
.and_then(serde_json::Value::as_str)
|
||||||
.or_else(|| metadata.get("client_family").and_then(serde_json::Value::as_str))
|
.or_else(|| {
|
||||||
|
metadata
|
||||||
|
.get("client_family")
|
||||||
|
.and_then(serde_json::Value::as_str)
|
||||||
|
})
|
||||||
})
|
})
|
||||||
.map(str::trim)
|
.map(str::trim)
|
||||||
.filter(|value| !value.is_empty())
|
.filter(|value| !value.is_empty())
|
||||||
|
|||||||
@@ -1166,9 +1166,21 @@ pub fn admin_usage_record_json(
|
|||||||
);
|
);
|
||||||
object.insert("client_is_stream".to_string(), json!(client_is_stream));
|
object.insert("client_is_stream".to_string(), json!(client_is_stream));
|
||||||
maybe_insert_string_field(object, "client_family", admin_usage_client_family(item));
|
maybe_insert_string_field(object, "client_family", admin_usage_client_family(item));
|
||||||
maybe_insert_string_field(object, "client_ip", admin_usage_metadata_string(item, "client_ip"));
|
maybe_insert_string_field(
|
||||||
maybe_insert_string_field(object, "user_agent", admin_usage_metadata_string(item, "user_agent"));
|
object,
|
||||||
maybe_insert_string_field(object, "request_path", admin_usage_metadata_string(item, "request_path"));
|
"client_ip",
|
||||||
|
admin_usage_metadata_string(item, "client_ip"),
|
||||||
|
);
|
||||||
|
maybe_insert_string_field(
|
||||||
|
object,
|
||||||
|
"user_agent",
|
||||||
|
admin_usage_metadata_string(item, "user_agent"),
|
||||||
|
);
|
||||||
|
maybe_insert_string_field(
|
||||||
|
object,
|
||||||
|
"request_path",
|
||||||
|
admin_usage_metadata_string(item, "request_path"),
|
||||||
|
);
|
||||||
maybe_insert_string_field(
|
maybe_insert_string_field(
|
||||||
object,
|
object,
|
||||||
"request_path_and_query",
|
"request_path_and_query",
|
||||||
|
|||||||
Reference in New Issue
Block a user