From 8003a799c270d14049c569a46c7255656cc6b4b4 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Wed, 28 May 2025 16:42:54 +0800
Subject: [PATCH] =?UTF-8?q?TASK:mannequin=EF=BC=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../model/dto/QueryLibraryPageServiceDTO.java |  2 ++
 .../da/service/impl/LibraryServiceImpl.java   |  4 ++++
 .../da/service/impl/WorkspaceServiceImpl.java | 21 ++++++++++++++++++-
 3 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/ai/da/model/dto/QueryLibraryPageServiceDTO.java b/src/main/java/com/ai/da/model/dto/QueryLibraryPageServiceDTO.java
index 5396036e..8c8c829c 100644
--- a/src/main/java/com/ai/da/model/dto/QueryLibraryPageServiceDTO.java
+++ b/src/main/java/com/ai/da/model/dto/QueryLibraryPageServiceDTO.java
@@ -31,6 +31,8 @@ public class QueryLibraryPageServiceDTO extends PageQueryBaseVo {
     @ApiModelProperty("1交集2并集")
     private Integer intersection;
 
+    private String ageGroup;
+
     private Long brandId;
 
 }
diff --git a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
index 8de84945..98956f0a 100644
--- a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
@@ -167,6 +167,10 @@ public class LibraryServiceImpl extends ServiceImpl<LibraryMapper, Library> impl
             queryWrapper.like("name", query.getPictureName());
         }
 
+        if (!StringUtils.isEmpty(query.getAgeGroup())) {
+            queryWrapper.eq("level3_type", query.getAgeGroup());
+        }
+
         if (query.getBrandId() != null) {
             QueryWrapper<BrandRelLibrary> qw = new QueryWrapper<>();
             qw.lambda().eq(BrandRelLibrary::getBrandId, query.getBrandId());
diff --git a/src/main/java/com/ai/da/service/impl/WorkspaceServiceImpl.java b/src/main/java/com/ai/da/service/impl/WorkspaceServiceImpl.java
index 3d9450f2..1536ffd3 100644
--- a/src/main/java/com/ai/da/service/impl/WorkspaceServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/WorkspaceServiceImpl.java
@@ -2,6 +2,7 @@ package com.ai.da.service.impl;
 
 import com.ai.da.common.config.exception.BusinessException;
 import com.ai.da.common.context.UserContext;
+import com.ai.da.common.enums.CollectionLevel1TypeEnum;
 import com.ai.da.common.enums.LibraryLevel1TypeEnum;
 import com.ai.da.common.response.PageBaseResponse;
 import com.ai.da.common.response.ResultEnum;
@@ -84,6 +85,10 @@ public class WorkspaceServiceImpl extends ServiceImpl<WorkspaceMapper, Workspace
     @Resource
     private WorkspaceService workspaceService;
 
+    @Resource
+    private CollectionElementMapper collectionElementMapper;
+    @Resource
+    private CollectionElementRelModelMapper collectionElementRelModelMapper;
 
     @Override
     public IPage<WorkspaceVO> selectWorkspacePage(IPage<WorkspaceVO> page, WorkspaceVO workspace) {
@@ -755,7 +760,21 @@ public class WorkspaceServiceImpl extends ServiceImpl<WorkspaceMapper, Workspace
 
                 if (projectDTO.getProcess().equals(DesignProcess.SERIES_DESIGN.name()) || projectDTO.getProcess().equals(DesignProcess.SKETCH_COLLAGE_PROCESS.name())) {
                     SysFile sysFile = sysFileService.getOneBySex(projectDTO.getStyleId(), projectDTO.getWorkspace().getSex(), projectDTO.getWorkspace().getAgeGroup());
-
+                    CollectionElement collectionElement = new CollectionElement();
+                    collectionElement.setAccountId(userInfo.getId());
+                    collectionElement.setProjectId(project.getId());
+                    collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MODEL.getRealName());
+                    collectionElement.setName(sysFile.getName());
+                    collectionElement.setUrl(sysFile.getUrl());
+                    collectionElement.setMd5(sysFile.getMd5());
+                    collectionElement.setCreateDate(new Date());
+                    collectionElement.setHasPin((byte) 0);
+                    collectionElementMapper.insert(collectionElement);
+                    CollectionElementRelModel collectionElementRelModel = new CollectionElementRelModel();
+                    collectionElementRelModel.setCollectionElementId(collectionElement.getId());
+                    collectionElementRelModel.setRelationId(sysFile.getId());
+                    collectionElementRelModel.setRelationType("System");
+                    collectionElementRelModelMapper.insert(collectionElementRelModel);
                     if (projectDTO.getWorkspace().getSex().equals(Sex.FEMALE.getValue())) {
                         workspace.setMannequinFemaleId(sysFile.getId());
                         workspace.setMannequinFemaleType("System");