Commit 1c0cf4b860596de08a88602b2e3b8725b6f46502

Authored by cherishsince
1 parent 2eade5ad

【解决todo】所有转换都不依赖 convert 采用 BeanUtils

yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageListRespVO.java renamed to yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImagePageMyRespVO.java
1 package cn.iocoder.yudao.module.ai.controller.admin.image.vo; 1 package cn.iocoder.yudao.module.ai.controller.admin.image.vo;
2 2
3 import cn.iocoder.yudao.framework.common.pojo.PageParam; 3 import cn.iocoder.yudao.framework.common.pojo.PageParam;
4 -import com.baomidou.mybatisplus.annotation.FieldFill;  
5 -import com.baomidou.mybatisplus.annotation.IdType;  
6 -import com.baomidou.mybatisplus.annotation.TableField;  
7 -import com.baomidou.mybatisplus.annotation.TableId;  
8 import io.swagger.v3.oas.annotations.media.Schema; 4 import io.swagger.v3.oas.annotations.media.Schema;
9 import lombok.Data; 5 import lombok.Data;
10 import lombok.experimental.Accessors; 6 import lombok.experimental.Accessors;
11 7
12 -import java.time.LocalDateTime;  
13 import java.util.Map; 8 import java.util.Map;
14 9
15 // TODO @fan:可以考虑,复用 AiImageDallRespVO,统一成 AIImageRespVO 10 // TODO @fan:可以考虑,复用 AiImageDallRespVO,统一成 AIImageRespVO
@@ -22,7 +17,7 @@ import java.util.Map; @@ -22,7 +17,7 @@ import java.util.Map;
22 */ 17 */
23 @Data 18 @Data
24 @Accessors(chain = true) 19 @Accessors(chain = true)
25 -public class AiImageListRespVO extends PageParam { 20 +public class AiImagePageMyRespVO extends PageParam {
26 21
27 @Schema(description = "id编号", example = "1") 22 @Schema(description = "id编号", example = "1")
28 private Long id; 23 private Long id;
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/convert/AiImageConvert.java deleted
1 -package cn.iocoder.yudao.module.ai.convert;  
2 -  
3 -import org.springframework.ai.models.midjourney.MidjourneyMessage;  
4 -import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageDallReqVO;  
5 -import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageDallRespVO;  
6 -import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageListRespVO;  
7 -import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageMidjourneyOperationsVO;  
8 -import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO;  
9 -import org.mapstruct.Mapper;  
10 -import org.mapstruct.factory.Mappers;  
11 -  
12 -import java.util.List;  
13 -  
14 -// TODO @fan:convert 可以考虑去掉,使用 BeanUtils.copy 替代  
15 -/**  
16 - * ai image convert  
17 - *  
18 - * @author fansili  
19 - * @time 2024/4/18 16:39  
20 - * @since 1.0  
21 - */  
22 -@Mapper  
23 -public interface AiImageConvert {  
24 -  
25 - AiImageConvert INSTANCE = Mappers.getMapper(AiImageConvert.class);  
26 -  
27 - /**  
28 - * 转换 - AiImageDallDrawingRespVO  
29 - *  
30 - * @param req  
31 - * @return  
32 - */  
33 - AiImageDallRespVO convertAiImageDallDrawingRespVO(AiImageDO req);  
34 -  
35 - /**  
36 - * 转换 - AiImageDallDrawingRespVO  
37 - *  
38 - * @param req  
39 - * @return  
40 - */  
41 - AiImageDallRespVO convertAiImageDallDrawingRespVO(AiImageDallReqVO req);  
42 -  
43 - /**  
44 - * 转换 - AiImageListRespVO  
45 - *  
46 - * @param list  
47 - * @return  
48 - */  
49 - List<AiImageListRespVO> convertAiImageListRespVO(List<AiImageDO> list);  
50 -  
51 - /**  
52 - * 转换 - AiImageListRespVO  
53 - *  
54 - * @param aiImageDO  
55 - * @return  
56 - */  
57 - AiImageListRespVO convertAiImageListRespVO(AiImageDO aiImageDO);  
58 -  
59 - /**  
60 - * 转换 - AiImageMidjourneyOperationsVO  
61 - *  
62 - * @param component  
63 - * @return  
64 - */  
65 - AiImageMidjourneyOperationsVO convertAiImageMidjourneyOperationsVO(MidjourneyMessage.Component component);  
66 -  
67 - /**  
68 - * 转换 - AiImageDO  
69 - *  
70 - * @param req  
71 - * @return  
72 - */  
73 - AiImageDO convertAiImageDO(AiImageDallReqVO req);  
74 -}  
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/midjourneyHandler/YuDaoMidjourneyMessageHandler.java
@@ -2,23 +2,19 @@ package cn.iocoder.yudao.module.ai.service.image.midjourneyHandler; @@ -2,23 +2,19 @@ package cn.iocoder.yudao.module.ai.service.image.midjourneyHandler;
2 2
3 import cn.hutool.core.collection.CollUtil; 3 import cn.hutool.core.collection.CollUtil;
4 import cn.hutool.core.util.StrUtil; 4 import cn.hutool.core.util.StrUtil;
5 -import com.alibaba.fastjson.JSON;  
6 -import org.springframework.ai.models.midjourney.MidjourneyMessage;  
7 -import org.springframework.ai.models.midjourney.constants.MidjourneyGennerateStatusEnum;  
8 -import org.springframework.ai.models.midjourney.webSocket.MidjourneyMessageHandler;  
9 -import cn.iocoder.yudao.framework.common.util.json.JsonUtils;  
10 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageMidjourneyOperationsVO; 5 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageMidjourneyOperationsVO;
11 -import cn.iocoder.yudao.module.ai.convert.AiImageConvert;  
12 import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO; 6 import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO;
13 import cn.iocoder.yudao.module.ai.dal.mysql.image.AiImageMapper; 7 import cn.iocoder.yudao.module.ai.dal.mysql.image.AiImageMapper;
14 import cn.iocoder.yudao.module.ai.enums.AiImageStatusEnum; 8 import cn.iocoder.yudao.module.ai.enums.AiImageStatusEnum;
  9 +import com.alibaba.fastjson.JSON;
15 import lombok.AllArgsConstructor; 10 import lombok.AllArgsConstructor;
16 import lombok.extern.slf4j.Slf4j; 11 import lombok.extern.slf4j.Slf4j;
  12 +import org.springframework.ai.models.midjourney.MidjourneyMessage;
  13 +import org.springframework.ai.models.midjourney.constants.MidjourneyGennerateStatusEnum;
  14 +import org.springframework.ai.models.midjourney.webSocket.MidjourneyMessageHandler;
17 import org.springframework.stereotype.Component; 15 import org.springframework.stereotype.Component;
18 16
19 -import java.util.Collections;  
20 import java.util.List; 17 import java.util.List;
21 -import java.util.stream.Collectors;  
22 18
23 /** 19 /**
24 * yudao message handler 20 * yudao message handler
@@ -113,15 +109,16 @@ public class YuDaoMidjourneyMessageHandler implements MidjourneyMessageHandler { @@ -113,15 +109,16 @@ public class YuDaoMidjourneyMessageHandler implements MidjourneyMessageHandler {
113 } 109 }
114 110
115 private List<AiImageMidjourneyOperationsVO> getMidjourneyOperationsList(MidjourneyMessage midjourneyMessage) { 111 private List<AiImageMidjourneyOperationsVO> getMidjourneyOperationsList(MidjourneyMessage midjourneyMessage) {
116 - // 为空直接返回  
117 - if (CollUtil.isEmpty(midjourneyMessage.getComponents())) {  
118 - return Collections.emptyList();  
119 - }  
120 - // 将 component 转成 AiImageMidjourneyOperationsVO  
121 - return midjourneyMessage.getComponents().stream()  
122 - .map(componentType -> componentType.getComponents().stream()  
123 - .map(AiImageConvert.INSTANCE::convertAiImageMidjourneyOperationsVO)  
124 - .collect(Collectors.toList()))  
125 - .toList().stream().flatMap(List::stream).toList(); 112 +// // 为空直接返回
  113 +// if (CollUtil.isEmpty(midjourneyMessage.getComponents())) {
  114 +// return Collections.emptyList();
  115 +// }
  116 +// // 将 component 转成 AiImageMidjourneyOperationsVO
  117 +// return midjourneyMessage.getComponents().stream()
  118 +// .map(componentType -> componentType.getComponents().stream()
  119 +// .map(AiImageConvert.INSTANCE::convertAiImageMidjourneyOperationsVO)
  120 +// .collect(Collectors.toList()))
  121 +// .toList().stream().flatMap(List::stream).toList();
  122 + return null;
126 } 123 }
127 } 124 }