diff --git a/src/main/java/com/ai/da/controller/GenerateController.java b/src/main/java/com/ai/da/controller/GenerateController.java index 064e3c4e..2db4ef8b 100644 --- a/src/main/java/com/ai/da/controller/GenerateController.java +++ b/src/main/java/com/ai/da/controller/GenerateController.java @@ -109,7 +109,7 @@ public class GenerateController { @ApiOperation(value = "获取姿势变换生成结果") @PostMapping("/poseTransformResult") public Response> getPoseTransformationResults(@Valid @RequestBody List taskIdList) { - List generateResult = generateService.getPoseTransformationResult(taskIdList); + List generateResult = generateService.getPoseTransformationResult(taskIdList, null, null); return Response.success(generateResult); } diff --git a/src/main/java/com/ai/da/service/GenerateService.java b/src/main/java/com/ai/da/service/GenerateService.java index 9e934941..8cea57ae 100644 --- a/src/main/java/com/ai/da/service/GenerateService.java +++ b/src/main/java/com/ai/da/service/GenerateService.java @@ -56,11 +56,10 @@ public interface GenerateService extends IService { void processPoseTransformResult(String taskId, String gifUrl, String videoUrl, String imageUrl); - List getPoseTransformationResult(List taskIdList); + List getPoseTransformationResult(List taskIdList, Long projectId, Boolean like); void updatePoseTransferStatus(String taskId, String status); - List getPoseTransformationResultList(Long projectId, boolean like); CollectionSort disOrLikePose(Long transformedId, String likeOrDislike, Long projectId, Long sortLikeParentId); diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java index 68bcce6d..ac085ec0 100644 --- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java @@ -2063,8 +2063,8 @@ public class UserLikeGroupServiceImpl extends ServiceImpl likedList = generateService.getPoseTransformationResultList(projectDTO.getId(), true); - List list = generateService.getPoseTransformationResultList(projectDTO.getId(), false); + List likedList = generateService.getPoseTransformationResult(null, projectDTO.getId(), true); + List list = generateService.getPoseTransformationResult(null, projectDTO.getId(), false); PoseTransferVO vo = new PoseTransferVO(); vo.setLikedList(likedList); vo.setList(list);