【增加】根据多个id获取 image 信息，用于前端轮训任务

cherishsince
1 parent 39ecc7a0
Showing 3 changed files with 34 additions and 0 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageService.java
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
 package cn.iocoder.yudao.module.ai.controller.admin.image;
  
+import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.ObjUtil;
 import cn.iocoder.yudao.framework.ai.core.model.midjourney.api.MidjourneyApi;
 import cn.iocoder.yudao.framework.common.pojo.CommonResult;
@@ -25,6 +26,10 @@ import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
  
+import java.util.Collections;
+import java.util.List;
+import java.util.Objects;
+
 import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
 import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
  
@@ -54,6 +59,19 @@ public class AiImageController {
         return success(BeanUtils.toBean(image, AiImageRespVO.class));
     }
  
+    @Operation(summary = "获取【我的】绘图记录 - ids")
+    @GetMapping("/get-my-ids")
+    public CommonResult<List<AiImageRespVO>> getImageMyIds(@RequestParam("ids") List<Long> ids) {
+        List<AiImageDO> imageList = imageService.getImageByIds(ids);
+        if (CollUtil.isEmpty(imageList)) {
+            return success(Collections.emptyList());
+        }
+        List<AiImageDO> userImageList = imageList.stream()
+                .map(item -> ObjUtil.equal(getLoginUserId(), item.getUserId()) ? item : null)
+                .filter(Objects::nonNull).toList();
+        return success(BeanUtils.toBean(userImageList, AiImageRespVO.class));
+    }
+
     @Operation(summary = "生成图片")
     @PostMapping("/draw")
     public CommonResult<Long> drawImage(@Validated @RequestBody AiImageDrawReqVO drawReqVO) {
@@ -5,12 +5,15 @@ import cn.iocoder.yudao.framework.common.pojo.PageParam;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageDrawReqVO;
 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImagePageReqVO;
+import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageRespVO;
 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImageUpdatePublicStatusReqVO;
 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.midjourney.AiMidjourneyActionReqVO;
 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.midjourney.AiMidjourneyImagineReqVO;
 import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO;
 import jakarta.validation.Valid;
  
+import java.util.List;
+
 /**
  * AI 绘图 Service 接口
  *
@@ -36,6 +39,14 @@ public interface AiImageService {
     AiImageDO getImage(Long id);
  
     /**
+     * 获得绘图记录 - 多个
+     *
+     * @param ids
+     * @return
+     */
+    List<AiImageDO> getImageByIds(List<Long> ids);
+
+    /**
      * 绘制图片
      *
      * @param userId 用户编号
@@ -75,6 +75,11 @@ public class AiImageServiceImpl implements AiImageService {
     }
  
     @Override
+    public List<AiImageDO> getImageByIds(List<Long> ids) {
+        return imageMapper.selectBatchIds(ids);
+    }
+
+    @Override
     public Long drawImage(Long userId, AiImageDrawReqVO drawReqVO) {
         // 1. 保存数据库
         AiImageDO image = BeanUtils.toBean(drawReqVO, AiImageDO.class).setUserId(userId).setPublicStatus(false)