feat: add select by foreign key

Author: DataEraserC

file_classification_core/src/internal/files.rs

@@ -287,4 +287,17 @@ pub fn decrease_files_reference_count_by_conditions(
 
     // 减少引用计数
     query.set(files::reference_count.eq(files::reference_count - 1)).execute(conn)
-}
+}
+
+pub fn select_file_by_group_id(
+    conn: &mut AnyConnection,
+    other_group_id: i64,
+) -> Result<Vec<File>, diesel::result::Error> {
+    use crate::model::schema::file_groups;
+
+    files::table
+        .inner_join(file_groups::table.on(files::id.eq(file_groups::file_id)))
+        .filter(file_groups::group_id.eq(other_group_id as i32))
+        .select(File::as_select())
+        .load(conn)
+}

file_classification_core/src/internal/groups.rs

@@ -356,3 +356,29 @@ pub fn decrease_groups_reference_count_by_conditions(
     // 减少引用计数
     query.set(groups::reference_count.eq(groups::reference_count - 1)).execute(conn)
 }
+
+pub fn select_group_by_file_id(
+    conn: &mut AnyConnection,
+    other_file_id: i32,
+) -> Result<Vec<Group>, diesel::result::Error> {
+    use crate::model::schema::file_groups;
+    
+    groups::table
+        .inner_join(file_groups::table.on(groups::id.eq(file_groups::group_id)))
+        .filter(file_groups::file_id.eq(other_file_id))
+        .select(Group::as_select())
+        .load(conn)
+}
+
+pub fn select_group_by_tag_id(
+    conn: &mut AnyConnection,
+    tag_id: i32,
+) -> Result<Vec<Group>, diesel::result::Error> {
+    use crate::model::schema::group_tags;
+    
+    groups::table
+        .inner_join(group_tags::table.on(groups::id.eq(group_tags::group_id)))
+        .filter(group_tags::tag_id.eq(tag_id))
+        .select(Group::as_select())
+        .load(conn)
+}

file_classification_core/src/internal/tags.rs

@@ -79,7 +79,7 @@ pub fn delete_tag(conn: &mut AnyConnection, tag_id: i32) -> Result<usize, diesel
 }
 
 use crate::model::schema::tags::dsl::*;
-use crate::model::schema::tags;
+use crate::model::schema::{groups, tags};
 use std::fmt::{Debug, Formatter, Result as fmtResult};
 
 impl Debug for Tag {
@@ -93,9 +93,10 @@ impl Debug for Tag {
 }
 
 use super::models::TagCondition;
-use crate::model::models::{OrderDirection, TagOrderBy, TagQueryOptions, UpdateTagDTO};
+use crate::model::models::{Group, OrderDirection, TagOrderBy, TagQueryOptions, UpdateTagDTO};
 use diesel::dsl::not;
 use diesel::sql_types::Bool;
+use crate::model::schema::group_tags::dsl::group_tags;
 
 // 将 TagCondition 转换为 diesel 查询条件的辅助函数
 fn build_tag_condition(condition: TagCondition) -> Box<dyn BoxableExpression<tags::table, <AnyConnection as Connection>::Backend, SqlType=diesel::sql_types::Bool>> {
@@ -280,3 +281,16 @@ pub fn decrease_tags_reference_count_by_conditions(
     // 减少引用计数
     query.set(tags::reference_count.eq(tags::reference_count - 1)).execute(conn)
 }
+
+pub fn select_tag_by_group_id(
+    conn: &mut AnyConnection,
+    group_id: i64,
+) -> Result<Vec<Tag>, diesel::result::Error> {
+    use crate::model::schema::group_tags;
+
+    tags::table
+        .inner_join(group_tags::table.on(tags::id.eq(group_tags::tag_id)))
+        .filter(group_tags::group_id.eq(group_id as i32))
+        .select(Tag::as_select())
+        .load(conn)
+}

file_classification_core/src/service/files.rs

@@ -242,3 +242,9 @@ pub fn delete_files_by_conditions(
     })
 }
 
+pub fn select_file_by_group_id(
+    conn: &mut AnyConnection,
+    other_group_id: i64,
+) -> Result<Vec<File>, diesel::result::Error> {
+    internal::files::select_file_by_group_id(conn, other_group_id)
+}

file_classification_core/src/service/groups.rs

@@ -2,7 +2,7 @@ use crate::model::models::{FileCondition, FileGroupCondition, GroupCondition, Gr
 use crate::service::AppError;
 use crate::{internal::groups, model::models::{CreateGroupDTO, Group, GroupFilter}};
 use diesel::result::Error;
-use diesel::Connection;
+use diesel::{Connection, JoinOnDsl};
 use crate::utils::database::AnyConnection;
 
 pub fn create_group(conn: &mut AnyConnection, name: &str) -> Result<usize, Error> {
@@ -184,3 +184,16 @@ pub fn delete_groups_by_conditions(
     })
 }
 
+pub fn select_group_by_file_id(
+    conn: &mut AnyConnection,
+    other_file_id: i32,
+) -> Result<Vec<Group>, diesel::result::Error> {
+    crate::internal::groups::select_group_by_file_id(conn, other_file_id)
+}
+
+pub fn select_group_by_tag_id(
+    conn: &mut AnyConnection,
+    tag_id: i32,
+) -> Result<Vec<Group>, diesel::result::Error> {
+    crate::internal::groups::select_group_by_tag_id(conn, tag_id)
+}

file_classification_core/src/service/tags.rs

@@ -3,7 +3,7 @@ use crate::{
     internal::tags,
     model::models::{CreateTagDTO, Tag, TagFilter},
 };
-use diesel::Connection;
+use diesel::{Connection, JoinOnDsl};
 use crate::utils::database::AnyConnection;
 
 pub fn create_tag(conn: &mut AnyConnection, name: &str) -> Result<Tag, diesel::result::Error> {
@@ -107,4 +107,11 @@ pub fn delete_tags_by_conditions(
 
         Ok(total_deleted)
     })
+}
+
+pub fn select_tag_by_group_id(
+    conn: &mut AnyConnection,
+    group_id: i64,
+) -> Result<Vec<Tag>, diesel::result::Error> {
+    crate::internal::tags::select_tag_by_group_id(conn, group_id)
 }