fix: preserve walk-dir internode metrics fallback (#5095)

Co-authored-by: heihutu <heihutu@gmail.com>
This commit is contained in:
houseme
2026-07-21 23:30:34 +08:00
committed by GitHub
parent 937b311316
commit f6e8ce4639
3 changed files with 59 additions and 2 deletions
+29
View File
@@ -1496,6 +1496,7 @@ mod tests {
use axum::{Router, body::Body, extract::State, http::StatusCode, response::IntoResponse, routing::get};
use futures::stream::{self, StreamExt as _};
use http_body_util::BodyExt as _;
use rustfs_io_metrics::internode_metrics::global_internode_metrics;
use std::io::{self, IoSlice};
use std::sync::{
Arc,
@@ -1741,6 +1742,7 @@ mod tests {
let addr = listener.local_addr().expect("listener local address should be available");
let app = Router::new()
.route("/stream", get(get_stream).head(reject_head).put(accept_put))
.route(WALK_DIR_PATH, get(get_stream))
.route("/reject-put", get(get_stream).put(reject_put))
.route("/stall", get(get_stalling_stream))
.route("/delayed-first", get(get_delayed_first_chunk))
@@ -1801,6 +1803,33 @@ mod tests {
handle.abort();
}
#[tokio::test]
async fn http_reader_records_walk_dir_recv_bytes() {
let state = TestState::default();
let Some((base_url, handle)) = start_test_server(state.clone()).await else {
return;
};
let metrics = global_internode_metrics();
let before = metrics.snapshot().recv_bytes_total;
let url = base_url.replace("/stream", WALK_DIR_PATH);
let mut reader = HttpReader::new(url, Method::GET, HeaderMap::new(), None)
.await
.expect("walk_dir reader should open");
let mut buf = Vec::new();
reader.read_to_end(&mut buf).await.expect("walk_dir body should read to EOF");
let after = metrics.snapshot().recv_bytes_total;
assert_eq!(buf, b"hello");
assert_eq!(state.get_count.load(Ordering::SeqCst), 1);
assert!(
after >= before.saturating_add(5),
"walk_dir HttpReader should record streamed bytes as internode recv bytes: before={before}, after={after}"
);
handle.abort();
}
#[tokio::test]
async fn http_reader_stall_timeout_triggers_after_progress_stops() {
let state = TestState::default();