diff --git a/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java b/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java index 280dca1..8a9c959 100644 --- a/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java +++ b/src/main/java/com/gxwebsoft/ai/entity/AiCloudDoc.java @@ -27,6 +27,9 @@ public class AiCloudDoc implements Serializable { @Schema(description = "云目录ID") private String categoryId; + @Schema(description = "项目ID") + private Integer projectId; + @Schema(description = "单位ID") private Integer companyId; diff --git a/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml b/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml index be5a5a5..08d04a1 100644 --- a/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml +++ b/src/main/java/com/gxwebsoft/ai/mapper/xml/AiCloudDocMapper.xml @@ -13,6 +13,9 @@ AND a.category_id = #{param.categoryId} + + AND a.project_id = #{param.projectId} + AND a.company_id = #{param.companyId} diff --git a/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java b/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java index f09aade..c5802c2 100644 --- a/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java +++ b/src/main/java/com/gxwebsoft/ai/param/AiCloudDocParam.java @@ -27,6 +27,10 @@ public class AiCloudDocParam extends BaseParam { @Schema(description = "云目录ID") @QueryField(type = QueryType.EQ) private String categoryId; + + @Schema(description = "项目ID") + @QueryField(type = QueryType.EQ) + private Integer projectId = 0; @Schema(description = "单位ID") @QueryField(type = QueryType.EQ)