feat(redis): redis整机替换单据值守 TencentBlueKing#16385

Author: xiepaup

dbm-ui/backend/core/translation/constants.py

@@ -47,6 +47,7 @@
     "backend/ticket/constants.py",
     # aidev 快捷指令中文
     "backend/dbm_aiagent/agent/commands/commands.py",
+    "backend/dbm_aiagent/agent/commands/redis_commands.py",
 ]
 ALL_EXCLUDE_DIRS = set(EXCLUDE_DIRS + EXCLUDE_FILE_PATHS)
 

dbm-ui/backend/db_services/redis/autofix/message.py

@@ -19,6 +19,8 @@
 from backend.core.notify.handlers import CmsiHandler
 from backend.db_meta.enums import ClusterType
 from backend.db_meta.models import AppCache
+from backend.dbm_aiagent.agent.constants import DBMAgentCode
+from backend.dbm_aiagent.agent.handlers import AgentHandler
 from backend.utils.time import date2str
 
 from .enums import AutofixItem
@@ -55,7 +57,12 @@ def send_msg_2_qywx(sub_title: str, msgs):
             content += _("业务DBA : {}(@{})\n".format(redis_DBA[0], redis_DBA[0]))
         else:
             content += _("{} : {}\n".format(k, v))
+    if db_type == DBType.Redis.value:
+        content = _("""查询这个{}集群最新的qps和CPU情况""".format("-".join(sub_title.split("-")[:-1])))
+        rest = AgentHandler.ask_agent_with_content(DBMAgentCode.REDIS_TASK_GUARDIAN.value, "", redis_DBA)
+        content += _("当前负载 : {}\n".format(rest))
     content += _("消息时间 : {}\n".format(date2str(datetime.datetime.now(), "%Y-%m-%d %H:%M:%S")))
+
     CmsiHandler(_("Tendis自愈"), content, msg_ids).send_wecom_robot()
 
 

dbm-ui/backend/dbm_aiagent/agent/commands/__init__.py

@@ -14,6 +14,7 @@
 from aidev_agent.services.command_handler import CommandProcessor as BaseCommandProcessor
 
 from .commands import *
+from .redis_commands import *
 from .register import command, register_command
 
 CommandProcessor = BaseCommandProcessor()

dbm-ui/backend/dbm_aiagent/agent/commands/redis_commands.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+"""
+TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-DB管理系统(BlueKing-BK-DBM) available.
+Copyright (C) 2017-2023 THL A29 Limited, a Tencent company. All rights reserved.
+Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
+You may obtain a copy of the License at https://opensource.org/licenses/MIT
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+"""
+from aidev_agent.services.command_handler import CommandHandler
+from django.utils.translation import gettext as _
+
+from ..constants import DBMAgentCode
+from .register import command
+
+
+@command
+class CheckRedisClusterCommand(CommandHandler):
+    name = _("查询单据运行期间这批Redis集群的是否存在风险性")
+    command = "check_redis_cluster_operating_status"
+    agent_code = DBMAgentCode.REDIS_TASK_GUARDIAN
+
+    def get_template(self) -> str:
+        return """
+        查询单据运行期间这批Redis集群的是否存在风险性:
+        业务ID：{{ bk_biz_id }}
+        集群主域名列表：{{ cluster_domains }}
+        查询起始时间点：{{ start_time }}
+        查询截止时间点：{{ end_time }}
+        """
+
+
+@command
+class GenRedisUpdReporterCommand(CommandHandler):
+    name = _("生成Redis变更报告")
+    command = "gen_redis_updater_report"
+    agent_code = DBMAgentCode.REDIS_TASK_GUARDIAN
+
+    def get_init(self) -> str:
+        return "查询{{ cluster_domain }}集群部署情况"
+
+    def get_report(self) -> str:
+        return "再次查询{{ cluster_domain }}集群部署情况，并与上一次查询结果进行对比，生成变更报告"

dbm-ui/backend/dbm_aiagent/agent/constants.py

@@ -24,3 +24,8 @@ class DBMAgentCode(StrStructuredEnum):
     MYSQL_TASK_GUARDIAN = EnumField("ai-mysql-taskgd", _("MySQL单据值守智能体"))
     SQLSERVER_TASK_GUARDIAN = EnumField("ai-sqlsvr-tgd", _("SQLServer单据值守智能体"))
     REDIS_MEMORY_ANALYSIS = EnumField("ai-redis-memchk", _("Redis内存增长分析智能体"))
+    REDIS_HELPER = EnumField("ai-tendis-agent", _("Redis智能助手"))
+    REDIS_TASK_GUARDIAN = EnumField("ai-redis-taskgd", _("Redis单据值守"))
+    REDIS_LOG_ANA = EnumField("ai-redis-logana", _("Redis日志解析"))
+    REDIS_TOOLS = EnumField("ai-redis-wb", _("Redis工具箱"))
+    REDIS_METRICS = EnumField("ai-tendismetrics", _("Redis指标助手"))

dbm-ui/backend/dbm_aiagent/agent/handlers.py

@@ -55,10 +55,12 @@ def create_chat_completion(cls, session_code, session_content_id):
         return result
 
     @classmethod
-    def ask_agent_with_content(cls, agent_code: DBMAgentCode, content: str, username=DEFAULT_USERNAME):
+    def ask_agent_with_content(
+        cls, agent_code: DBMAgentCode, content: str, username=DEFAULT_USERNAME, session_code=None
+    ):
         """根据agent直接内容询问agent"""
         # 创建临时会话
-        session_code = cls.create_temporary_session(username)
+        session_code = session_code or cls.create_temporary_session(username)
 
         # 创建会话内容
         # 主智能体直接询问，子智能体走快捷指令切换询问
@@ -73,6 +75,15 @@ def ask_agent_with_content(cls, agent_code: DBMAgentCode, content: str, username
         ai_response = cls.create_chat_completion(session_code, session_content_id)
         return ai_response["choices"][0]["delta"]["content"]
 
+    @classmethod
+    def ask_agent_with_content_in_session(
+        cls, agent_code: DBMAgentCode, content: str, username=DEFAULT_USERNAME, session_code=None
+    ):
+        """根据agent直接内容询问agent, 连续对话"""
+        session_code = session_code or cls.create_temporary_session(username)
+        ai_response = cls.ask_agent_with_content(agent_code, content, username, session_code)
+        return ai_response, session_code
+
     @classmethod
     def ask_agent_with_command(cls, command: str, command_params: dict, username=DEFAULT_USERNAME):
         """根据快捷指令询问agent"""

dbm-ui/backend/flow/engine/bamboo/scene/redis/redis_cluster_scene_cmr.py

@@ -220,9 +220,10 @@ def __init_builder(self, operate_name: str):
     # 这里整理替换所需要的参数
     def complete_machine_replace(self):
         redis_pipeline, act_kwargs = self.__init_builder(_("REDIS-整机替换"))
-        sub_pipelines = []
+        sub_pipelines, cluster_ids = [], []
         for cluster_replacement in self.data["infos"]:
             for cluster_id in cluster_replacement["cluster_ids"]:
+                cluster_ids.append(int(cluster_id))
                 cluster_kwargs = deepcopy(act_kwargs)
                 cluster_info = self.get_cluster_info(self.data["bk_biz_id"], cluster_id)
                 sync_type = SyncType.SYNC_MMS.value  # ssd sync from master
@@ -244,7 +245,8 @@ def complete_machine_replace(self):
                     sub_pipelines.append(sub_pipeline)
         if len(sub_pipelines) > 0:
             redis_pipeline.add_parallel_sub_pipeline(sub_flow_list=sub_pipelines)
-        return redis_pipeline.run_pipeline()
+        # return redis_pipeline.run_pipeline()
+        return redis_pipeline.run_pipeline_with_sidecar(check_ai_monitor_cluster_list=cluster_ids)
 
     # 组装&控制 集群替换流程
     def generate_cluster_replacement(self, flow_data, act_kwargs, replacement_param):

dbm-ui/backend/flow/plugins/components/collections/common/check_cluster_alarm_for_ai.py

@@ -31,6 +31,7 @@
     DBType.MySQL: commands.CheckMysqlClusterCommand,
     DBType.TenDBCluster: commands.CheckMysqlClusterCommand,
     DBType.Sqlserver: commands.CheckSQLServerClusterCommand,
+    DBType.Redis: commands.CheckRedisClusterCommand,
 }