support compress performance items.

Author: wenfeng.wf

src/alibabacloud_rds_openapi_mcp_server/server.py

@@ -78,10 +78,25 @@ async def describe_db_instance_performance(region_id: str,
  region_id: db instance region(e.g. cn-hangzhou)
  db_instance_id: db instance id(e.g. rm-xxx)
  db_type: the db instance database type(e.g. mysql,pgsql,sqlserver)
- perf_keys: Performance Key (e.g. ["MemCpuUsage", "QPSTPS", "Sessions", "COMDML", "RowDML"])
+ perf_keys: Performance Key (e.g. ["MemCpuUsage", "QPSTPS", "Sessions", "COMDML", "RowDML", "ThreadStatus", "MBPS", "DetailedSpaceUsage"])
  start_time: start time(e.g. 2023-01-01 00:00)
  end_time: end time(e.g. 2023-01-01 00:00)
  """
+ def _compress_performance(performance_value, max_items=10):
+ if len(performance_value) > max_items:
+ result = []
+ offset = len(performance_value) / 10
+ for i in range(0, len(performance_value), int(offset)):
+ _item = None
+ for j in range(i, min(i + int(offset), len(performance_value))):
+ if _item is None or sum([float(v) for v in performance_value[j].value.split('&')]) > sum([float(v) for v in _item.value.split('&')]):
+ _item = performance_value[j]
+ else:
+ result.append(_item)
+ return result
+ else:
+ return performance_value
+
  try:
  start_time = transform_to_datetime(start_time)
  end_time = transform_to_datetime(end_time)
@@ -98,7 +113,7 @@ async def describe_db_instance_performance(region_id: str,
  response = client.describe_dbinstance_performance(request)
  responses = []
  for perf_key in response.body.performance_keys.performance_key:
- perf_key_info = f"""Key={perf_key.key}; Unit={perf_key.unit}; ValueFormat={perf_key.value_format}; Values={compress_json_array([item.to_map() for item in perf_key.values.performance_value])}"""
+ perf_key_info = f"""Key={perf_key.key}; Unit={perf_key.unit}; ValueFormat={perf_key.value_format}; Values={compress_json_array([item.to_map() for item in _compress_performance(perf_key.values.performance_value)])}"""
  responses.append(perf_key_info)
  return responses
  except Exception as e:

src/alibabacloud_rds_openapi_mcp_server/utils.py

@@ -13,15 +13,21 @@
  "Sessions": ["MySQL_Sessions"],
  "COMDML": ["MySQL_COMDML"],
  "RowDML": ["MySQL_RowDML"],
- "SpaceUsage": ["MySQL_DetailedSpaceUsage"]
+ "SpaceUsage": ["MySQL_DetailedSpaceUsage"],
+ "ThreadStatus": ["MySQL_ThreadStatus"],
+ "MBPS": ["MySQL_MBPS"],
+ "DetailedSpaceUsage": ["MySQL_DetailedSpaceUsage"]
  },
  "pgsql": {
  "MemCpuUsage": ["MemoryUsage", "CpuUsage"],
  "QPSTPS": ["PolarDBQPSTPS"],
  "Sessions": ["PgSQL_Session"],
  "COMDML": ["PgSQL_COMDML"],
  "RowDML": ["PolarDBRowDML"],
- "SpaceUsage": ["PgSQL_SpaceUsage"]
+ "SpaceUsage": ["PgSQL_SpaceUsage"],
+ "ThreadStatus": [],
+ "MBPS": [],
+ "DetailedSpaceUsage": ["SQLServer_DetailedSpaceUsage"]
  },
  "sqlserver": {
  "MemCpuUsage": ["SQLServer_CPUUsage"],
@@ -30,6 +36,9 @@
  "COMDML": [],
  "RowDML": [],
  "SpaceUsage": ["SQLServer_DetailedSpaceUsage"],
+ "ThreadStatus": [],
+ "MBPS": [],
+ "DetailedSpaceUsage": ["PgSQL_SpaceUsage"]
  }
 
 }