【增加】获取单个 image 接口

cherishsince
1 parent e97408b3
Showing 4 changed files with 28 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/convert/AiImageConvert.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
@@ -36,6 +36,12 @@ public class AiImageController {
         return CommonResult.success(aiImageService.list(req));
     }
+    @Operation(summary = "获取image信息", description = "获取image信息")
+    @GetMapping("/get")
+    public CommonResult<AiImageListRespVO> get(@RequestParam("id") Long id) {
+        return CommonResult.success(aiImageService.get(id));
+    }
+
     @Operation(summary = "dall2/dall3绘画", description = "openAi dall3是付费的!")
     @PostMapping("/dall")
     public AiImageDallRespVO dallDrawing(@Validated @RequestBody AiImageDallReqVO req) {
@@ -48,6 +48,14 @@ public interface AiImageConvert {
     List<AiImageListRespVO> convertAiImageListRespVO(List<AiImageDO> list);
     /**
+     * 转换 - AiImageListRespVO
+     *
+     * @param aiImageDO
+     * @return
+     */
+    AiImageListRespVO convertAiImageListRespVO(AiImageDO aiImageDO);
+
+    /**
      * 转换 - AiImageMidjourneyOperationsVO
      *
      * @param component
@@ -21,6 +21,14 @@ public interface AiImageService {
     PageResult<AiImageListRespVO> list(AiImageListReqVO req);
     /**
+     * 获取 - image 信息
+     *
+     * @param id
+     * @return
+     */
+    AiImageListRespVO get(Long id);
+
+    /**
      * ai绘画 - dall2/dall3 绘画
      *
      * @param req
@@ -93,6 +93,12 @@ public class AiImageServiceImpl implements AiImageService {
     }
     @Override
+    public AiImageListRespVO get(Long id) {
+        AiImageDO aiImageDO = aiImageMapper.selectById(id);
+        return AiImageConvert.INSTANCE.convertAiImageListRespVO(aiImageDO);
+    }
+
+    @Override
     public AiImageDallRespVO dallDrawing(AiImageDallReqVO req) {
         // 保存数据库
         AiImageDO aiImageDO = doSave(req.getPrompt(), req.getSize(), req.getModel(),