保存消息逻辑

core/src/main/java/com/wecloud/im/service/ImMessageService.java

@@ -180,10 +180,4 @@ public interface ImMessageService extends BaseService<ImMessage> {
 
     List<WeCloudMessageVo> syncListMessage(SyncListMessageParam param);
 
-    /**
-     * 保存消息到数据库
-     * @param imMessage
-     */
-    void saveMessageToDb(ImMessage imMessage);
-
 }

core/src/main/java/com/wecloud/im/service/SaveMessageService.java

+package com.wecloud.im.service;
+
+import com.baomidou.lock.annotation.Lock4j;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.wecloud.im.entity.ImClient;
+import com.wecloud.im.entity.ImConversationMembers;
+import com.wecloud.im.entity.ImMessage;
+import com.wecloud.im.param.ImConversationQueryVo;
+import com.wecloud.im.ws.utils.RedisUtils;
+import com.wecloud.utils.SnowflakeUtil;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.cache.Cache;
+import org.springframework.cache.CacheManager;
+import org.springframework.stereotype.Service;
+
+import java.time.Duration;
+import java.util.Date;
+import java.util.List;
+
+
+/**
+ * @Author Future
+ * @Date 2022/8/24 17:18
+ * @Description 本地缓存服务
+ */
+@Slf4j
+@Service
+public class SaveMessageService {
+
+    private static final String LAST_MSG_PREFIX = "last_msg_";
+
+
+    @Autowired
+    private ImMessageService imMessageService;
+
+    @Autowired
+    private RedisUtils redisUtils;
+
+    /**
+     * 持久化消息到数据库并更新相关缓存
+     * expire 锁过期时间 防止死锁
+     *
+     * @param imMessage
+     */
+    @Lock4j(keys = {"#imMessage.fkConversationId"}, expire = 3000, acquireTimeout = 1000)
+    public void saveMessageToDb(ImMessage imMessage) {
+        imMessage.setId(SnowflakeUtil.getId());
+        imMessage.setCreateTime(new Date());
+        this.setLastMessageIdAndFreshCash(imMessage);
+        imMessageService.save(imMessage);
+    }
+
+    /**
+     * 查找会话的最新一条消息
+     *
+     * @param message
+     * @return
+     */
+    private void setLastMessageIdAndFreshCash(ImMessage message) {
+        try {
+            // 执行业务方法
+            String key = LAST_MSG_PREFIX + message.getFkConversationId();
+            String messageId = redisUtils.getKey(key);
+            if (StringUtils.isNotBlank(messageId)) {
+                // 将Redis中最新的messageId存入
+                message.setPreMessageId(Long.valueOf(messageId));
+            } else {
+                // 落库查
+                ImMessage preMessage = imMessageService.getOne(new QueryWrapper<ImMessage>().lambda()
+                        .eq(ImMessage::getFkConversationId, message.getFkConversationId())
+                        .orderByDesc(ImMessage::getId)
+                        .last("limit 1")
+                );
+                if (preMessage != null) {
+                    message.setPreMessageId(preMessage.getId());
+                }
+            }
+            redisUtils.addKey(key, message.getId().toString(), Duration.ofMinutes(30));
+        } catch (Exception e) {
+            log.info("获取会话最后一条消息异常 ", e);
+        }
+    }
+
+}

core/src/main/java/com/wecloud/im/service/impl/ImConversationServiceImpl.java

@@ -62,6 +62,7 @@ import com.wecloud.im.service.ImConversationMembersService;
 import com.wecloud.im.service.ImConversationService;
 import com.wecloud.im.service.ImInboxService;
 import com.wecloud.im.service.ImMessageService;
+import com.wecloud.im.service.SaveMessageService;
 import com.wecloud.im.vo.ChatRoomMemberVo;
 import com.wecloud.im.vo.ConversationCountVo;
 import com.wecloud.im.vo.ConversationMemberVo;
@@ -162,6 +163,9 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
     @Resource
     private EhcacheService ehcacheService;
 
+    @Autowired
+    private SaveMessageService saveMessageService;
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public ImConversationCreateVo createImConversation(ImConversationCreate imConversationCreate) {
@@ -341,7 +345,7 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
             content.put("operator", imClientSender.getClientId()); //操作的client ID
             content.put("passivityOperator", clientToConversation.getClientId()); //被操作的client ID
             ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.INVITE_CLIENT_JOIN_CONVERSATION, imApplication, createClient, imConversation, JsonUtils.encodeJson(content));
-            boolean save = imMessageService.save(imMessage);
+            saveMessageService.saveMessageToDb(imMessage);
             // 发送给在群内的成员
             sendMsgToMembers(imConversation, membersList, createClient, imMessage, content);
             // 发送给被邀请人
@@ -403,7 +407,7 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
             content.put("passivityOperator", clientToBeRemove.getClientId()); //被操作的client ID
 
             ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.REMOVE_CLIENT_CONVERSATION, imApplication, createClient, imConversation, JsonUtils.encodeJson(content));
-            imMessageService.save(imMessage);
+            saveMessageService.saveMessageToDb(imMessage);
 
             membersList.removeIf(e -> e.getId().equals(members.getId()));
 
@@ -451,7 +455,7 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
         Map<String, Object> content = new HashMap<>();
         content.put("operator", currentClient.getClientId());
         ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.CONVERSATION_DISBAND, imApplication, currentClient, imConversation, JsonUtils.encodeJson(content));
-        imMessageService.save(imMessage);
+        saveMessageService.saveMessageToDb(imMessage);
 
         sendMsgToMembers(imConversation, membersList, currentClient, imMessage, content);
     }
@@ -527,11 +531,7 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
         // 生成消息id
         ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.LEAVE_CONVERSATION, imApplication, currentClient, imConversation, "");
         //  保存消息至消息表
-        boolean save = imMessageService.save(imMessage);
-        if (!save) {
-            throw new BusinessException("退出群聊错误");
-        }
-
+        saveMessageService.saveMessageToDb(imMessage);
         sendMsgToMembers(imConversation, membersList, currentClient, imMessage, null);
         // 群主退出 转移给下一个人
         if (GroupRoleEnum.OWNER.getCode().equals(members.getRole())) {
@@ -1022,7 +1022,7 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
         if (!membersList.isEmpty()) {
             // 保存事件消息
             ImMessage imMessage = MessageBuilder.buildEventMessage(msgType, application, operatorClient, group, JsonUtils.encodeJson(content));
-            imMessageService.save(imMessage);
+            saveMessageService.saveMessageToDb(imMessage);
             // 发送消息至群成员
             sendMsgToMembers(group, membersList, operatorClient, imMessage, content);
         }
@@ -1541,7 +1541,7 @@ public class ImConversationServiceImpl extends BaseServiceImpl<ImConversationMap
             content.put("operator", creator.getClientId()); //操作的client ID
             content.put("passivityOperator", member.getClientId()); //被操作的client ID
             ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.CLIENT_JOIN_NEW_CONVERSATION, application, creator, conversation, JsonUtils.encodeJson(content));
-            imMessageService.save(imMessage);
+            saveMessageService.saveMessageToDb(imMessage);
             // 投递消息
             couriers.deliver(imMessage, content, creator, member, WsResponseCmdEnum.CONVERSATION_EVENT_MSG);
         }

core/src/main/java/com/wecloud/im/service/impl/ImGroupServiceImpl.java

 package com.wecloud.im.service.impl;
 
+import com.wecloud.im.service.SaveMessageService;
 import io.geekidea.springbootplus.framework.common.exception.BusinessException;
 import io.geekidea.springbootplus.framework.shiro.util.SecurityUtils;
 import lombok.AllArgsConstructor;
@@ -12,6 +13,7 @@ import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
 
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -86,10 +88,8 @@ public class ImGroupServiceImpl implements ImGroupService {
      */
     private final ImConversationMapper imConversationMapper;
 
-    /**
-     * 消息服务
-     */
-    private final ImMessageService imMessageService;
+    @Autowired
+    private SaveMessageService saveMessageService;
 
     @Override
     public Long createGroup(String creatorClientId, String groupName, List<String> memberIds) {
@@ -163,7 +163,7 @@ public class ImGroupServiceImpl implements ImGroupService {
         ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.CONVERSATION_DISBAND, application,
                 operator, conversation, JsonUtils.encodeJson(content));
         //  保存消息至消息表
-        imMessageService.save(imMessage);
+        saveMessageService.saveMessageToDb(imMessage);
         conversationService.sendMsgToMembers(conversation, membersList, operator, imMessage, content);
         return true;
 
@@ -236,7 +236,7 @@ public class ImGroupServiceImpl implements ImGroupService {
             content.put("passivityOperator", newMember.getClientId());
             ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.INVITE_CLIENT_JOIN_CONVERSATION,
                     imApplication, inviter, conversation, JsonUtils.encodeJson(content));
-            boolean save = imMessageService.save(imMessage);
+            saveMessageService.saveMessageToDb(imMessage);
             // 发送给在群内的成员
             conversationService.sendMsgToMembers(conversation, oldMembers, inviter, imMessage, content);
             // 发送给被邀请人
@@ -286,7 +286,7 @@ public class ImGroupServiceImpl implements ImGroupService {
                 content.put("operator", operator.getClientId());
                 // 自己主动退出
                 ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.LEAVE_CONVERSATION, imApplication, operator, conversation, JsonUtils.encodeJson(content));
-                imMessageService.save(imMessage);
+                saveMessageService.saveMessageToDb(imMessage);
                 // 发送给在群内的成员
                 conversationService.sendMsgToMembers(conversation, membersList, operator, imMessage, content);
             } else {
@@ -294,7 +294,7 @@ public class ImGroupServiceImpl implements ImGroupService {
                 // 被操作的client ID
                 content.put("passivityOperator", members.getClientId());
                 ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.REMOVE_CLIENT_CONVERSATION, imApplication, operator, conversation, JsonUtils.encodeJson(content));
-                imMessageService.save(imMessage);
+                saveMessageService.saveMessageToDb(imMessage);
                 // 发送给在群内的成员
                 conversationService.sendMsgToMembers(conversation, membersList, operator, imMessage, content);
             }
@@ -394,7 +394,7 @@ public class ImGroupServiceImpl implements ImGroupService {
         Map<String, Object> content = new HashMap<>();
         content.put("operator", newGroupOwner.getClientId());
         ImMessage imMessage = MessageBuilder.buildEventMessage(MsgTypeEnum.CONVERSATION_NEW_CREATOR, imApplication, newGroupOwnerClient, conversation, JsonUtils.encodeJson(content));
-        imMessageService.save(imMessage);
+        saveMessageService.saveMessageToDb(imMessage);
         // 发送给在群内的成员
         List<ImConversationMembers> existMemberList =
                 conversationMembersService.list(Wrappers.<ImConversationMembers>lambdaQuery()