From cd74022d04acccd7d801dc87a077574e8e620086 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Thu, 16 Oct 2025 10:03:57 +0800
Subject: [PATCH] =?UTF-8?q?BUGFIX:generateLike=E6=B7=BB=E5=8A=A0=E5=B9=B4?=
 =?UTF-8?q?=E9=BE=84=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/ai/da/model/dto/GenerateLikeDTO.java       | 3 +++
 .../java/com/ai/da/service/impl/GenerateServiceImpl.java     | 5 +++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/ai/da/model/dto/GenerateLikeDTO.java b/src/main/java/com/ai/da/model/dto/GenerateLikeDTO.java
index 58734272..9e76dae3 100644
--- a/src/main/java/com/ai/da/model/dto/GenerateLikeDTO.java
+++ b/src/main/java/com/ai/da/model/dto/GenerateLikeDTO.java
@@ -22,6 +22,9 @@ public class GenerateLikeDTO {
     @ApiModelProperty("当一级类型为Sketchboard时，二级类型 Outwear Dress Blouse Skirt Trousers；当一级类型为Printboard时，二级类型 Slogan Logo Pattern")
     private String level2Type;
 
+    @ApiModelProperty("年龄段")
+    private String ageGroup;
+
     @ApiModelProperty("性别")
     private String gender;
 
diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index a4a8f628..5e7243ea 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -521,7 +521,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         // 2.2、添加到library
         AuthPrincipalVo userInfo = UserContext.getUserHolder();
         Long accountId = userInfo.getId();
-        Library library = setLibrary(accountId, generateLikeDTO, generateDetail.getUrl());
+        Library library = setLibrary(accountId, generateLikeDTO, generateDetail.getUrl(), generateLikeDTO.getAgeGroup());
         libraryService.save(library);
 
         // 3、更新generateDetail表的isLike列和libraryId列
@@ -555,12 +555,13 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         return Boolean.TRUE;
     }
 
-    public Library setLibrary(Long accountId, GenerateLikeDTO generateLikeDTO, String imageUrl) {
+    public Library setLibrary(Long accountId, GenerateLikeDTO generateLikeDTO, String imageUrl, String ageGroup) {
         Library library = new Library();
         library.setAccountId(accountId);
         library.setLevel1Type(generateLikeDTO.getLevel1Type());
         library.setLevel2Type(StringUtil.isNullOrEmpty(generateLikeDTO.getLevel2Type()) ? null : generateLikeDTO.getLevel2Type());
         library.setLevel3Type(StringUtil.isNullOrEmpty(generateLikeDTO.getGender()) ? null : generateLikeDTO.getGender());
+        library.setAgeGroup(ageGroup);
         library.setName(DateUtil.getTimeStamp(generateLikeDTO.getTimeZone()) + "_N_G");
         library.setUrl(imageUrl);
         try {