diff --git a/src/main/java/com/ai/da/model/vo/UserLikeChooseVO.java b/src/main/java/com/ai/da/model/vo/UserLikeChooseVO.java index 69dc235e..6c78317c 100644 --- a/src/main/java/com/ai/da/model/vo/UserLikeChooseVO.java +++ b/src/main/java/com/ai/da/model/vo/UserLikeChooseVO.java @@ -1,5 +1,6 @@ package com.ai.da.model.vo; +import com.ai.da.mapper.primary.entity.Portfolio; import io.swagger.annotations.ApiModel; import io.swagger.annotations.ApiModelProperty; import lombok.AllArgsConstructor; @@ -24,4 +25,6 @@ public class UserLikeChooseVO { private String sex; private Integer beenPublished; + + private Portfolio portfolio; } 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 9dff1679..99f1c49e 100644 --- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java @@ -155,10 +155,12 @@ public class UserLikeGroupServiceImpl extends ServiceImpl qw = new QueryWrapper<>(); qw.lambda().eq(Portfolio::getUserLikeGroupSourceId, userGroupId); List portfolios = portfolioMapper.selectList(qw); + Portfolio portfolio = new Portfolio(); if (CollectionUtil.isNotEmpty(portfolios)) { + portfolio = portfolios.get(0); beenPublished = 1; } - return new UserLikeChooseVO(userGroupId, userLikeVOS, userLikeCollection, sex, beenPublished); + return new UserLikeChooseVO(userGroupId, userLikeVOS, userLikeCollection, sex, beenPublished, portfolio); } @Override