feat(database): 切换默认数据库为PostgreSQL并优化连接配置

- 修改默认数据库配置为PostgreSQL
- 在MySQL和PostgreSQL连接中添加外键约束检查关闭逻辑
- 为WebAPI模块添加数据库连接初始化逻辑
- 支持MySQL和PostgreSQL的外键约束动态关闭功能

Author: DataEraserC

.file_classification_env

@@ -6,8 +6,14 @@
 #DATABASE_URL=file_classification.db
 #DATABASE_TYPE=sqlite
 
-DATABASE_URL=mysql://root:@localhost:3306/file_classification
-DATABASE_TYPE=mysql
+#DATABASE_URL=postgres:mypassword//postgres:@localhost:5432/file_classification
+#DATABASE_TYPE=postgres
+
+#DATABASE_URL=mysql://root:@localhost:3306/file_classification
+#DATABASE_TYPE=mysql
+
+DATABASE_URL=postgres://postgres:@localhost:5432/file_classification
+DATABASE_TYPE=postgres
 
 # Web API 配置
 BIND_ADDRESS=127.0.0.1

file_classification_core/src/utils/database.rs

@@ -55,18 +55,32 @@ pub fn establish_connection(database_url: &str, database_type: &str) -> AnyConne
 
         #[cfg(feature = "mysql")]
         "mysql" => {
-            AnyConnection::Mysql(
+            let mut conn = AnyConnection::Mysql(
                 diesel::MysqlConnection::establish(database_url)
                     .unwrap_or_else(|_| panic!("Error connecting to {}", database_url)),
-            )
+            );
+            
+            // 关闭外键约束检查
+            diesel::sql_query("SET FOREIGN_KEY_CHECKS = 0")
+                .execute(&mut conn)
+                .expect("Error executing SET FOREIGN_KEY_CHECKS = 0");
+                
+            conn
         },
 
         #[cfg(feature = "postgres")]
         "postgres" => {
-            AnyConnection::Postgresql(
+            let mut conn = AnyConnection::Postgresql(
                 diesel::PgConnection::establish(database_url)
                     .unwrap_or_else(|_| panic!("Error connecting to {}", database_url)),
-            )
+            );
+            
+            // 关闭外键约束检查
+            diesel::sql_query("SET session_replication_role = 'replica'")
+                .execute(&mut conn)
+                .expect("Error executing SET session_replication_role = 'replica'");
+                
+            conn
         },
 
         // 不支持的数据库类型直接 panic

file_classification_webapi/src/bin/utils/database.rs

@@ -24,6 +24,20 @@ impl diesel::r2d2::CustomizeConnection<AnyConnection, diesel::r2d2::Error> for C
                     .execute(conn)
                     .map_err(diesel::r2d2::Error::QueryError)?;
             }
+            #[cfg(feature = "mysql")]
+            // MySQL 连接关闭外键约束检查
+            AnyConnection::Mysql(_) => {
+                diesel::sql_query("SET FOREIGN_KEY_CHECKS = 0")
+                    .execute(conn)
+                    .map_err(diesel::r2d2::Error::QueryError)?;
+            }
+            #[cfg(feature = "postgres")]
+            // PostgreSQL 连接关闭外键约束检查
+            AnyConnection::Postgresql(_) => {
+                diesel::sql_query("SET session_replication_role = 'replica'")
+                    .execute(conn)
+                    .map_err(diesel::r2d2::Error::QueryError)?;
+            }
             _ => {}
         }
         Ok(())