From 7ef4892f9f24f941aca37e6b3991b808a0aca619 Mon Sep 17 00:00:00 2001 From: zhanghua <314079846@qq.com> Date: 星期五, 08 九月 2023 11:16:35 +0800 Subject: [PATCH] 优化 --- ycl-platform/src/main/java/com/ycl/service/message/impl/MessageColumnSetServiceImpl.java | 15 +++++++++++++++ 1 files changed, 15 insertions(+), 0 deletions(-) diff --git a/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageColumnSetServiceImpl.java b/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageColumnSetServiceImpl.java index be23e71..9f134c3 100644 --- a/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageColumnSetServiceImpl.java +++ b/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageColumnSetServiceImpl.java @@ -4,7 +4,11 @@ import com.ycl.entity.message.MessageColumnSet; import com.ycl.mapper.message.MessageColumnSetMapper; import com.ycl.service.message.IMessageColumnSetService; +import com.ycl.vo.message.MessageColumnSetVO; import org.springframework.stereotype.Service; + +import javax.annotation.Resource; +import java.util.List; /** * <p> @@ -17,4 +21,15 @@ @Service public class MessageColumnSetServiceImpl extends ServiceImpl<MessageColumnSetMapper, MessageColumnSet> implements IMessageColumnSetService { + @Resource + MessageColumnSetMapper messageColumnSetMapper; + + @Override + public List<MessageColumnSetVO> listColumnAndPage(Long userId) { + List<MessageColumnSetVO> messageColumnSetVOS = messageColumnSetMapper.selectColumn(userId); + messageColumnSetVOS.forEach(item -> { + item.setMessageNumber(messageColumnSetMapper.selectColumnCount(userId, item.getId())); + }); + return messageColumnSetVOS; + } } -- Gitblit v1.8.0