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