zhanghua
2025-04-14 1cad14bca191807e18705c3a5526eda8151be439
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;
    }
}