rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-25 17:10:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

[proxy] Make NUM_BYTES_PROXIED_COUNTER more precise

Browse Source
This commit is contained in:
Dmitry Ivanov
2022-01-20 02:17:53 +03:00
parent 924d8d489a
commit 127df96635
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
proxy/src/proxy.rs
View File

@@ -360,8 +360,8 @@ fn proxy(
// `std::io::copy` is guaranteed to exit if we return an error,
// so we can afford to lose `res` in case `flush` fails
let res = self.0.write(buf);
if res.is_ok() {
NUM_BYTES_PROXIED_COUNTER.inc_by(buf.len() as u64);
if let Ok(count) = res {
NUM_BYTES_PROXIED_COUNTER.inc_by(count as u64);
self.flush()?;
}
res