diff --git a/src/db/mod.rs b/src/db/mod.rs index 605bf50..3165e30 100644 --- a/src/db/mod.rs +++ b/src/db/mod.rs @@ -2,6 +2,7 @@ pub mod config; #[cfg(feature = "db_sqlite")] pub mod sqlite; pub mod traits; +#[cfg(feature = "server")] pub mod user; pub use config::check_db_config; @@ -11,6 +12,7 @@ pub use config::PixivDownloaderSqliteConfig; #[cfg(feature = "db_sqlite")] pub use sqlite::{PixivDownloaderSqlite, SqliteError}; pub use traits::PixivDownloaderDb; +#[cfg(feature = "server")] pub use user::User; pub type PixivDownloaderDbError = anyhow::Error; diff --git a/src/db/sqlite/db.rs b/src/db/sqlite/db.rs index 3fbcc0d..130154c 100644 --- a/src/db/sqlite/db.rs +++ b/src/db/sqlite/db.rs @@ -1,6 +1,7 @@ +#[cfg(feature = "server")] +use super::super::User; use super::super::{ - PixivDownloaderDb, PixivDownloaderDbConfig, PixivDownloaderDbError, - PixivDownloaderSqliteConfig, User, + PixivDownloaderDb, PixivDownloaderDbConfig, PixivDownloaderDbError, PixivDownloaderSqliteConfig, }; use super::SqliteError; use bytes::BytesMut; @@ -171,6 +172,7 @@ impl PixivDownloaderSqlite { Ok(tables) } + #[cfg(feature = "server")] async fn _get_user(&self, id: u64) -> Result, SqliteError> { let con = self.db.lock().await; Ok(con @@ -239,6 +241,7 @@ impl PixivDownloaderDb for PixivDownloaderSqlite { } } + #[cfg(feature = "server")] async fn get_user(&self, id: u64) -> Result, PixivDownloaderDbError> { Ok(self._get_user(id).await?) } diff --git a/src/db/traits.rs b/src/db/traits.rs index 379f4fd..598813c 100644 --- a/src/db/traits.rs +++ b/src/db/traits.rs @@ -1,5 +1,6 @@ use super::PixivDownloaderDbConfig; use super::PixivDownloaderDbError; +#[cfg(feature = "server")] use super::User; #[async_trait] @@ -11,6 +12,7 @@ pub trait PixivDownloaderDb { ) -> Result where Self: Sized + Send + Sync; + #[cfg(feature = "server")] /// Get a user by ID /// * `id`: The user's ID async fn get_user(&self, id: u64) -> Result, PixivDownloaderDbError>;