use async_trait::async_trait;
use anyhow::Result;
use serde::{Deserialize, Serialize};
use crate::domain::entity::FileLock;
use crate::infrastructure::persistence::{FileLockRepository, PaginationParams, PaginatedResult, FileLockFilter};
use super::QueryHandler;
#[deprecated(since = "0.2.0", note = "Use QueryHandler<Q> trait instead")]
#[async_trait]
pub trait FileLockQuery: Send + Sync {
type Output;
async fn execute(&self) -> Result<Self::Output>;
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GetFileLockByIdQuery {
pub id: String,
}
pub struct GetFileLockByIdHandler<R: FileLockRepository> {
repository: std::sync::Arc<R>,
}
impl<R: FileLockRepository> GetFileLockByIdHandler<R> {
pub fn new(repository: std::sync::Arc<R>) -> Self {
Self { repository }
}
}
#[async_trait]
impl<R: FileLockRepository + 'static> QueryHandler<GetFileLockByIdQuery> for GetFileLockByIdHandler<R> {
type Output = Option<FileLock>;
async fn handle(&self, query: GetFileLockByIdQuery) -> Result<Self::Output> {
self.repository.find_by_id(&query.id).await
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ListFileLockQuery {
pub page: u32,
pub per_page: u32,
pub filter_file_id: Option<Uuid>,
pub filter_user_id: Option<Uuid>,
pub filter_status: Option<LockStatus>,
}
impl Default for ListFileLockQuery {
fn default() -> Self {
Self {
page: 1,
per_page: 20,
filter_file_id: None,
filter_user_id: None,
filter_status: None,
}
}
}
pub struct ListFileLockHandler<R: FileLockRepository> {
repository: std::sync::Arc<R>,
}
impl<R: FileLockRepository> ListFileLockHandler<R> {
pub fn new(repository: std::sync::Arc<R>) -> Self {
Self { repository }
}
}
#[async_trait]
impl<R: FileLockRepository + 'static> QueryHandler<ListFileLockQuery> for ListFileLockHandler<R> {
type Output = PaginatedResult<FileLock>;
async fn handle(&self, query: ListFileLockQuery) -> Result<Self::Output> {
let params = PaginationParams::new(query.page, query.per_page);
let filters = FileLockFilter {
file_id: query.filter_file_id.clone(),
user_id: query.filter_user_id.clone(),
status: query.filter_status.clone(),
..Default::default()
};
if filters.has_filters() {
self.repository.list_with_filters(params, filters).await
} else {
self.repository.list(params).await
}
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GetFileLockByFileIdQuery {
pub file_id: Uuid,
}
pub struct GetFileLockByFileIdHandler<R: FileLockRepository> {
repository: std::sync::Arc<R>,
}
impl<R: FileLockRepository> GetFileLockByFileIdHandler<R> {
pub fn new(repository: std::sync::Arc<R>) -> Self {
Self { repository }
}
}
#[async_trait]
impl<R: FileLockRepository + 'static> QueryHandler<GetFileLockByFileIdQuery> for GetFileLockByFileIdHandler<R> {
type Output = Option<FileLock>;
async fn handle(&self, query: GetFileLockByFileIdQuery) -> Result<Self::Output> {
self.repository.find_by_file_id(query.file_id).await
}
}