From 3af957a67cf00b8d64eb61406df1f8b8072cb08c Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期一, 27 五月 2024 11:19:53 +0800
Subject: [PATCH] 修改秒杀/团购 websocket消息推送

---
 ruoyi-modules/ruoyi-goods/src/main/java/com/ruoyi/goods/service/impl/GoodsSeckillServiceImpl.java |   31 +++++++++++++++++++++++++------
 1 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/ruoyi-modules/ruoyi-goods/src/main/java/com/ruoyi/goods/service/impl/GoodsSeckillServiceImpl.java b/ruoyi-modules/ruoyi-goods/src/main/java/com/ruoyi/goods/service/impl/GoodsSeckillServiceImpl.java
index 8cf04e0..60955b8 100644
--- a/ruoyi-modules/ruoyi-goods/src/main/java/com/ruoyi/goods/service/impl/GoodsSeckillServiceImpl.java
+++ b/ruoyi-modules/ruoyi-goods/src/main/java/com/ruoyi/goods/service/impl/GoodsSeckillServiceImpl.java
@@ -2,6 +2,8 @@
 
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.ruoyi.common.core.constant.CacheConstants;
 import com.ruoyi.common.core.enums.StartStatusEnum;
 import com.ruoyi.common.core.exception.ServiceException;
@@ -17,18 +19,22 @@
 import com.ruoyi.goods.service.IGoodsSeckillService;
 import com.ruoyi.goods.service.IGoodsSkuService;
 import com.ruoyi.goods.service.async.AsyncMethodService;
-import com.ruoyi.system.api.WebSocketUsers;
+import com.ruoyi.system.api.constants.NotificationTypeConstant;
 import com.ruoyi.system.api.domain.GoodsSeckill;
 import com.ruoyi.system.api.domain.GoodsSku;
 import com.ruoyi.system.api.domain.dto.ListStatusDTO;
 import com.ruoyi.system.api.feignClient.OrderClient;
 import com.ruoyi.system.api.feignClient.SysUserClient;
+import java.time.LocalDateTime;
 import java.util.List;
+import java.util.Map;
 import java.util.Optional;
+import java.util.concurrent.ConcurrentHashMap;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import util.WebSocketUsers;
 
 /**
  * <p>
@@ -48,6 +54,8 @@
     private final RedisService redisService;
     private final SysUserClient sysUserClient;
     private final AsyncMethodService asyncMethodService;
+    // 创建一个静态共享的ObjectMapper实例以重用
+    private static final ObjectMapper objectMapper = new ObjectMapper();
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void addGoodsSeckill(GoodsSeckillDTO dto) {
@@ -147,7 +155,7 @@
      * @param seckillId 秒杀id
      */
     @Override
-    public void startSeckill(Long seckillId) {
+    public void startSeckill(Long seckillId) throws JsonProcessingException {
         log.info(">>>>>>>>>>>>>>>>>>>>{}秒杀开始<<<<<<<<<<<<<<<<<<<<", seckillId);
         GoodsSeckill goodsSeckill = this.getById(seckillId);
         //秒杀商品不能为空且状态为未开始
@@ -162,7 +170,13 @@
                     goodsSeckill.getSeckillStock());
         }
         //推送秒杀开始消息
-        WebSocketUsers.sendMessageToUsersByText("秒杀活动已开始");
+        Map<String, Object> map = new ConcurrentHashMap<>();
+        map.put("notification_type", NotificationTypeConstant.SECKILL);
+        map.put("notification_time", LocalDateTime.now());
+        map.put("message_type", "start");
+        String msg = objectMapper.writeValueAsString(map);
+        WebSocketUsers.sendMessageToUsersByText(msg);
+        log.info("===================>发送websocket通知,消息体{}", msg);
     }
 
     /**
@@ -171,7 +185,7 @@
      * @param seckillId 秒杀id
      */
     @Override
-    public void endSeckill(Long seckillId) {
+    public void endSeckill(Long seckillId) throws JsonProcessingException {
         log.info(">>>>>>>>>>>>>>>>>>>>{}秒杀结束<<<<<<<<<<<<<<<<<<<<", seckillId);
         GoodsSeckill goodsSeckill = this.getById(seckillId);
         if (StringUtils.isNotNull(goodsSeckill)
@@ -182,7 +196,12 @@
 //            将秒杀商品从缓存中移除
             redisService.deleteObject(CacheConstants.SECKILL_GOODS + goodsSeckill.getId());
         }
-        //TODO websocket 推送秒杀结束消息
-        WebSocketUsers.sendMessageToUsersByText("秒杀活动已结束");
+        Map<String, Object> map = new ConcurrentHashMap<>();
+        map.put("notification_type", NotificationTypeConstant.SECKILL);
+        map.put("notification_time", LocalDateTime.now());
+        map.put("message_type", "end");
+        String msg = objectMapper.writeValueAsString(map);
+        WebSocketUsers.sendMessageToUsersByText(msg);
+        log.info("===================>发送websocket通知,消息体{}", msg);
     }
 }

--
Gitblit v1.7.1