recycleactor commited on
Commit
5ed7694
·
verified ·
1 Parent(s): a90bd19

Upload main.rs

Browse files
Files changed (1) hide show
  1. src/main.rs +8 -1
src/main.rs CHANGED
@@ -981,7 +981,14 @@ async fn stream(
981
  let upstream_total = upstream_headers
982
  .get(reqwest::header::CONTENT_RANGE)
983
  .and_then(|v| v.to_str().ok())
984
- .and_then(parse_total_from_content_range);
 
 
 
 
 
 
 
985
 
986
  let mut upstream = upstream.bytes_stream();
987
 
 
981
  let upstream_total = upstream_headers
982
  .get(reqwest::header::CONTENT_RANGE)
983
  .and_then(|v| v.to_str().ok())
984
+ .and_then(parse_total_from_content_range)
985
+ // For full (non-range) requests, CDN returns Content-Length directly
986
+ .or_else(|| {
987
+ upstream_headers
988
+ .get(reqwest::header::CONTENT_LENGTH)
989
+ .and_then(|v| v.to_str().ok())
990
+ .and_then(|v| v.parse::<u64>().ok())
991
+ });
992
 
993
  let mut upstream = upstream.bytes_stream();
994