♻️ Extract serve_file fn
_
parent
722eac97e8
commit
ec5b770ae3
|
|
@ -50,76 +50,14 @@ fn parse_range_header (range_str: &str) -> (Option <u64>, Option <u64>) {
|
||||||
(start, end)
|
(start, end)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
async fn serve_file (
|
||||||
async fn main () -> Result <(), Box <dyn Error>> {
|
client: Arc <Client>,
|
||||||
let client = Arc::new (Client::new ());
|
mut f: File,
|
||||||
|
req_id: String,
|
||||||
let mut backoff_delay = 0;
|
should_send_body: bool,
|
||||||
|
range_start: Option <u64>,
|
||||||
loop {
|
range_end: Option <u64>
|
||||||
if backoff_delay > 0 {
|
) {
|
||||||
delay_for (Duration::from_millis (backoff_delay)).await;
|
|
||||||
}
|
|
||||||
|
|
||||||
let _uri = Uri::builder ()
|
|
||||||
.scheme ("http")
|
|
||||||
.authority ("127.0.0.1:4000")
|
|
||||||
.path_and_query ("/http_listen/alien_wildlands")
|
|
||||||
.build ().unwrap ();
|
|
||||||
|
|
||||||
let req_req = client.get ("http://127.0.0.1:4000/http_listen/alien_wildlands");
|
|
||||||
|
|
||||||
//println! ("Step 1");
|
|
||||||
let req_resp = match req_req.send ().await {
|
|
||||||
Err (e) => {
|
|
||||||
println! ("Err: {:?}", e);
|
|
||||||
backoff_delay = backoff_delay * 2 + 500;
|
|
||||||
continue;
|
|
||||||
},
|
|
||||||
Ok (r) => {
|
|
||||||
backoff_delay = 0;
|
|
||||||
r
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
if req_resp.status () != StatusCode::OK {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
println! ("Step 3");
|
|
||||||
|
|
||||||
let body = req_resp.bytes ().await?;
|
|
||||||
let parts: http_serde::RequestParts = match rmp_serde::from_read_ref (&body)
|
|
||||||
{
|
|
||||||
Ok (x) => x,
|
|
||||||
_ => continue,
|
|
||||||
};
|
|
||||||
|
|
||||||
let (req_id, uri) = (parts.id, parts.uri);
|
|
||||||
|
|
||||||
println! ("Client requested {}", uri);
|
|
||||||
let mut range_start = None;
|
|
||||||
let mut range_end = None;
|
|
||||||
|
|
||||||
for (k, v) in parts.headers.iter () {
|
|
||||||
let v = std::str::from_utf8 (v).unwrap ();
|
|
||||||
//println! ("{}: {}", k, v);
|
|
||||||
|
|
||||||
if k == "range" {
|
|
||||||
let (start, end) = parse_range_header (v);
|
|
||||||
range_start = start;
|
|
||||||
range_end = end;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let should_send_body = match &parts.method {
|
|
||||||
http_serde::Method::Get => true,
|
|
||||||
_ => false,
|
|
||||||
};
|
|
||||||
|
|
||||||
//println! ("Step 6");
|
|
||||||
let client = client.clone ();
|
|
||||||
tokio::spawn (async move {
|
|
||||||
let (tx, rx) = channel (2);
|
let (tx, rx) = channel (2);
|
||||||
let body = if should_send_body {
|
let body = if should_send_body {
|
||||||
Some (Body::wrap_stream (rx))
|
Some (Body::wrap_stream (rx))
|
||||||
|
|
@ -128,10 +66,6 @@ async fn main () -> Result <(), Box <dyn Error>> {
|
||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut path = PathBuf::from ("/home/user");
|
|
||||||
path.push (&uri [1..]);
|
|
||||||
let mut f = File::open (path).await.unwrap ();
|
|
||||||
|
|
||||||
let file_md = f.metadata ().await.unwrap ();
|
let file_md = f.metadata ().await.unwrap ();
|
||||||
let file_len = file_md.len ();
|
let file_len = file_md.len ();
|
||||||
|
|
||||||
|
|
@ -213,6 +147,90 @@ async fn main () -> Result <(), Box <dyn Error>> {
|
||||||
if let Err (e) = resp_req.send ().await {
|
if let Err (e) = resp_req.send ().await {
|
||||||
println! ("Err: {:?}", e);
|
println! ("Err: {:?}", e);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::main]
|
||||||
|
async fn main () -> Result <(), Box <dyn Error>> {
|
||||||
|
let client = Arc::new (Client::new ());
|
||||||
|
|
||||||
|
let mut backoff_delay = 0;
|
||||||
|
|
||||||
|
loop {
|
||||||
|
if backoff_delay > 0 {
|
||||||
|
delay_for (Duration::from_millis (backoff_delay)).await;
|
||||||
|
}
|
||||||
|
|
||||||
|
let _uri = Uri::builder ()
|
||||||
|
.scheme ("http")
|
||||||
|
.authority ("127.0.0.1:4000")
|
||||||
|
.path_and_query ("/http_listen/alien_wildlands")
|
||||||
|
.build ().unwrap ();
|
||||||
|
|
||||||
|
let req_req = client.get ("http://127.0.0.1:4000/http_listen/alien_wildlands");
|
||||||
|
|
||||||
|
//println! ("Step 1");
|
||||||
|
let req_resp = match req_req.send ().await {
|
||||||
|
Err (e) => {
|
||||||
|
println! ("Err: {:?}", e);
|
||||||
|
backoff_delay = backoff_delay * 2 + 500;
|
||||||
|
continue;
|
||||||
|
},
|
||||||
|
Ok (r) => {
|
||||||
|
backoff_delay = 0;
|
||||||
|
r
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
if req_resp.status () != StatusCode::OK {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
println! ("Step 3");
|
||||||
|
|
||||||
|
let body = req_resp.bytes ().await?;
|
||||||
|
let parts: http_serde::RequestParts = match rmp_serde::from_read_ref (&body)
|
||||||
|
{
|
||||||
|
Ok (x) => x,
|
||||||
|
_ => continue,
|
||||||
|
};
|
||||||
|
|
||||||
|
let (req_id, uri) = (parts.id, parts.uri);
|
||||||
|
|
||||||
|
println! ("Client requested {}", uri);
|
||||||
|
let mut range_start = None;
|
||||||
|
let mut range_end = None;
|
||||||
|
|
||||||
|
for (k, v) in parts.headers.iter () {
|
||||||
|
let v = std::str::from_utf8 (v).unwrap ();
|
||||||
|
//println! ("{}: {}", k, v);
|
||||||
|
|
||||||
|
if k == "range" {
|
||||||
|
let (start, end) = parse_range_header (v);
|
||||||
|
range_start = start;
|
||||||
|
range_end = end;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let should_send_body = match &parts.method {
|
||||||
|
http_serde::Method::Get => true,
|
||||||
|
_ => false,
|
||||||
|
};
|
||||||
|
|
||||||
|
//println! ("Step 6");
|
||||||
|
let client = client.clone ();
|
||||||
|
tokio::spawn (async move {
|
||||||
|
let mut path = PathBuf::from ("/home/user");
|
||||||
|
path.push (&uri [1..]);
|
||||||
|
let f = File::open (path).await.unwrap ();
|
||||||
|
|
||||||
|
serve_file (
|
||||||
|
client,
|
||||||
|
f,
|
||||||
|
req_id,
|
||||||
|
should_send_body,
|
||||||
|
range_start,
|
||||||
|
range_end
|
||||||
|
).await;
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue