feat(database): 禁用 SQLite 外键约束以提升兼容性

- 在建立 SQLite 数据库连接后执行 PRAGMA foreign_keys = OFF

Author: DataEraserC

file_classification_core/src/utils/database.rs

@@ -5,7 +5,7 @@
 //! 并封装了通用的数据库连接类型。
 
 pub use diesel::{Connection, QueryResult};
-
+use diesel::RunQueryDsl;
 /// 通用数据库连接枚举
 ///
 /// 使用 `diesel::MultiConnection` 宏定义的枚举类型，支持多种数据库连接类型。
@@ -31,10 +31,19 @@ pub enum AnyConnection {
 pub fn establish_connection(database_url: &str, database_type: &str) -> AnyConnection {
     // 根据数据库类型建立相应的连接
     match database_type {
-        "sqlite" => AnyConnection::Sqlite(
-            diesel::SqliteConnection::establish(database_url)
-                .unwrap_or_else(|_| panic!("Error connecting to {}", database_url)),
-        ),
+        "sqlite" => {
+            let mut conn = AnyConnection::Sqlite(
+                diesel::SqliteConnection::establish(database_url)
+                    .unwrap_or_else(|_| panic!("Error connecting to {}", database_url)),
+            );
+            
+            // 关闭外键约束检查
+            diesel::sql_query("PRAGMA foreign_keys = OFF")
+                .execute(&mut conn)
+                .expect("Error executing PRAGMA foreign_keys = OFF");
+                
+            conn
+        },
         // 不支持的数据库类型直接 panic
         _ => panic!("Unsupported database type: {}", database_type),
     }

file_classification_webapi/src/bin/utils/database.rs

@@ -2,12 +2,32 @@ pub use diesel::r2d2::{ConnectionManager, Pool, PooledConnection};
 use file_classification_common::env_loader::load_env_file;
 use file_classification_core::utils::database::AnyConnection;
 use std::env;
+use diesel::RunQueryDsl;
 
 // 定义连接池类型
 pub type DbPool = Pool<ConnectionManager<AnyConnection>>;
 // 定义池化连接类型
 pub type DbPooledConnection = PooledConnection<ConnectionManager<AnyConnection>>;
 
+/// 自定义连接创建器，用于在创建连接后执行 PRAGMA 命令
+#[derive(Debug)]
+struct ConnectionCreator;
+
+impl diesel::r2d2::CustomizeConnection<AnyConnection, diesel::r2d2::Error> for ConnectionCreator {
+    fn on_acquire(&self, conn: &mut AnyConnection) -> Result<(), diesel::r2d2::Error> {
+        // 只有 SQLite 连接才这样关闭外键约束
+        match conn {
+            AnyConnection::Sqlite(_) => {
+                // 关闭外键约束检查
+                diesel::sql_query("PRAGMA foreign_keys = OFF")
+                    .execute(conn)
+                    .map_err(diesel::r2d2::Error::QueryError)?;
+            }
+        }
+        Ok(())
+    }
+}
+
 pub fn establish_connection_pool() -> DbPool {
     // 加载环境变量文件
     if let Err(e) = load_env_file() {
@@ -23,5 +43,8 @@ pub fn establish_connection_pool() -> DbPool {
         _ => panic!("Unsupported database type: {}", database_type),
     };
 
-    Pool::builder().build(manager).expect("Failed to create pool.")
+    Pool::builder()
+        .connection_customizer(Box::new(ConnectionCreator))
+        .build(manager)
+        .expect("Failed to create pool.")
 }