浏览代码

返回异常信息给前端

刘立冬 1 月之前
父节点
当前提交
bb2dee23d9

+ 3 - 0
risk-control-core/src/main/java/com/tzld/piaoquan/risk/control/model/dto/WorkWechatRoomDTO.java

@@ -1,5 +1,6 @@
 package com.tzld.piaoquan.risk.control.model.dto;
 
+import io.swagger.models.auth.In;
 import lombok.Data;
 
 @Data
@@ -12,5 +13,7 @@ public class WorkWechatRoomDTO {
     private String roomName;
 
     private Integer autoRemoveUserSwitch;
+    private Integer showTip;
+    private String tipInfo;
 
 }

+ 15 - 0
risk-control-core/src/main/java/com/tzld/piaoquan/risk/control/service/impl/WorkWechatServiceImpl.java

@@ -75,6 +75,8 @@ public class WorkWechatServiceImpl implements WorkWechatService {
 
         JSONObject chatRoomInfo = this.getChatRoomInfo(uuid, 0);
         Map<String, Integer> roomAutoRemoveUserStatusMap = this.getRoomAutoRemoveUserStatusMap();
+        Map<String, Integer> roomShowTip = this.getRoomShowTipMap();
+        Map<String, String> roomTipInfoMap = this.getRoomTipInfoMap();
 
         JSONArray roomList = chatRoomInfo.getJSONArray("roomList");
 
@@ -92,6 +94,8 @@ public class WorkWechatServiceImpl implements WorkWechatService {
             workWechatRoomDTO.setUuid(uuid);
             workWechatRoomDTO.setRoomId(roomId);
             workWechatRoomDTO.setRoomName(room.getString("nickname"));
+            workWechatRoomDTO.setShowTip(roomShowTip.getOrDefault(roomId, 0));
+            workWechatRoomDTO.setTipInfo(roomTipInfoMap.getOrDefault(roomId, ""));
             workWechatRoomDTO.setAutoRemoveUserSwitch(roomAutoRemoveUserStatusMap.getOrDefault(roomId, 0));
             filteredRooms.add(workWechatRoomDTO);
         }
@@ -207,4 +211,15 @@ public class WorkWechatServiceImpl implements WorkWechatService {
         List<WorkWechatRoomInfo> workWechatRoomInfos = workWechatRoomInfoMapper.selectByExample(example);
         return workWechatRoomInfos.stream().collect(Collectors.toMap(WorkWechatRoomInfo::getRoomId, WorkWechatRoomInfo::getAutoRemoveUserSwitch, (o1, o2) -> o2));
     }
+
+    private Map<String, Integer> getRoomShowTipMap() {
+        WorkWechatRoomInfoExample example = new WorkWechatRoomInfoExample();
+        List<WorkWechatRoomInfo> workWechatRoomInfos = workWechatRoomInfoMapper.selectByExample(example);
+        return workWechatRoomInfos.stream().collect(Collectors.toMap(WorkWechatRoomInfo::getRoomId, WorkWechatRoomInfo::getShowtip, (o1, o2) -> o2));
+    }
+    private Map<String, String> getRoomTipInfoMap() {
+        WorkWechatRoomInfoExample example = new WorkWechatRoomInfoExample();
+        List<WorkWechatRoomInfo> workWechatRoomInfos = workWechatRoomInfoMapper.selectByExample(example);
+        return workWechatRoomInfos.stream().collect(Collectors.toMap(WorkWechatRoomInfo::getRoomId, WorkWechatRoomInfo::getTipinfo, (o1, o2) -> o2));
+    }
 }