rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-17 02:12:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Limit number of buffered responses

Browse Source
This commit is contained in:
Konstantin Knizhnik
2022-10-29 22:49:57 +03:00
parent ab7a4d8574
commit ad0c5410d7
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pageserver/src/page_service.rs
View File

@@ -275,6 +275,7 @@ impl PageServerHandler {
pgb.flush().await?;
let metrics = PageRequestMetrics::new(&tenant_id, &timeline_id);
let mut n_buffered_responses = 0usize;
loop {
let msg = tokio::select! {
@@ -312,7 +313,7 @@ impl PageServerHandler {
}
PagestreamFeMessage::GetPage(req) => {
let _timer = metrics.get_page_at_lsn.start_timer();
do_flush = !req.prefetch;
do_flush = !req.prefetch || n_buffered_responses > MAX_BUFFERED_RESPONSES;
self.handle_get_page_at_lsn_request(&timeline, &req).await
}
PagestreamFeMessage::DbSize(req) => {
@@ -331,8 +332,10 @@ impl PageServerHandler {
});
pgb.write_message(&BeMessage::CopyData(&response.serialize()))?;
n_buffered_responses += 1;
if do_flush {
pgb.flush().await?;
n_buffered_responses = 0;
}
}
Ok(())