YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-02 18:55:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

benchmark byte conversion as well; bittorrent_udp: use buffers

Browse source
This commit is contained in:
Joakim Frostegård 2020-04-07 19:49:29 +02:00
parent 51590a3d6c
commit a99688a5a2
11 changed files with 231 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

16
aquatic/src/lib/network.rs
View file

@ -1,6 +1,6 @@
use std::sync::atomic::Ordering;
use std::net::SocketAddr;
use std::io::ErrorKind;
use std::io::{Cursor, ErrorKind};
use mio::{Events, Poll, Interest, Token};
use mio::net::UdpSocket;
@ -115,7 +115,7 @@ fn handle_readable_socket(
let mut responses_sent: usize = 0;
loop {
match socket.recv_from(buffer) {
match socket.recv_from(&mut buffer[..]) {
Ok((amt, src)) => {
let request = request_from_bytes(
&buffer[..amt],
@ -179,10 +179,16 @@ fn handle_readable_socket(
scrape_requests.drain(..),
);
for (response, src) in responses.drain(..) {
let bytes = response_to_bytes(&response, IpVersion::IPv4);
let mut cursor = Cursor::new(buffer);
match socket.send_to(&bytes[..], src){
for (response, src) in responses.drain(..) {
cursor.set_position(0);
response_to_bytes(&mut cursor, response, IpVersion::IPv4);
let amt = cursor.position() as usize;
match socket.send_to(&cursor.get_ref()[..amt], src){
Ok(_bytes_sent) => {
responses_sent += 1;
},