feat(dynamodb-mcp-server): Add MySQL integration to support DynamoDB Data Modeling and Migration planning

src/dynamodb-mcp-server/README.md

@@ -2,13 +2,16 @@
 
 The official MCP Server for interacting with AWS DynamoDB
 
-This comprehensive server provides both operational DynamoDB management and expert design guidance, featuring 30+ operational tools for managing DynamoDB tables, items, indexes, backups, and more, expert data modeling guidance.
+This comprehensive server provides both operational DynamoDB management and expert design guidance, featuring 30+ operational tools for managing DynamoDB tables, items, indexes, backups, and more, expert data modeling guidance and also MySQL query capabilities for database analysis and migration planning.
 
 ## Available MCP Tools
 
 ### Design & Modeling
 - `dynamodb_data_modeling` - Retrieves the complete DynamoDB Data Modeling Expert prompt
 
+### MySQL Integration
+- `mysql_run_query` - Run SQL queries against MySQL databases for analysis and migration planning
+
 ### Table Operations
 - `create_table` - Creates a new DynamoDB table with optional secondary indexes
 - `delete_table` - Deletes a table and all of its items
@@ -61,6 +64,22 @@ To use these tools, ensure you have proper AWS credentials configured with appro
 
 All tools support an optional `region_name` parameter to specify which AWS region to operate in. If not provided, it will use the AWS_REGION environment variable or default to 'us-west-2'.
 
+## MySQL Integration
+
+The DynamoDB MCP server includes MySQL integration to run mysql queries within DynamoDB MCP server. When configured, it adds a `mysql_run_query` tool for database analysis and migration planning.
+
+### MySQL Environment Variables
+
+Add these environment variables to DynamoDB MCP Server configuration to enable MySQL integration:
+
+- `MYSQL_CLUSTER_ARN`: RDS / Aurora MySQL cluster ARN
+- `MYSQL_SECRET_ARN`: AWS Secrets Manager secret ARN with database credentials
+- `MYSQL_DATABASE`: Database name to connect to
+- `MYSQL_AWS_REGION`: AWS region for MySQL cluster (default: "us-west-2")
+- `MYSQL_READONLY`: optional (default: "true")
+
+**Note**: MySQL integration tool is available only when these variables are set.
+
 ## Prerequisites
 
 1. Install `uv` from [Astral](https://docs.astral.sh/uv/getting-started/installation/) or the [GitHub README](https://github.com/astral-sh/uv#installation)
@@ -86,7 +105,12 @@ Add the MCP to your favorite agentic tools. (e.g. for Amazon Q Developer CLI MCP
         "DDB-MCP-READONLY": "true",
         "AWS_PROFILE": "default",
         "AWS_REGION": "us-west-2",
-        "FASTMCP_LOG_LEVEL": "ERROR"
+        "FASTMCP_LOG_LEVEL": "ERROR",
+        "MYSQL_CLUSTER_ARN": "[your data]",
+        "MYSQL_SECRET_ARN": "[your data]",
+        "MYSQL_DATABASE": "[your data]",
+        "MYSQL_AWS_REGION": "us-west-2",
+        "MYSQL_READONLY": "true"
       },
       "disabled": false,
       "autoApprove": []
@@ -116,7 +140,12 @@ For Windows users, the MCP server configuration format is slightly different:
       "env": {
         "FASTMCP_LOG_LEVEL": "ERROR",
         "AWS_PROFILE": "your-aws-profile",
-        "AWS_REGION": "us-east-1"
+        "AWS_REGION": "us-east-1",,
+        "MYSQL_CLUSTER_ARN": "[your data]",
+        "MYSQL_SECRET_ARN": "[your data]",
+        "MYSQL_DATABASE": "[your data]",
+        "MYSQL_AWS_REGION": "us-west-2",
+        "MYSQL_READONLY": "true"
       }
     }
   }

src/dynamodb-mcp-server/awslabs/dynamodb_mcp_server/server.py

@@ -50,14 +50,19 @@
 from mcp.server.fastmcp import FastMCP
 from pathlib import Path
 from pydantic import Field
-from typing import Any, Dict, List, Literal, Union
+from typing import Any, Dict, List, Literal, Union, Optional, Annotated
+
+from awslabs.mysql_mcp_server.server import DBConnectionSingleton, run_query as mysql_query
 
 
 # Define server instructions and dependencies
 SERVER_INSTRUCTIONS = """The official MCP Server for interacting with AWS DynamoDB
 
 This server provides comprehensive DynamoDB capabilities with over 30 operational tools for managing DynamoDB tables,
-items, indexes, backups, and more, plus expert data modeling guidance through DynamoDB data modeling expert prompt
+items, indexes, backups, and more, plus expert data modeling guidance through DynamoDB data modeling expert prompt.
+
+MySQL Integration: This server also includes MySQL query capabilities through the run_query tool, enabling database
+analysis and migration planning from MySQL to DynamoDB.
 
 IMPORTANT: DynamoDB Attribute Value Format
 -----------------------------------------
@@ -942,6 +947,39 @@ async def list_imports(
     }
 
 
+@app.tool()
+@handle_exceptions
+async def mysql_run_query(
+    sql: Annotated[str, Field(description='The SQL query to run')],
+    query_parameters: Annotated[
+        Optional[List[Dict[str, Any]]], Field(description='Parameters for the SQL query')
+    ] = None,
+) -> list[dict]:
+    """Run a SQL query against a MySQL database for analysis and migration planning."""
+
+    if all([os.getenv('MYSQL_CLUSTER_ARN'), os.getenv('MYSQL_SECRET_ARN'), os.getenv('MYSQL_DATABASE')]):
+
+        cluster_arn = os.getenv('MYSQL_CLUSTER_ARN')
+        secret_arn = os.getenv('MYSQL_SECRET_ARN')
+        database = os.getenv('MYSQL_DATABASE')
+        region = os.getenv('AWS_REGION', 'us-west-2')
+        readonly = os.getenv('MYSQL_READONLY', 'true').lower() == 'true'
+
+        try:
+            DBConnectionSingleton.initialize(cluster_arn, secret_arn, database, region, readonly)
+        except Exception as e:
+            return [{'error': f'MySQL initialization failed: {str(e)}'}]
+    else:
+        return [{'error': 'MySQL configuration not properly configured or available'}]
+
+    class DummyContext:
+        async def error(self, message): pass
+
+    try:
+        return await mysql_query(sql, DummyContext(), None, query_parameters)
+    except Exception as e:
+        return [{'error': f'MySQL query failed: {str(e)}'}]
+
 def main():
     """Main entry point for the MCP server application."""
     app.run()

src/dynamodb-mcp-server/pyproject.toml

@@ -11,7 +11,8 @@ dependencies = [
     "pydantic==2.11.7",
     "typing-extensions==4.14.1",
     "strands-agents>=1.5.0",
-    "dspy-ai>=2.6.27"
+    "dspy-ai>=2.6.27",
+    "awslabs-mysql-mcp-server>=1.0.5",
 ]
 license = {text = "Apache-2.0"}
 license-files = ["LICENSE", "NOTICE" ]

src/dynamodb-mcp-server/tests/test_mysql_integration.py

@@ -0,0 +1,44 @@
+import os
+import pytest
+import pytest_asyncio
+from awslabs.dynamodb_mcp_server.server import mysql_run_query
+
+
+class TestMySQLIntegration:
+    """Test MySQL integration functionality."""
+
+    @pytest.mark.asyncio
+    async def test_mysql_connection_and_query(self, monkeypatch):
+        """Test MySQL connection initialization and query execution."""
+        # Set environment variables using monkeypatch
+        monkeypatch.setenv('MYSQL_CLUSTER_ARN', 'arn:aws:rds:us-west-2:123456789012:cluster:test-cluster')
+        monkeypatch.setenv('MYSQL_SECRET_ARN', 'arn:aws:secretsmanager:us-west-2:123456789012:secret:test-secret')
+        monkeypatch.setenv('MYSQL_DATABASE', 'test_database')
+        monkeypatch.setenv('MYSQL_AWS_REGION', 'us-west-2')
+        monkeypatch.setenv('MYSQL_READONLY', 'true')
+
+        # Mock the MySQL functions using monkeypatch
+        def mock_initialize(*args, **kwargs):
+            return True
+
+        async def mock_query(*args, **kwargs):
+            return [{'id': 1, 'name': 'test'}]
+
+        monkeypatch.setattr('awslabs.dynamodb_mcp_server.server.DBConnectionSingleton.initialize', mock_initialize)
+        monkeypatch.setattr('awslabs.dynamodb_mcp_server.server.mysql_query', mock_query)
+
+        result = await mysql_run_query("SELECT * FROM users")
+
+        assert result == [{'id': 1, 'name': 'test'}]
+
+    @pytest.mark.asyncio
+    async def test_mysql_missing_config(self, monkeypatch):
+        """Test MySQL query when configuration is missing."""
+        # Clear MySQL env vars using monkeypatch
+        monkeypatch.delenv('MYSQL_CLUSTER_ARN', raising=False)
+        monkeypatch.delenv('MYSQL_SECRET_ARN', raising=False)
+        monkeypatch.delenv('MYSQL_DATABASE', raising=False)
+
+        result = await mysql_run_query("SELECT * FROM users")
+        assert 'error' in result[0]
+        assert 'MySQL configuration not properly configured' in result[0]['error']