Aode (lion)
6ed592c432
Some checks failed
continuous-integration/drone/push Build is failing
- Is this a good idea? it definitely will make interacting with pict-rs harder. Maybe it's best not to do this
96 lines
2.6 KiB
Rust
96 lines
2.6 KiB
Rust
use crate::error::Error;
|
|
use actix_web::web::Bytes;
|
|
use futures_util::stream::Stream;
|
|
use std::fmt::Debug;
|
|
use tokio::io::{AsyncRead, AsyncWrite};
|
|
|
|
pub(crate) mod file_store;
|
|
pub(crate) mod object_store;
|
|
|
|
pub(crate) trait Identifier: Send + Sync + Clone + Debug {
|
|
fn to_bytes(&self) -> Result<Vec<u8>, Error>;
|
|
|
|
fn from_bytes(bytes: Vec<u8>) -> Result<Self, Error>
|
|
where
|
|
Self: Sized;
|
|
}
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
pub(crate) trait Store: Send + Sync + Clone + Debug {
|
|
type Identifier: Identifier + 'static;
|
|
type Stream: Stream<Item = std::io::Result<Bytes>> + 'static;
|
|
|
|
async fn save_async_read<Reader>(&self, reader: &mut Reader) -> Result<Self::Identifier, Error>
|
|
where
|
|
Reader: AsyncRead + Unpin;
|
|
|
|
async fn save_bytes(&self, bytes: Bytes) -> Result<Self::Identifier, Error>;
|
|
|
|
async fn to_stream(
|
|
&self,
|
|
identifier: &Self::Identifier,
|
|
from_start: Option<u64>,
|
|
len: Option<u64>,
|
|
) -> Result<Self::Stream, Error>;
|
|
|
|
async fn read_into<Writer>(
|
|
&self,
|
|
identifier: &Self::Identifier,
|
|
writer: &mut Writer,
|
|
) -> Result<(), std::io::Error>
|
|
where
|
|
Writer: AsyncWrite + Send + Unpin;
|
|
|
|
async fn len(&self, identifier: &Self::Identifier) -> Result<u64, Error>;
|
|
|
|
async fn remove(&self, identifier: &Self::Identifier) -> Result<(), Error>;
|
|
}
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
impl<'a, T> Store for &'a T
|
|
where
|
|
T: Store,
|
|
{
|
|
type Identifier = T::Identifier;
|
|
type Stream = T::Stream;
|
|
|
|
async fn save_async_read<Reader>(&self, reader: &mut Reader) -> Result<Self::Identifier, Error>
|
|
where
|
|
Reader: AsyncRead + Unpin,
|
|
{
|
|
T::save_async_read(self, reader).await
|
|
}
|
|
|
|
async fn save_bytes(&self, bytes: Bytes) -> Result<Self::Identifier, Error> {
|
|
T::save_bytes(self, bytes).await
|
|
}
|
|
|
|
async fn to_stream(
|
|
&self,
|
|
identifier: &Self::Identifier,
|
|
from_start: Option<u64>,
|
|
len: Option<u64>,
|
|
) -> Result<Self::Stream, Error> {
|
|
T::to_stream(self, identifier, from_start, len).await
|
|
}
|
|
|
|
async fn read_into<Writer>(
|
|
&self,
|
|
identifier: &Self::Identifier,
|
|
writer: &mut Writer,
|
|
) -> Result<(), std::io::Error>
|
|
where
|
|
Writer: AsyncWrite + Send + Unpin,
|
|
{
|
|
T::read_into(self, identifier, writer).await
|
|
}
|
|
|
|
async fn len(&self, identifier: &Self::Identifier) -> Result<u64, Error> {
|
|
T::len(self, identifier).await
|
|
}
|
|
|
|
async fn remove(&self, identifier: &Self::Identifier) -> Result<(), Error> {
|
|
T::remove(self, identifier).await
|
|
}
|
|
}
|