添加机构

yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/alertmessage/AlertMessageServiceImpl.java

@@ -117,6 +117,7 @@ public class AlertMessageServiceImpl implements AlertMessageService {
 
         // 转换按日期分组的数据
         List<AlertMessageStatisticsRespVO.DailyAlertData> dailyData = new ArrayList<>();
+        if (dailyStatistics != null) {
             for (Map<String, Object> stat : dailyStatistics) {
                 AlertMessageStatisticsRespVO.DailyAlertData dailyAlertData = new AlertMessageStatisticsRespVO.DailyAlertData();
                 // 处理日期类型转换
@@ -126,32 +127,59 @@ public class AlertMessageServiceImpl implements AlertMessageService {
                 } else if (dateObj instanceof LocalDate) {
                     dailyAlertData.setDate((LocalDate) dateObj);
                 }
-            dailyAlertData.setTotalCount(((Number) stat.get("totalCount")).longValue());
-            dailyAlertData.setReadCount(((Number) stat.get("readCount")).longValue());
-            dailyAlertData.setUnreadCount(((Number) stat.get("unreadCount")).longValue());
-            dailyAlertData.setSosCount(((Number) stat.get("sosCount")).longValue());
-            dailyAlertData.setAnalysisCount(((Number) stat.get("analysisCount")).longValue());
-            dailyAlertData.setSosReadCount(((Number) stat.get("sosReadCount")).longValue());
-            dailyAlertData.setSosUnreadCount(((Number) stat.get("sosUnreadCount")).longValue());
-            dailyAlertData.setAnalysisReadCount(((Number) stat.get("analysisReadCount")).longValue());
-            dailyAlertData.setAnalysisUnreadCount(((Number) stat.get("analysisUnreadCount")).longValue());
+                dailyAlertData.setTotalCount(getNumberValue(stat, "totalCount"));
+                dailyAlertData.setReadCount(getNumberValue(stat, "readCount"));
+                dailyAlertData.setUnreadCount(getNumberValue(stat, "unreadCount"));
+                dailyAlertData.setSosCount(getNumberValue(stat, "sosCount"));
+                dailyAlertData.setAnalysisCount(getNumberValue(stat, "analysisCount"));
+                dailyAlertData.setSosReadCount(getNumberValue(stat, "sosReadCount"));
+                dailyAlertData.setSosUnreadCount(getNumberValue(stat, "sosUnreadCount"));
+                dailyAlertData.setAnalysisReadCount(getNumberValue(stat, "analysisReadCount"));
+                dailyAlertData.setAnalysisUnreadCount(getNumberValue(stat, "analysisUnreadCount"));
                 dailyData.add(dailyAlertData);
             }
+        }
 
         // 构建返回结果
         AlertMessageStatisticsRespVO result = new AlertMessageStatisticsRespVO();
-        result.setTotalCount(((Number) totalStats.get("totalCount")).longValue());
-        result.setReadCount(((Number) totalStats.get("readCount")).longValue());
-        result.setUnreadCount(((Number) totalStats.get("unreadCount")).longValue());
-        result.setSosCount(((Number) totalStats.get("sosCount")).longValue());
-        result.setAnalysisCount(((Number) totalStats.get("analysisCount")).longValue());
-        result.setSosReadCount(((Number) totalStats.get("sosReadCount")).longValue());
-        result.setSosUnreadCount(((Number) totalStats.get("sosUnreadCount")).longValue());
-        result.setAnalysisReadCount(((Number) totalStats.get("analysisReadCount")).longValue());
-        result.setAnalysisUnreadCount(((Number) totalStats.get("analysisUnreadCount")).longValue());
+        if (totalStats != null) {
+            result.setTotalCount(getNumberValue(totalStats, "totalCount"));
+            result.setReadCount(getNumberValue(totalStats, "readCount"));
+            result.setUnreadCount(getNumberValue(totalStats, "unreadCount"));
+            result.setSosCount(getNumberValue(totalStats, "sosCount"));
+            result.setAnalysisCount(getNumberValue(totalStats, "analysisCount"));
+            result.setSosReadCount(getNumberValue(totalStats, "sosReadCount"));
+            result.setSosUnreadCount(getNumberValue(totalStats, "sosUnreadCount"));
+            result.setAnalysisReadCount(getNumberValue(totalStats, "analysisReadCount"));
+            result.setAnalysisUnreadCount(getNumberValue(totalStats, "analysisUnreadCount"));
+        } else {
+            // 如果查询结果为空，设置默认值
+            result.setTotalCount(0L);
+            result.setReadCount(0L);
+            result.setUnreadCount(0L);
+            result.setSosCount(0L);
+            result.setAnalysisCount(0L);
+            result.setSosReadCount(0L);
+            result.setSosUnreadCount(0L);
+            result.setAnalysisReadCount(0L);
+            result.setAnalysisUnreadCount(0L);
+        }
         result.setDailyData(dailyData);
 
         return result;
+    }
 
+    /**
+     * 安全地从Map中获取Number值并转换为Long
+     */
+    private Long getNumberValue(Map<String, Object> map, String key) {
+        Object value = map.get(key);
+        if (value == null) {
+            return 0L;
+        }
+        if (value instanceof Number) {
+            return ((Number) value).longValue();
+        }
+        return 0L;
     }
 }