45 lines
1.2 KiB
Rust
45 lines
1.2 KiB
Rust
use crate::{error::Error, jobs::JobState};
|
|
use background_jobs::ActixJob;
|
|
use std::{future::Future, pin::Pin};
|
|
use uuid::Uuid;
|
|
|
|
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
|
pub(crate) struct CacheMedia {
|
|
uuid: Uuid,
|
|
}
|
|
|
|
impl CacheMedia {
|
|
pub(crate) fn new(uuid: Uuid) -> Self {
|
|
CacheMedia { uuid }
|
|
}
|
|
|
|
#[tracing::instrument(name = "Cache media", skip(state))]
|
|
async fn perform(self, state: JobState) -> Result<(), Error> {
|
|
if !state.media.is_outdated(self.uuid).await? {
|
|
return Ok(());
|
|
}
|
|
|
|
if let Some(url) = state.media.get_url(self.uuid).await? {
|
|
let (content_type, bytes) = state.requests.fetch_bytes(url.as_str()).await?;
|
|
|
|
state
|
|
.media
|
|
.store_bytes(self.uuid, content_type, bytes)
|
|
.await?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl ActixJob for CacheMedia {
|
|
type State = JobState;
|
|
type Future = Pin<Box<dyn Future<Output = Result<(), anyhow::Error>>>>;
|
|
|
|
const NAME: &'static str = "relay::jobs::CacheMedia";
|
|
|
|
fn run(self, state: Self::State) -> Self::Future {
|
|
Box::pin(async move { self.perform(state).await.map_err(Into::into) })
|
|
}
|
|
}
|