feat(other): 新增根ip查询集群信息mcp-tool #16445

Author: ymakedaq

.gitignore

@@ -22,9 +22,11 @@ package-lock.json
 .cursorrules
 .cursor/
 # Claude
+CLAUDE.md
 .claude/
 .history/
 dbm-services/.history/
 dbm-ui/.history/
 # trellis
-.trellis
+.trellis
+

dbm-ui/backend/dbm_aiagent/mcp_tools/mysql/impl/query_cluster_by_ip.py

@@ -0,0 +1,70 @@
+# -*- 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 typing import List
+
+from backend.db_meta.enums import ClusterType
+from backend.db_meta.models import Machine, ProxyInstance, StorageInstance
+
+
+def query_cluster_by_ip(ip_list: List[str], bk_cloud_id: int = 0) -> List[dict]:
+    """根据 IP 列表查询关联的集群信息"""
+    machines = Machine.objects.filter(ip__in=ip_list, bk_cloud_id=bk_cloud_id)
+    machine_map = {m.ip: m for m in machines}
+
+    results = []
+    seen = set()
+
+    for ip in ip_list:
+        machine = machine_map.get(ip)
+        if not machine:
+            continue
+
+        for inst in StorageInstance.objects.filter(machine=machine).prefetch_related("cluster"):
+            for cluster in inst.cluster.all():
+                key = (ip, cluster.id)
+                if key in seen:
+                    continue
+                seen.add(key)
+                results.append(
+                    {
+                        "ip": ip,
+                        "cluster_id": cluster.id,
+                        "cluster_domain": cluster.immute_domain,
+                        "cluster_type": cluster.cluster_type,
+                        "db_type": ClusterType.cluster_type_to_db_type(cluster.cluster_type),
+                        "machine_type": machine.machine_type,
+                        "bk_sub_zone": machine.bk_sub_zone,
+                        "bk_sub_zone_id": machine.bk_sub_zone_id,
+                        "bk_city": machine.bk_city.bk_idc_city_name,
+                    }
+                )
+
+        for inst in ProxyInstance.objects.filter(machine=machine).prefetch_related("cluster"):
+            for cluster in inst.cluster.all():
+                key = (ip, cluster.id)
+                if key in seen:
+                    continue
+                seen.add(key)
+                results.append(
+                    {
+                        "ip": ip,
+                        "cluster_id": cluster.id,
+                        "cluster_domain": cluster.immute_domain,
+                        "cluster_type": cluster.cluster_type,
+                        "db_type": ClusterType.cluster_type_to_db_type(cluster.cluster_type),
+                        "machine_type": machine.machine_type,
+                        "bk_sub_zone": machine.bk_sub_zone,
+                        "bk_sub_zone_id": machine.bk_sub_zone_id,
+                        "bk_city": machine.bk_city.bk_idc_city_name,
+                    }
+                )
+
+    return results

dbm-ui/backend/dbm_aiagent/mcp_tools/mysql/serializers/query_cluster_by_ip.py

@@ -0,0 +1,33 @@
+# -*- 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 django.utils.translation import gettext_lazy as _
+from rest_framework import serializers
+
+
+class QueryClusterByIpInputSerializer(serializers.Serializer):
+    ip_list = serializers.ListField(child=serializers.CharField(), help_text=_("主机 IP 列表"))
+    bk_cloud_id = serializers.IntegerField(help_text=_("云区域 ID"), required=False, default=0)
+
+
+class ClusterInfoByIpSerializer(serializers.Serializer):
+    cluster_id = serializers.IntegerField(help_text=_("集群 ID"))
+    cluster_domain = serializers.CharField(help_text=_("集群域名"))
+    cluster_type = serializers.CharField(help_text=_("集群类型"))
+    db_type = serializers.CharField(help_text=_("主机的 DB 类型"))
+    machine_type = serializers.CharField(help_text=_("主机的机器类型"))
+    ip = serializers.CharField(help_text=_("主机 IP"))
+    bk_sub_zone = serializers.CharField(help_text=_("子 Zone"), allow_null=True)
+    bk_sub_zone_id = serializers.IntegerField(help_text=_("子 Zone ID"))
+    bk_city = serializers.CharField(help_text=_("IDC 城市名"))
+
+
+class QueryClusterByIpOutputSerializer(serializers.Serializer):
+    clusters = serializers.ListSerializer(child=ClusterInfoByIpSerializer(), help_text=_("集群信息列表"))

dbm-ui/backend/dbm_aiagent/mcp_tools/mysql/views/mysql_query_mcp.py

@@ -25,6 +25,7 @@
 )
 from backend.dbm_aiagent.mcp_tools.mysql.impl.cluster_topo import mysql_cluster_topo
 from backend.dbm_aiagent.mcp_tools.mysql.impl.explain_sql import explain_sql
+from backend.dbm_aiagent.mcp_tools.mysql.impl.query_cluster_by_ip import query_cluster_by_ip
 from backend.dbm_aiagent.mcp_tools.mysql.impl.show_create_table import show_create_table
 from backend.dbm_aiagent.mcp_tools.mysql.impl.show_priv_template import show_biz_mysql_privilege_template
 from backend.dbm_aiagent.mcp_tools.mysql.impl.show_processlist import show_cluster_processlist_summary
@@ -38,6 +39,10 @@
     ExplainSQLInputSerializer,
     ExplainSQLOutputSerializer,
 )
+from backend.dbm_aiagent.mcp_tools.mysql.serializers.query_cluster_by_ip import (
+    QueryClusterByIpInputSerializer,
+    QueryClusterByIpOutputSerializer,
+)
 from backend.dbm_aiagent.mcp_tools.mysql.serializers.show_create_table import (
     ShowCreateTableInputSerializer,
     ShowCreateTableOutputSerializer,
@@ -62,7 +67,7 @@
 from backend.dbm_aiagent.mcp_tools.mysql.serializers.usefully_choices import MySQLProcessListInstanceGroupType
 from backend.dbm_aiagent.mcp_tools.views import McpToolsViewSet
 from backend.iam_app.handlers.drf_perm.base import DBManagePermission
-from backend.iam_app.handlers.drf_perm.mcp import McpClusterManagePermission
+from backend.iam_app.handlers.drf_perm.mcp import McpClusterManagePermission, McpIsDbaPermission
 
 logger = logging.getLogger("root")
 
@@ -275,6 +280,21 @@ def show_biz_mysql_privilege_template(self, request, *args, **kwargs):
             }
         )
 
+    @mcp_tools_api_decorator(
+        description=str(_("根据 IP 列表查询所属集群信息，返回集群域名、集群ID和主机DB类型，仅 DBA 可调用")),
+        request_slz=QueryClusterByIpInputSerializer,
+        response_slz=QueryClusterByIpOutputSerializer,
+        permission_classes=[McpIsDbaPermission],
+        tags=[DBMMCPTags.READ],
+        mcp=[DBMMcpTools.MYSQL_QUERY],
+        name_prefix="mysql_query",
+    )
+    def query_cluster_by_ip(self, request, *args, **kwargs):
+        ip_list = self.get_param("ip_list")
+        bk_cloud_id = self.get_param("bk_cloud_id")
+
+        return Response({"clusters": query_cluster_by_ip(ip_list=ip_list, bk_cloud_id=bk_cloud_id)})
+
 
 def _validate_and_get_machine(bk_cloud_id: int | None, address: str) -> Machine:
     """验证并获取机器对象"""