use crate::{AppContext, ImageData, ImageId, SharedUrl, Task};use collections::HashMap;use futures::{future::Shared, AsyncReadExt, FutureExt, TryFutureExt};use image::ImageError;use parking_lot::Mutex;use std::sync::Arc;use thiserror::Error;use util::http::{self, HttpClient}; #[derive(PartialEq, Eq, Hash, Clone)]pub(crate) struct RenderImageParams { pub(crate) image_id: ImageId,} #[derive(Debug, Error, Clone)]pub(crate) enum Error { #[error("http error: {0}")] Client(#[from] http::Error), #[error("IO error: {0}")] Io(Arc<std::io::Error>), #[error("unexpected http status: {status}, body: {body}")] BadStatus { status: http::StatusCode, body: String, }, #[error("image error: {0}")] Image(Arc<ImageError>),} impl From<std::io::Error> for Error { fn from(error: std::io::Error) -> Self { Error::Io(Arc::new(error)) }} impl From<ImageError> for Error { fn from(error: ImageError) -> Self { Error::Image(Arc::new(error)) }} pub(crate) struct ImageCache { client: Arc<dyn HttpClient>, images: Arc<Mutex<HashMap<SharedUrl, FetchImageTask>>>,} type FetchImageTask = Shared<Task<Result<Arc<ImageData>, Error>>>; impl ImageCache { pub fn new(client: Arc<dyn HttpClient>) -> Self { ImageCache { client, images: Default::default(), } } pub fn get(&self, uri: impl Into<SharedUrl>, cx: &AppContext) -> FetchImageTask { let uri = uri.into(); let mut images = self.images.lock(); match images.get(&uri) { Some(future) => future.clone(), None => { let client = self.client.clone(); let future = cx .background_executor() .spawn( { let uri = uri.clone(); async move { let mut response = client.get(uri.as_ref(), ().into(), true).await?; let mut body = Vec::new(); response.body_mut().read_to_end(&mut body).await?; if !response.status().is_success() { return Err(Error::BadStatus { status: response.status(), body: String::from_utf8_lossy(&body).into_owned(), }); } let format = image::guess_format(&body)?; let image = image::load_from_memory_with_format(&body, format)? .into_bgra8(); Ok(Arc::new(ImageData::new(image))) } } .map_err({ let uri = uri.clone(); move |error| { log::log!(log::Level::Error, "{:?} {:?}", &uri, &error); error } }), ) .shared(); images.insert(uri, future.clone()); future } } }}
use crate::{AppContext, ImageData, ImageId, SharedUrl, Task};
use collections::HashMap;
use futures::{future::Shared, AsyncReadExt, FutureExt, TryFutureExt};
use image::ImageError;
use parking_lot::Mutex;
use std::sync::Arc;
use thiserror::Error;
use util::http::{self, HttpClient};
#[derive(PartialEq, Eq, Hash, Clone)]
pub(crate) struct RenderImageParams {
pub(crate) image_id: ImageId,
}
#[derive(Debug, Error, Clone)]
pub(crate) enum Error {
#[error("http error: {0}")]
Client(#[from] http::Error),
#[error("IO error: {0}")]
Io(Arc<std::io::Error>),
#[error("unexpected http status: {status}, body: {body}")]
BadStatus {
status: http::StatusCode,
body: String,
},
#[error("image error: {0}")]
Image(Arc<ImageError>),
}
impl From<std::io::Error> for Error {
fn from(error: std::io::Error) -> Self {
Error::Io(Arc::new(error))
}
}
impl From<ImageError> for Error {
fn from(error: ImageError) -> Self {
Error::Image(Arc::new(error))
}
}
pub(crate) struct ImageCache {
client: Arc<dyn HttpClient>,
images: Arc<Mutex<HashMap<SharedUrl, FetchImageTask>>>,
}
type FetchImageTask = Shared<Task<Result<Arc<ImageData>, Error>>>;
impl ImageCache {
pub fn new(client: Arc<dyn HttpClient>) -> Self {
ImageCache {
client,
images: Default::default(),
}
}
pub fn get(&self, uri: impl Into<SharedUrl>, cx: &AppContext) -> FetchImageTask {
let uri = uri.into();
let mut images = self.images.lock();
match images.get(&uri) {
Some(future) => future.clone(),
None => {
let client = self.client.clone();
let future = cx
.background_executor()
.spawn(
{
let uri = uri.clone();
async move {
let mut response =
client.get(uri.as_ref(), ().into(), true).await?;
let mut body = Vec::new();
response.body_mut().read_to_end(&mut body).await?;
if !response.status().is_success() {
return Err(Error::BadStatus {
status: response.status(),
body: String::from_utf8_lossy(&body).into_owned(),
});
}
let format = image::guess_format(&body)?;
let image = image::load_from_memory_with_format(&body, format)?
.into_bgra8();
Ok(Arc::new(ImageData::new(image)))
}
}
.map_err({
let uri = uri.clone();
move |error| {
log::log!(log::Level::Error, "{:?} {:?}", &uri, &error);
error
}
}),
)
.shared();
images.insert(uri, future.clone());
future
}
}
}
}