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/MessageServiceImpl.java | 25 ++++++++++++++++++++++++- 1 files changed, 24 insertions(+), 1 deletions(-) diff --git a/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageServiceImpl.java b/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageServiceImpl.java index 6c5e7e1..cbf68e8 100644 --- a/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageServiceImpl.java +++ b/ycl-platform/src/main/java/com/ycl/service/message/impl/MessageServiceImpl.java @@ -1,6 +1,8 @@ package com.ycl.service.message.impl; +import cn.hutool.core.collection.CollUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.fasterxml.jackson.databind.ObjectMapper; @@ -33,7 +35,6 @@ import javax.annotation.Resource; import javax.mail.internet.MimeMessage; -import javax.xml.ws.Response; import java.util.ArrayList; import java.util.Date; import java.util.List; @@ -254,6 +255,28 @@ } @Override + public Page<Message> queryMyMessage(Page<Message> page, Long userId) { + List<Message> list = list(); + ArrayList<Long> ids = new ArrayList<>(); + list.forEach(item -> { + String targetTo = item.getTargetTo(); + if (StringUtils.isNotBlank(targetTo)) { + String[] split = targetTo.split(","); + for (String s : split) { + if (userId.equals(Long.getLong(s))) { + ids.add(item.getId()); + break; + } + } + } + }); + if (CollUtil.isEmpty(ids)) { + return new Page<>(); + } + return baseMapper.selectPage(page, new QueryWrapper<Message>().lambda().in(Message::getId, ids)); + } + + @Override public Boolean deleteMessages(List<Long> ids) { LambdaQueryWrapper<Message> queryWrapper = new LambdaQueryWrapper<>(); queryWrapper.in(Message::getId, ids).or().in(Message::getParentId, ids); -- Gitblit v1.8.0