BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 9878abb03c7ebee3e234ee0a1f12491d9a6cb348

Authored by cherishsince
1 parent e97408b3

【增加】获取单个 image 接口

Inline Side-by-side
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
  View file @9878abb
... ... @@ -36,6 +36,12 @@ public class AiImageController {
36 36 return CommonResult.success(aiImageService.list(req));
37 37 }
38 38  
  39 + @Operation(summary = "获取image信息", description = "获取image信息")
  40 + @GetMapping("/get")
  41 + public CommonResult<AiImageListRespVO> get(@RequestParam("id") Long id) {
  42 + return CommonResult.success(aiImageService.get(id));
  43 + }
  44 +
39 45 @Operation(summary = "dall2/dall3绘画", description = "openAi dall3是付费的!")
40 46 @PostMapping("/dall")
41 47 public AiImageDallRespVO dallDrawing(@Validated @RequestBody AiImageDallReqVO req) {
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/convert/AiImageConvert.java
  View file @9878abb
... ... @@ -48,6 +48,14 @@ public interface AiImageConvert {
48 48 List<AiImageListRespVO> convertAiImageListRespVO(List<AiImageDO> list);
49 49  
50 50 /**
  51 + * 转换 - AiImageListRespVO
  52 + *
  53 + * @param aiImageDO
  54 + * @return
  55 + */
  56 + AiImageListRespVO convertAiImageListRespVO(AiImageDO aiImageDO);
  57 +
  58 + /**
51 59 * 转换 - AiImageMidjourneyOperationsVO
52 60 *
53 61 * @param component
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageService.java
  View file @9878abb
... ... @@ -21,6 +21,14 @@ public interface AiImageService {
21 21 PageResult<AiImageListRespVO> list(AiImageListReqVO req);
22 22  
23 23 /**
  24 + * 获取 - image 信息
  25 + *
  26 + * @param id
  27 + * @return
  28 + */
  29 + AiImageListRespVO get(Long id);
  30 +
  31 + /**
24 32 * ai绘画 - dall2/dall3 绘画
25 33 *
26 34 * @param req
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
  View file @9878abb
... ... @@ -93,6 +93,12 @@ public class AiImageServiceImpl implements AiImageService {
93 93 }
94 94  
95 95 @Override
  96 + public AiImageListRespVO get(Long id) {
  97 + AiImageDO aiImageDO = aiImageMapper.selectById(id);
  98 + return AiImageConvert.INSTANCE.convertAiImageListRespVO(aiImageDO);
  99 + }
  100 +
  101 + @Override
96 102 public AiImageDallRespVO dallDrawing(AiImageDallReqVO req) {
97 103 // 保存数据库
98 104 AiImageDO aiImageDO = doSave(req.getPrompt(), req.getSize(), req.getModel(),
... ...