Spaces:
Running
Running
Upload main.rs
Browse files- 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 |
|