Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Load images in the background

Browse source
This commit is contained in:
Antonio Scandurra 2024-01-24 13:15:31 +01:00
parent e3157f7212
commit 92b0184036
2 changed files with 37 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

2
crates/gpui/src/elements/img.rs
View file

@ -104,7 +104,7 @@ impl Element for Img {
cx.with_z_index(1, |cx| { cx.with_z_index(1, |cx| {
match source { match source {
ImageSource::Uri(uri) => { ImageSource::Uri(uri) => {
let image_future = cx.image_cache.get(uri.clone()); let image_future = cx.image_cache.get(uri.clone(), cx);
if let Some(data) = image_future if let Some(data) = image_future
.clone() .clone()
.now_or_never() .now_or_never()

33
crates/gpui/src/image_cache.rs
View file

@ -1,9 +1,6 @@
use crate::{ImageData, ImageId, SharedUrl}; use crate::{AppContext, ImageData, ImageId, SharedUrl, Task};
use collections::HashMap; use collections::HashMap;
use futures::{ use futures::{future::Shared, AsyncReadExt, FutureExt, TryFutureExt};
future::{BoxFuture, Shared},
AsyncReadExt, FutureExt, TryFutureExt,
};
use image::ImageError; use image::ImageError;
use parking_lot::Mutex; use parking_lot::Mutex;
use std::sync::Arc; use std::sync::Arc;
@ -44,10 +41,10 @@ impl From<ImageError> for Error {
pub(crate) struct ImageCache { pub(crate) struct ImageCache {
client: Arc<dyn HttpClient>, client: Arc<dyn HttpClient>,
images: Arc<Mutex<HashMap<SharedUrl, FetchImageFuture>>>, images: Arc<Mutex<HashMap<SharedUrl, FetchImageTask>>>,
} }
type FetchImageFuture = Shared<BoxFuture<'static, Result<Arc<ImageData>, Error>>>; type FetchImageTask = Shared<Task<Result<Arc<ImageData>, Error>>>;
impl ImageCache { impl ImageCache {
pub fn new(client: Arc<dyn HttpClient>) -> Self { pub fn new(client: Arc<dyn HttpClient>) -> Self {
@ -57,10 +54,7 @@ impl ImageCache {
} }
} }
pub fn get( pub fn get(&self, uri: impl Into<SharedUrl>, cx: &AppContext) -> FetchImageTask {
&self,
uri: impl Into<SharedUrl>,
) -> Shared<BoxFuture<'static, Result<Arc<ImageData>, Error>>> {
let uri = uri.into(); let uri = uri.into();
let mut images = self.images.lock(); let mut images = self.images.lock();
@ -68,10 +62,14 @@ impl ImageCache {
Some(future) => future.clone(), Some(future) => future.clone(),
None => { None => {
let client = self.client.clone(); let client = self.client.clone();
let future = { let future = cx
.background_executor()
.spawn(
{
let uri = uri.clone(); let uri = uri.clone();
async move { async move {
let mut response = client.get(uri.as_ref(), ().into(), true).await?; let mut response =
client.get(uri.as_ref(), ().into(), true).await?;
let mut body = Vec::new(); let mut body = Vec::new();
response.body_mut().read_to_end(&mut body).await?; response.body_mut().read_to_end(&mut body).await?;
@ -83,20 +81,19 @@ impl ImageCache {
} }
let format = image::guess_format(&body)?; let format = image::guess_format(&body)?;
let image = let image = image::load_from_memory_with_format(&body, format)?
image::load_from_memory_with_format(&body, format)?.into_bgra8(); .into_bgra8();
Ok(Arc::new(ImageData::new(image))) Ok(Arc::new(ImageData::new(image)))
} }
} }
.map_err({ .map_err({
let uri = uri.clone(); let uri = uri.clone();
move |error| { move |error| {
log::log!(log::Level::Error, "{:?} {:?}", &uri, &error); log::log!(log::Level::Error, "{:?} {:?}", &uri, &error);
error error
} }
}) }),
.boxed() )
.shared(); .shared();
images.insert(uri, future.clone()); images.insert(uri, future.clone());