add restart instance tool

Author: fdwills

README.md

@@ -102,11 +102,12 @@ Add the following configuration to the MCP client configuration file:
 * `describe_slow_log_records`: Query slow log records for an RDS instance.
 * `describe_vpcs`: Query VPC list.
 * `describe_vswitches`: Query VSwitch list.
-* `modify_security_ips`: Modify security ips
+* `modify_security_ips`: Modify RDS instance security IP whitelist.
 * `get_current_time`: Get the current time.
 * `modify_db_instance_description`: Modify RDS instance descriptions.
 * `modify_db_instance_spec`: Modify RDS instance specifications.
 * `modify_parameter`: Modify RDS instance parameters.
+* `restart_db_instance`: Restart an RDS instance.
 
 ### Resources
 None at this time

README_CN.md

@@ -104,6 +104,7 @@ git clone https://github.com/aliyun/alibabacloud-rds-openapi-mcp-server.git
 * `modify_db_instance_description`: 修改RDS实例描述
 * `modify_db_instance_spec`: 修改RDS实例规格
 * `modify_parameter`: 修改RDS实例参数
+* `restart_db_instance`: 重启RDS实例
 
 ### 资源
 当前暂无资源

src/alibabacloud_rds_openapi_mcp_server/server.py

@@ -1299,6 +1299,57 @@ async def modify_security_ips(
  raise OpenAPIError(f"modify rds instance security ips failed: {str(e)}")
 
 
+@mcp.tool()
+async def restart_db_instance(
+ region_id: str,
+ dbinstance_id: str,
+ effective_time: str = "Immediate",
+ switch_time: str = None,
+ client_token: str = None
+) -> Dict[str, Any]:
+ """Restart an RDS instance.
+
+ Args:
+ region_id (str): The region ID of the RDS instance.
+ dbinstance_id (str): The ID of the RDS instance.
+ effective_time (str, optional): When to restart the instance. Options:
+ - Immediate: Restart immediately
+ - MaintainTime: Restart during maintenance window
+ - ScheduleTime: Restart at specified time
+ Default: Immediate
+ switch_time (str, optional): The scheduled restart time in format: yyyy-MM-ddTHH:mm:ssZ (UTC time).
+ Required when effective_time is ScheduleTime.
+ client_token (str, optional): Idempotency token, max 64 ASCII characters.
+
+ Returns:
+ Dict[str, Any]: Response containing the request ID.
+ """
+ try:
+ # Initialize client
+ client = get_rds_client(region_id)
+
+ # Create request
+ request = rds_20140815_models.RestartDBInstanceRequest(
+ dbinstance_id=dbinstance_id
+ )
+
+ # Add optional parameters
+ if effective_time:
+ request.effective_time = effective_time
+ if switch_time:
+ request.switch_time = switch_time
+ if client_token:
+ request.client_token = client_token
+
+ # Make the API request
+ response = client.restart_dbinstance(request)
+ return response.body.to_map()
+
+ except Exception as e:
+ logger.error(f"Error occurred while restarting instance: {str(e)}")
+ raise OpenAPIError(f"Failed to restart RDS instance: {str(e)}")
+
+
 def main():
  mcp.run(transport=os.getenv('SERVER_TRANSPORT', 'stdio'))