feat(database): 支持多数据库后端连接

- 尝试添加对 MySQL 和 PostgreSQL 数据库的支持
- 使用条件编译特性启用不同数据库驱动
- 更新依赖配置以支持可选的数据库功能
- 修改连接管理逻辑适配多种数据库类型
- 调整插入操作实现以兼容各数据库语法差异
- 移除旧的单数据库限制并增强扩展性

Author: DataEraserC

Cargo.lock

@@ -9,19 +9,8 @@ panic = "abort"
 strip = true
 
 [workspace.dependencies]
-# NOTE: 可选数据库后端（todo）
-#pq-sys = { version = "0.7", features = ["bundled"] }
-#libsqlite3-sys = { version = "0.35", features = ["bundled"] }
-#mysqlclient-sys = { version = "0.4", features = ["bundled"] }
-
-#openssl-sys = { version = "0.9", features = ["vendored"] }
-libsqlite3-sys = { version = "0.35", features = ["bundled"] }
 diesel = { version = "2.2", features = [
-    "sqlite",
-#    "mysql",
-#    "postgres",
     "chrono",
-    "returning_clauses_for_sqlite_3_35",
     "r2d2",
 ] }
 diesel_migrations = "2.2"

Cargo.toml

@@ -20,4 +20,9 @@ path = "src/main.rs"
 assert_cmd = "2.0.14"
 
 [target.'cfg(not(target_env = "musl"))'.dependencies]
-libsqlite3-sys = { workspace = true }
+
+[features]
+default = ["sqlite"]
+sqlite = ["file_classification_core/sqlite"]
+mysql = ["file_classification_core/mysql"]
+postgres = ["file_classification_core/postgres"]

file_classification_cli/Cargo.toml

@@ -6,15 +6,17 @@ edition = "2024"
 [dependencies]
 diesel = { workspace = true }
 diesel_migrations = { workspace = true }
-libsqlite3-sys = { workspace = true }
+libsqlite3-sys = { version = "0.35", features = ["bundled"], optional = true }
+mysqlclient-sys = { version = "0.4", features = ["bundled"], optional = true }
+pq-sys = { version = "0.7", features = ["bundled"], optional = true }
+openssl-sys = { version = "0.9", features = ["vendored"], optional = true }
 chrono = { workspace = true }
 serde = { workspace = true }
 utoipa = { workspace = true, optional = true }
-#mysqlclient-sys = { workspace = true }
-#libsqlite3-sys = { workspace = true }
-#openssl-sys = { workspace = true }
-#pq-sys = { workspace = true }
 
 [features]
-default = []
+default = ["sqlite"]
+sqlite = ["libsqlite3-sys", "diesel/sqlite", "diesel/returning_clauses_for_sqlite_3_35"]
+mysql = ["mysqlclient-sys", "diesel/mysql"]
+postgres = ["pq-sys", "openssl-sys", "diesel/postgres"]
 swagger = ["utoipa"]

file_classification_core/Cargo.toml

@@ -25,12 +25,13 @@ pub fn insert_file(
 ) -> Result<i32, diesel::result::Error> {
     // 使用 match 表达式根据连接类型选择实现方式
     match conn {
+        #[cfg(feature = "sqlite")]
         // 对于 SQLite 连接，使用 returning 子句
         AnyConnection::Sqlite(_) => {
             diesel::insert_into(files::table).values(new_file).returning(files::id).get_result(conn)
         }
-        // 对于 MySQL 连接，使用事务方式（暂时注释掉，因为目前没有启用mysql，注意不要删除以下注释内容，将来会用到）
-        /*
+        #[cfg(feature = "mysql")]
+        // 对于 MySQL 连接，使用事务方式
         AnyConnection::Mysql(_) => {
                 conn.transaction(|conn| {
                         // 执行插入操作
@@ -45,7 +46,6 @@ pub fn insert_file(
                         Ok(last_id)
                 })
         },
-        */
         // 默认情况（如其他数据库类型）使用 returning 子句
         _ => diesel::insert_into(files::table).values(new_file).returning(files::id).get_result(conn),
     }

file_classification_core/src/internal/files.rs

@@ -26,12 +26,13 @@ pub fn insert_group(
 ) -> Result<i32, diesel::result::Error> {
     // 使用 match 表达式根据连接类型选择实现方式
     match conn {
+        #[cfg(feature = "sqlite")]
         // 对于 SQLite 连接，使用 returning 子句
         AnyConnection::Sqlite(_) => {
             diesel::insert_into(groups::table).values(new_group).returning(groups::id).get_result(conn)
         }
-        // 对于 MySQL 连接，使用事务方式（暂时注释掉，因为目前没有启用mysql，注意不要删除以下注释内容，将来会用到）
-        /*
+        #[cfg(feature = "mysql")]
+        // 对于 MySQL 连接，使用事务方式
         AnyConnection::Mysql(_) => {
                 conn.transaction(|conn| {
                         // 执行插入操作
@@ -46,7 +47,6 @@ pub fn insert_group(
                         Ok(last_id)
                 })
         },
-        */
         // 默认情况（如其他数据库类型）使用 returning 子句
         _ => {
             diesel::insert_into(groups::table).values(new_group).returning(groups::id).get_result(conn)

file_classification_core/src/internal/groups.rs

@@ -25,12 +25,13 @@ pub fn insert_tag(
 ) -> Result<i32, diesel::result::Error> {
     // 使用 match 表达式根据连接类型选择实现方式
     match conn {
+        #[cfg(feature = "sqlite")]
         // 对于 SQLite 连接，使用 returning 子句
         AnyConnection::Sqlite(_) => {
             diesel::insert_into(tags::table).values(new_tag).returning(tags::id).get_result(conn)
         }
+        #[cfg(feature = "mysql")]
         // 对于 MySQL 连接，使用事务方式（暂时注释掉，因为目前没有启用mysql，注意不要删除以下注释内容，将来会用到）
-        /*
         AnyConnection::Mysql(_) => {
                 conn.transaction(|conn| {
                         // 执行插入操作
@@ -45,7 +46,6 @@ pub fn insert_tag(
                         Ok(last_id)
                 })
         },
-        */
         // 默认情况（如其他数据库类型）使用 returning 子句
         _ => diesel::insert_into(tags::table).values(new_tag).returning(tags::id).get_result(conn),
     }

file_classification_core/src/internal/tags.rs

@@ -1,19 +1,26 @@
 // database.rs
 //! 数据库连接管理模块
 //!
-//! 提供数据库连接的建立和管理功能，支持多种数据库类型（当前仅支持 SQLite），
+//! 提供数据库连接的建立和管理功能，支持多种数据库类型，
 //! 并封装了通用的数据库连接类型。
 
 pub use diesel::{Connection, QueryResult};
 use diesel::RunQueryDsl;
+
 /// 通用数据库连接枚举
 ///
 /// 使用 `diesel::MultiConnection` 宏定义的枚举类型，支持多种数据库连接类型。
-/// 当前仅支持 SQLite 连接，但预留了扩展其他数据库类型的接口。
 #[derive(diesel::MultiConnection)]
 pub enum AnyConnection {
-    // Postgresql(diesel::PgConnection),
-    // Mysql(diesel::MysqlConnection),
+    #[cfg(feature = "postgres")]
+    /// PostgreSQL 数据库连接
+    Postgresql(diesel::PgConnection),
+    
+    #[cfg(feature = "mysql")]
+    /// MySQL 数据库连接
+    Mysql(diesel::MysqlConnection),
+    
+    #[cfg(feature = "sqlite")]
     /// SQLite 数据库连接
     Sqlite(diesel::SqliteConnection),
 }
@@ -24,13 +31,14 @@ pub enum AnyConnection {
 ///
 /// 参数：
 /// - `database_url`: 数据库连接字符串
-/// - `database_type`: 数据库类型（当前仅支持 "sqlite"）
+/// - `database_type`: 数据库类型
 ///
 /// 返回值：
 /// 成功时返回封装好的数据库连接对象，失败时会 panic 并输出错误信息
 pub fn establish_connection(database_url: &str, database_type: &str) -> AnyConnection {
     // 根据数据库类型建立相应的连接
     match database_type {
+        #[cfg(feature = "sqlite")]
         "sqlite" => {
             let mut conn = AnyConnection::Sqlite(
                 diesel::SqliteConnection::establish(database_url)
@@ -44,8 +52,25 @@ pub fn establish_connection(database_url: &str, database_type: &str) -> AnyConne
 
             conn
         },
+        
+        #[cfg(feature = "mysql")]
+        "mysql" => {
+            AnyConnection::Mysql(
+                diesel::MysqlConnection::establish(database_url)
+                    .unwrap_or_else(|_| panic!("Error connecting to {}", database_url)),
+            )
+        },
+        
+        #[cfg(feature = "postgres")]
+        "postgres" => {
+            AnyConnection::Postgresql(
+                diesel::PgConnection::establish(database_url)
+                    .unwrap_or_else(|_| panic!("Error connecting to {}", database_url)),
+            )
+        },
+        
         // 不支持的数据库类型直接 panic
-        _ => panic!("Unsupported database type: {}", database_type),
+        _ => panic!("Unsupported database type: {} or feature not enabled", database_type),
     }
 }
 

file_classification_core/src/utils/database.rs

@@ -26,6 +26,12 @@ mime_guess = "2.0.5"
 fern = "0.6"
 dotenvy = "0.15"
 
+[features]
+default = ["sqlite"]
+sqlite = ["file_classification_core/sqlite", "diesel/sqlite", "diesel/returning_clauses_for_sqlite_3_35"]
+mysql = ["file_classification_core/mysql", "diesel/mysql"]
+postgres = ["file_classification_core/postgres", "diesel/postgres"]
+
 [[bin]]
 name = "file_classification_webapi"
 doc = false