language-servers/deno
1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2025-09-26 12:19:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix(upgrade): fallback to Content-Length header for progress bar (#24923)

Browse source 
If the "size hint" does not provide a value, fall back to using a
"Content-Length" header.
This commit is contained in:
Bartek Iwańczuk 2024-08-07 10:16:27 +01:00 • committed by GitHub
parent d1b2a9822d
commit fe64cbd88b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
cli/http_util.rs
View file

@ -23,6 +23,7 @@ use http::header::HeaderName;
use http::header::HeaderValue; use http::header::HeaderValue;
use http::header::ACCEPT; use http::header::ACCEPT;
use http::header::AUTHORIZATION; use http::header::AUTHORIZATION;
use http::header::CONTENT_LENGTH;
use http::header::IF_NONE_MATCH; use http::header::IF_NONE_MATCH;
use http::header::LOCATION; use http::header::LOCATION;
use http::StatusCode; use http::StatusCode;
@ -579,7 +580,15 @@ async fn get_response_body_with_progress(
) -> Result<Vec<u8>, AnyError> { ) -> Result<Vec<u8>, AnyError> {
use http_body::Body as _; use http_body::Body as _;
if let Some(progress_guard) = progress_guard { if let Some(progress_guard) = progress_guard {
if let Some(total_size) = response.body().size_hint().exact() { let mut total_size = response.body().size_hint().exact();
if total_size.is_none() {
total_size = response
.headers()
.get(CONTENT_LENGTH)
.and_then(|val| val.to_str().ok())
.and_then(|s| s.parse::<u64>().ok());
}
if let Some(total_size) = total_size {
progress_guard.set_total_size(total_size); progress_guard.set_total_size(total_size);
let mut current_size = 0; let mut current_size = 0;
let mut data = Vec::with_capacity(total_size as usize); let mut data = Vec::with_capacity(total_size as usize);