diff --git a/src/main/java/com/ai/da/common/enums/CreditsEventsEnum.java b/src/main/java/com/ai/da/common/enums/CreditsEventsEnum.java index 2711d04b..5dc8ff30 100644 --- a/src/main/java/com/ai/da/common/enums/CreditsEventsEnum.java +++ b/src/main/java/com/ai/da/common/enums/CreditsEventsEnum.java @@ -3,6 +3,9 @@ package com.ai.da.common.enums; import lombok.AllArgsConstructor; import lombok.Getter; +import java.util.Arrays; +import java.util.List; + @AllArgsConstructor @Getter public enum CreditsEventsEnum { @@ -45,4 +48,9 @@ public enum CreditsEventsEnum { * 对应事件需要消耗or获得的积分 */ private String value; + + public static List generateFunctionNames() { + return Arrays.asList(SLOGAN.name, LOGO.name, PATTERN.name, MOOD_BOARD.name, SKETCH_BOARD.name, + TO_PRODUCT_IMAGE.name, RELIGHT.name, IMAGE_TO_SKETCH.name, POSE_TRANSFORMATION.name); + } } diff --git a/src/main/java/com/ai/da/controller/ConvenientInquiryController.java b/src/main/java/com/ai/da/controller/ConvenientInquiryController.java index 9925474f..9acaa9a4 100644 --- a/src/main/java/com/ai/da/controller/ConvenientInquiryController.java +++ b/src/main/java/com/ai/da/controller/ConvenientInquiryController.java @@ -199,4 +199,11 @@ public class ConvenientInquiryController { public Response> getGenerateFrequency(@Valid @RequestBody AccountCreditsUsageQueryDTO queryDTO){ return Response.success(convenientInquiryService.getGenerateFrequency(queryDTO)); } + + @ApiOperation("获取所有生成功能的名字") + @GetMapping("/getAllGenerateFuncName") + public Response> getAllGenerateFuncName(){ + return Response.success(convenientInquiryService.getAllGenerateFuncName()); + } + } diff --git a/src/main/java/com/ai/da/controller/ElementController.java b/src/main/java/com/ai/da/controller/ElementController.java index ee0b65f6..96010a03 100644 --- a/src/main/java/com/ai/da/controller/ElementController.java +++ b/src/main/java/com/ai/da/controller/ElementController.java @@ -112,12 +112,12 @@ public class ElementController { @ApiOperation(value = "图片分割") @PostMapping("/imageSegmentation") public Response> selectedImageSeg( - @RequestPart(value = "files", required = false) MultipartFile[] files, + @RequestPart(value = "file", required = false) MultipartFile[] file, @RequestParam(value = "type", required = false) @Pattern(regexp = "sketch|product", message = "类型必须是sketch或product") String type, @RequestParam(value = "id", required = false) Long id) { // 过滤空文件 - List nonEmptyFiles = Arrays.stream(files) - .filter(file -> !file.isEmpty()) + List nonEmptyFiles = Arrays.stream(file) + .filter(item -> !item.isEmpty()) .collect(Collectors.toList()); // 参数校验 if ((nonEmptyFiles.isEmpty()) && id == null) { diff --git a/src/main/java/com/ai/da/service/ConvenientInquiryService.java b/src/main/java/com/ai/da/service/ConvenientInquiryService.java index 8cd4eb6b..64bf0e85 100644 --- a/src/main/java/com/ai/da/service/ConvenientInquiryService.java +++ b/src/main/java/com/ai/da/service/ConvenientInquiryService.java @@ -56,4 +56,6 @@ public interface ConvenientInquiryService extends IService { String exportTransactionRecords(QueryPaymentInfoDTO queryPaymentInfoDTO, HttpServletResponse response); PageBaseResponse getGenerateFrequency(AccountCreditsUsageQueryDTO queryDTO); + + List getAllGenerateFuncName(); } diff --git a/src/main/java/com/ai/da/service/impl/ConvenientInquiryServiceImpl.java b/src/main/java/com/ai/da/service/impl/ConvenientInquiryServiceImpl.java index 00192bc7..747e5b62 100644 --- a/src/main/java/com/ai/da/service/impl/ConvenientInquiryServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/ConvenientInquiryServiceImpl.java @@ -905,4 +905,8 @@ public class ConvenientInquiryServiceImpl extends ServiceImpl getAllGenerateFuncName(){ + return CreditsEventsEnum.generateFunctionNames(); + } + } diff --git a/src/main/resources/mapper/primary/AccountMapper.xml b/src/main/resources/mapper/primary/AccountMapper.xml index c41ba29d..c21f0c6b 100644 --- a/src/main/resources/mapper/primary/AccountMapper.xml +++ b/src/main/resources/mapper/primary/AccountMapper.xml @@ -99,7 +99,7 @@