snapshot

Author: DataEraserC

Cargo.lock

@@ -127,3 +127,5 @@ WHERE fg.file_id = 'specified_file_id';
 4. **用户权限管理**：如果系统对多用户开放，需要考虑用户权限管理，确保用户只能访问和操作自己的文件。
 5. **文件版本控制**：可以增加版本控制功能，记录文件的修改历史。
 6. **文件搜索优化**：除了标签搜索，还可以实现基于文件内容的全文搜索，提高搜索的准确性。
+7. **多种数据库兼容**：由于某些数据库不支持returning子句，所以若要兼容这个需要抛弃一旦插入就能直接返回对象的想法。
+8. **支持批量操作**

README_cn.md

@@ -15,10 +15,6 @@ doc = false
 name = "list_files_by_conditions"
 doc = false
 
-[[bin]]
-name = "create_file"
-doc = false
-
 [[bin]]
 name = "delete_file"
 doc = false

file_classification_cli/Cargo.toml

@@ -1,4 +1,4 @@
-use file_classification_core::service::group_tag::delete_group_tag;
+use file_classification_core::service::group_tag::delete_group_tag_by_id;
 use file_classification_core::utils::database::establish_connection;
 use std::io::{stdin, stdout, Write};
 use file_classification_core::model::models::GroupTagDTO;
@@ -19,7 +19,7 @@ fn main() {
     stdin().read_line(&mut tag_id_input).unwrap();
     let tag_id: i32 = tag_id_input.trim().parse().expect("Invalid Tag ID");
 
-    let result = delete_group_tag(connection, GroupTagDTO{group_id, tag_id});
+    let result = delete_group_tag_by_id(connection, GroupTagDTO{group_id, tag_id});
     match result {
         Ok(deleted_group_tag_num) => {
             if deleted_group_tag_num == 0 {

file_classification_cli/src/bin/create_file.rs

@@ -1,52 +1,27 @@
-use super::{models::FileGroupDTO, AppError};
-use crate::internal::{
-    files::increase_file_reference_count,
-    groups::{find_group_by_id, increase_group_reference_count},
-};
+use super::models::{FileGroupDTO,FileGroupCondition};
 use crate::model::schema::{file_groups};
 use diesel::prelude::*;
 use std::fmt::{Debug, Formatter, Result as fmtResult};
 
-pub fn create_file_group(
+pub fn insert_file_group(
     conn: &mut SqliteConnection,
-    file_group_dto: FileGroupDTO,
-) -> Result<FileGroupDTO, AppError> {
-    let group = find_group_by_id(conn, file_group_dto.group_id)?.ok_or(AppError::GroupNotFound)?;
-
-    if group.is_primary {
-        return Err(AppError::CannotAssociateWithPrimary);
-    }
-
-    let result = conn.transaction::<_, AppError, _>(|conn| {
-        increase_file_reference_count(conn, file_group_dto.file_id)?;
-        increase_group_reference_count(conn, file_group_dto.group_id)?;
-        Ok(diesel::insert_into(file_groups::table).values(&file_group_dto).execute(conn))
-    });
-    let _ = result?;
-    Ok(file_group_dto)
+    file_group_dto: &FileGroupDTO,
+) -> Result<usize, diesel::result::Error>{
+    diesel::insert_into(file_groups::table)
+        .values(file_group_dto)
+        .execute(conn)
 }
 
-pub fn delete_file_group(
+pub fn delete_file_group_by_id(
     conn: &mut SqliteConnection,
-    file_group_dto: FileGroupDTO,
-) -> Result<usize, AppError> {
-    let result = conn.transaction::<_, AppError, _>(|conn| {
-        // 减少组和标签的引用计数
-        decrease_group_reference_count(conn, file_group_dto.group_id)?;
-        decrease_file_reference_count(conn, file_group_dto.file_id)?;
-
-        // 删除关联记录
-        let deleted_count = diesel::delete(
-            file_groups::table
-                .filter(file_groups::group_id.eq(file_group_dto.group_id))
-                .filter(file_groups::file_id.eq(file_group_dto.file_id))
-        )
-            .execute(conn)?;
-
-        Ok(deleted_count)
-    });
-
-    result.map_err(|e| e)
+    file_group_dto: &FileGroupDTO,
+) -> Result<usize, diesel::result::Error> {
+    // 删除关联记录
+    diesel::delete(
+        file_groups::table
+            .filter(file_groups::group_id.eq(file_group_dto.group_id))
+            .filter(file_groups::file_id.eq(file_group_dto.file_id))
+    ).execute(conn)
 }
 
 impl Debug for FileGroupDTO {
@@ -55,14 +30,7 @@ impl Debug for FileGroupDTO {
     }
 }
 
-// 在 files.rs 文件中添加以下代码（需要添加到文件末尾，在其他 use 语句之后）
-
-use super::models::FileGroupCondition;
-use diesel::dsl::not;
-use diesel::sql_types::Bool;
 use diesel::sqlite::Sqlite;
-use crate::internal::files::decrease_file_reference_count;
-use crate::internal::groups::decrease_group_reference_count;
 
 // 将 FileGroupCondition 转换为 diesel 查询条件的辅助函数
 fn build_file_group_condition(condition: FileGroupCondition) -> Box<dyn BoxableExpression<file_groups::table, Sqlite, SqlType = diesel::sql_types::Bool>> {
@@ -84,7 +52,7 @@ fn build_file_group_condition(condition: FileGroupCondition) -> Box<dyn BoxableE
                     Some(prev) => result = Some(Box::new(prev.and(expr))),
                 }
             }
-            result.unwrap_or_else(|| Box::new(true.into_sql::<Bool>()))
+            result.unwrap_or_else(|| Box::new(true.into_sql::<diesel::sql_types::Bool>()))
         },
         FileGroupCondition::Or(conditions) => {
             let mut result: Option<Box<dyn BoxableExpression<file_groups::table, Sqlite, SqlType = diesel::sql_types::Bool>>> = None;
@@ -95,11 +63,11 @@ fn build_file_group_condition(condition: FileGroupCondition) -> Box<dyn BoxableE
                     Some(prev) => result = Some(Box::new(prev.or(expr))),
                 }
             }
-            result.unwrap_or_else(|| Box::new(false.into_sql::<Bool>()))
+            result.unwrap_or_else(|| Box::new(false.into_sql::<diesel::sql_types::Bool>()))
         },
         FileGroupCondition::Not(condition) => {
             let expr = build_file_group_condition(*condition);
-            Box::new(not(expr))
+            Box::new(diesel::dsl::not(expr))
         }
     }
 }
@@ -123,3 +91,19 @@ pub fn select_file_groups_by_conditions(
         .select((file_groups::file_id, file_groups::group_id))
         .load(conn)
 }
+
+pub fn delete_file_groups_by_conditions(
+    conn: &mut SqliteConnection,
+    conditions: Vec<FileGroupCondition>,
+) -> Result<usize, diesel::result::Error> {
+    let mut query = diesel::delete(file_groups::table).into_boxed::<Sqlite>();
+
+    // 对每个条件应用 AND 逻辑
+    for condition in conditions {
+        let boxed_condition = build_file_group_condition(condition);
+        query = query.filter(boxed_condition);
+    }
+
+    query.execute(conn)
+}
+

file_classification_cli/src/bin/delete_group_tag.rs

@@ -1,37 +1,34 @@
-use super::{models::{CreateFileDTO, File, FileFilter, FileCondition}, AppError};
+use super::{models::{UpdateFileDTO, CreateFileDTO, File, FileFilter, FileCondition}};
 use diesel::prelude::*;
 
-pub fn create_file(conn: &mut SqliteConnection, new_file: &CreateFileDTO) -> Result<File, AppError> {
+pub fn create_file(conn: &mut SqliteConnection, new_file: &CreateFileDTO) -> Result<usize, diesel::result::Error>  {
     diesel::insert_into(files::table)
-        .values(new_file)
-        // NOTE: as_returning 只有一部分数据库支持
-        // TODO: 若不支持则需要另外处理
-        .returning(File::as_returning())
-        .get_result(conn)
-        .map_err(|e| AppError::CreateFileFailed(e.to_string()))
+        .values(new_file).execute(conn)
 }
 
+pub fn find_file_by_id(conn: &mut SqliteConnection, _id: i32) -> Result<Option<File>, diesel::result::Error> {
+    files::table
+        .filter(files::id.eq(_id))
+        .select(File::as_select())
+        .first(conn)
+        .optional()
+}
 pub fn increase_file_reference_count(
     conn: &mut SqliteConnection,
     file_id: i32,
-) -> Result<(), AppError> {
+) -> Result<usize, diesel::result::Error> {
     diesel::update(files::table.find(file_id))
         .set(files::reference_count.eq(files::reference_count + 1))
-        .execute(conn)?;
-
-    Ok(())
+        .execute(conn)
 }
 
-#[allow(dead_code)]
 pub fn decrease_file_reference_count(
     conn: &mut SqliteConnection,
     file_id: i32,
-) -> Result<(), AppError> {
+) -> Result<usize, diesel::result::Error> {
     diesel::update(files::table.find(file_id))
         .set(files::reference_count.eq(files::reference_count - 1))
-        .execute(conn)?;
-
-    Ok(())
+        .execute(conn)
 }
 
 #[deprecated]
@@ -64,52 +61,10 @@ pub fn select_files(
     base_query.load(conn)
 }
 
-// pub fn update_file(
-// 	conn: &mut SqliteConnection,
-// 	update_input: UpdateFile,
-// ) -> Result<usize, diesel::result::Error> {
-// 	// 初始化更新查询
-// 	let mut query = diesel::update(files).into_boxed();
-
-// 	// 动态设置需要更新的字段
-// 	if let Some(new_path) = update_input.set.path {
-// 		query = query.set(path.eq(new_path));
-// 	}
-// 	if let Some(new_type) = update_input.set.type_ {
-// 		query = query.set(type_.eq(new_type));
-// 	}
-// 	if let Some(new_ref_count) = update_input.set.reference_count {
-// 		query = query.set(reference_count.eq(new_ref_count));
-// 	}
-// 	if let Some(new_group) = update_input.set.group_id {
-// 		query = query.set(group_id.eq(new_group));
-// 	}
-
-// 	// 动态添加过滤条件
-// 	if let Some(file_id) = update_input.filter.id {
-// 		query = query.filter(id.eq(file_id));
-// 	}
-// 	if let Some(file_type) = update_input.filter.type_ {
-// 		query = query.filter(type_.eq(file_type));
-// 	}
-// 	if let Some(file_path) = update_input.filter.path {
-// 		query = query.filter(path.eq(file_path));
-// 	}
-// 	if let Some(ref_count) = update_input.filter.reference_count {
-// 		query = query.filter(reference_count.eq(ref_count));
-// 	}
-// 	if let Some(group) = update_input.filter.group_id {
-// 		query = query.filter(group_id.eq(group));
-// 	}
-// 	query.load(conn)
-// }
-
-pub fn delete_file(conn: &mut SqliteConnection, file_id: i32) -> Result<(), diesel::result::Error> {
-    match diesel::delete(files.filter(files::id.eq(file_id))).execute(conn) {
-        Ok(_) => Ok(()),
-        Err(e) => Err(e),
-    }
+pub fn delete_file_by_id(conn: &mut SqliteConnection, file_id: i32) -> Result<usize, diesel::result::Error> {
+    diesel::delete(files.filter(files::id.eq(file_id))).execute(conn)
 }
+
 use crate::model::schema::files;
 use crate::model::schema::files::dsl::*;
 use std::fmt::{Debug, Formatter, Result as fmtResult};
@@ -126,8 +81,6 @@ impl Debug for File {
     }
 }
 
-use diesel::dsl::not;
-use crate::model::models::UpdateFileDTO;
 
 // 将 FileCondition 转换为 diesel 查询条件的辅助函数
 // 更新 build_condition 函数以处理新增的条件类型
@@ -173,7 +126,7 @@ fn build_file_condition(condition: FileCondition) -> Box<dyn BoxableExpression<f
         },
         FileCondition::Not(condition) => {
             let expr = build_file_condition(*condition);
-            Box::new(not(expr))
+            Box::new(diesel::dsl::not(expr))
         }
     }
 }
@@ -203,7 +156,7 @@ pub fn update_files_by_conditions(
     conn: &mut SqliteConnection,
     conditions: Vec<FileCondition>,
     update_set: UpdateFileDTO,
-) -> Result<usize, AppError> {
+) -> Result<usize, diesel::result::Error> {
     let mut query = diesel::update(files::table).into_boxed::<Sqlite>();
 
     // 应用所有条件
@@ -212,7 +165,20 @@ pub fn update_files_by_conditions(
         query = query.filter(boxed_condition);
     }
 
-    let result = query.set(update_set).execute(conn)?;
-    Ok(result)
+    query.set(update_set).execute(conn)
 }
 
+pub fn delete_files_by_conditions(
+    conn: &mut SqliteConnection,
+    conditions: Vec<FileCondition>,
+) -> Result<usize, diesel::result::Error> {
+    let mut query = diesel::delete(files::table).into_boxed::<Sqlite>();
+
+    // 应用所有条件
+    for condition in conditions {
+        let boxed_condition = build_file_condition(condition);
+        query = query.filter(boxed_condition);
+    }
+
+    query.execute(conn)
+}