From aa4c2a9c4efb904b5664a2974c97a30365ce1039 Mon Sep 17 00:00:00 2001
From: haotian <2421912570@qq.com>
Date: Fri, 1 Aug 2025 15:05:58 +0800
Subject: [PATCH] =?UTF-8?q?1.=E5=AE=9E=E7=8E=B0=E8=8E=B7=E5=8F=96=E5=91=8A?=
 =?UTF-8?q?=E8=AD=A6=E6=B6=88=E6=81=AF=E7=B1=BB=E5=9E=8B=E5=88=86=E5=B8=83?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../module_admin/controller/message_controller.py    |  3 ++-
 .../module_admin/dao/message_dao.py                  |  9 +++++++++
 .../module_admin/service/message_service.py          | 12 ++++++++++++
 3 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/ruoyi-fastapi-backend/module_admin/controller/message_controller.py b/ruoyi-fastapi-backend/module_admin/controller/message_controller.py
index db228de..6f0836d 100644
--- a/ruoyi-fastapi-backend/module_admin/controller/message_controller.py
+++ b/ruoyi-fastapi-backend/module_admin/controller/message_controller.py
@@ -29,7 +29,8 @@ async def get_alarm_type_distribution(
     query_db: AsyncSession = Depends(get_db),
 ):
     result = await MessageService.get_alarm_type_distribution_services(query_db)
-    pass
+    return ResponseUtil.success(data=result)
+
 
 
 # 一键处理未处理的告警消息
diff --git a/ruoyi-fastapi-backend/module_admin/dao/message_dao.py b/ruoyi-fastapi-backend/module_admin/dao/message_dao.py
index db13636..d2b90c0 100644
--- a/ruoyi-fastapi-backend/module_admin/dao/message_dao.py
+++ b/ruoyi-fastapi-backend/module_admin/dao/message_dao.py
@@ -12,6 +12,15 @@ class MessageDao:
     系统消息模块数据库操作层
     """
 
+    @classmethod
+    async def get_alarm_type_distribution(cls, db: AsyncSession):
+        stmt = (
+            select(Message.specific_type, func.count(Message.specific_type).label("cnt"))
+            .where(Message.type == '0')
+            .group_by(Message.specific_type)
+        )
+        return (await db.execute(stmt)).mappings().all()
+
     @classmethod
     async def handle_all_message(cls, db: AsyncSession):
         query = (
diff --git a/ruoyi-fastapi-backend/module_admin/service/message_service.py b/ruoyi-fastapi-backend/module_admin/service/message_service.py
index b83d5a9..dc1d05a 100644
--- a/ruoyi-fastapi-backend/module_admin/service/message_service.py
+++ b/ruoyi-fastapi-backend/module_admin/service/message_service.py
@@ -13,6 +13,18 @@ class MessageService:
     """
     系统消息模块服务层
     """
+    # 获取告警消息分布
+    @classmethod
+    async def get_alarm_type_distribution_services(cls, query_db: AsyncSession):
+        result = await MessageDao.get_alarm_type_distribution(query_db)
+        # d = dict()
+        # t = 0
+        # for k, v in result.items():
+        #     d[k] = v
+        #     t += v
+        # for k, v in d.items():
+        #     result[k] = round(v/t, 2)
+        return result
 
     @classmethod
     async def handle_all_message_services(cls,query_db: AsyncSession):