me/iced
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #2419 from skygrango/wasm/support-download_progress

Browse source 
Support wasm target for `download_progress`
This commit is contained in:
Héctor Ramón 2024-09-10 19:33:38 +02:00 • committed by GitHub
commit 7683447f87
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 60 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/download_progress/Cargo.toml
View file

@ -12,4 +12,4 @@ iced.features = ["tokio"]
[dependencies.reqwest]
version = "0.12"
default-features = false
features = ["rustls-tls"]
features = ["stream", "rustls-tls"]

12
examples/download_progress/index.html Normal file
View file

@ -0,0 +1,12 @@
<!DOCTYPE html>
<html lang="en" style="height: 100%">
<head>
<meta charset="utf-8" content="text/html; charset=utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1" />
<title>Download_Progress - Iced</title>
<base data-trunk-public-url />
</head>
<body style="height: 100%; margin: 0">
<link data-trunk rel="rust" href="Cargo.toml" data-wasm-opt="z" data-bin="download_progress" />
</body>
</html>

103
examples/download_progress/src/download.rs
View file

@ -1,91 +1,62 @@
use iced::futures;
use iced::futures::{SinkExt, Stream, StreamExt};
use iced::stream::try_channel;
use iced::Subscription;
use std::hash::Hash;
use std::sync::Arc;
// Just a little utility function
pub fn file<I: 'static + Hash + Copy + Send + Sync, T: ToString>(
id: I,
url: T,
) -> iced::Subscription<(I, Progress)> {
) -> iced::Subscription<(I, Result<Progress, Error>)> {
Subscription::run_with_id(
id,
futures::stream::unfold(State::Ready(url.to_string()), move |state| {
use iced::futures::FutureExt;
download(id, state).map(Some)
}),
download(url.to_string()).map(move |progress| (id, progress)),
)
}
async fn download<I: Copy>(id: I, state: State) -> ((I, Progress), State) {
match state {
State::Ready(url) => {
let response = reqwest::get(&url).await;
fn download(url: String) -> impl Stream<Item = Result<Progress, Error>> {
try_channel(1, move |mut output| async move {
let response = reqwest::get(&url).await?;
let total = response.content_length().ok_or(Error::NoContentLength)?;
match response {
Ok(response) => {
if let Some(total) = response.content_length() {
(
(id, Progress::Started),
State::Downloading {
response,
total,
downloaded: 0,
},
)
} else {
((id, Progress::Errored), State::Finished)
}
}
Err(_) => ((id, Progress::Errored), State::Finished),
}
let _ = output.send(Progress::Downloading { percent: 0.0 }).await;
let mut byte_stream = response.bytes_stream();
let mut downloaded = 0;
while let Some(next_bytes) = byte_stream.next().await {
let bytes = next_bytes?;
downloaded += bytes.len();
let _ = output
.send(Progress::Downloading {
percent: 100.0 * downloaded as f32 / total as f32,
})
.await;
}
State::Downloading {
mut response,
total,
downloaded,
} => match response.chunk().await {
Ok(Some(chunk)) => {
let downloaded = downloaded + chunk.len() as u64;
let percentage = (downloaded as f32 / total as f32) * 100.0;
let _ = output.send(Progress::Finished).await;
(
(id, Progress::Advanced(percentage)),
State::Downloading {
response,
total,
downloaded,
},
)
}
Ok(None) => ((id, Progress::Finished), State::Finished),
Err(_) => ((id, Progress::Errored), State::Finished),
},
State::Finished => {
// We do not let the stream die, as it would start a
// new download repeatedly if the user is not careful
// in case of errors.
iced::futures::future::pending().await
}
}
Ok(())
})
}
#[derive(Debug, Clone)]
pub enum Progress {
Started,
Advanced(f32),
Downloading { percent: f32 },
Finished,
Errored,
}
pub enum State {
Ready(String),
Downloading {
response: reqwest::Response,
total: u64,
downloaded: u64,
},
Finished,
#[derive(Debug, Clone)]
pub enum Error {
RequestFailed(Arc<reqwest::Error>),
NoContentLength,
}
impl From<reqwest::Error> for Error {
fn from(error: reqwest::Error) -> Self {
Error::RequestFailed(Arc::new(error))
}
}

20
examples/download_progress/src/main.rs
View file

@ -23,7 +23,7 @@ struct Example {
pub enum Message {
Add,
Download(usize),
DownloadProgressed((usize, download::Progress)),
DownloadProgressed((usize, Result<download::Progress, download::Error>)),
}
impl Example {
@ -114,19 +114,19 @@ impl Download {
}
}
pub fn progress(&mut self, new_progress: download::Progress) {
pub fn progress(
&mut self,
new_progress: Result<download::Progress, download::Error>,
) {
if let State::Downloading { progress } = &mut self.state {
match new_progress {
download::Progress::Started => {
*progress = 0.0;
Ok(download::Progress::Downloading { percent }) => {
*progress = percent;
}
download::Progress::Advanced(percentage) => {
*progress = percentage;
}
download::Progress::Finished => {
Ok(download::Progress::Finished) => {
self.state = State::Finished;
}
download::Progress::Errored => {
Err(_error) => {
self.state = State::Errored;
}
}
@ -136,7 +136,7 @@ impl Download {
pub fn subscription(&self) -> Subscription<Message> {
match self.state {
State::Downloading { .. } => {
download::file(self.id, "https://speed.hetzner.de/100MB.bin?")
download::file(self.id, "https://huggingface.co/mattshumer/Reflection-Llama-3.1-70B/resolve/main/model-00001-of-00162.safetensors")
.map(Message::DownloadProgressed)
}
_ => Subscription::none(),