YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-01 10:15:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic http load test: track and display bandwidth

Browse source
This commit is contained in:
Joakim Frostegård 2020-07-20 18:02:16 +02:00
parent 45940a05a9
commit 1c6f22db85
3 changed files with 27 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

17
aquatic_http_load_test/src/network.rs
View file

@ -78,6 +78,9 @@ impl Connection {
}
};
state.statistics.bytes_received
.fetch_add(self.bytes_read, Ordering::SeqCst);
let res_response = Response::from_bytes(
&self.read_buffer[..self.bytes_read]
);
@ -123,7 +126,7 @@ impl Connection {
rng
);
match self.send_request_inner(&request.as_bytes()){
match self.send_request_inner(state, &request.as_bytes()){
Ok(_) => {
state.statistics.requests.fetch_add(1, Ordering::SeqCst);
},
@ -135,8 +138,16 @@ impl Connection {
self.can_send_initial = false;
}
fn send_request_inner(&mut self, request: &[u8]) -> ::std::io::Result<()> {
self.stream.write(request)?;
fn send_request_inner(
&mut self,
state: &LoadTestState,
request: &[u8]
) -> ::std::io::Result<()> {
let bytes_sent = self.stream.write(request)?;
state.statistics.bytes_sent
.fetch_add(bytes_sent, Ordering::SeqCst);
self.stream.flush()?;
Ok(())