Trino MCP Server

Model Context Protocol (MCP) сервер для анализа схем базы данных Trino

Этот проект представляет собой MCP сервер, который предоставляет мощные инструменты для анализа структуры баз данных Trino. Сервер принимает параметры подключения динамически и поддерживает работу с множественными подключениями одновременно.

🚀 Основные возможности

Базовые операции

Проверка подключения к Trino серверам
Навигация по структуре БД: каталоги → схемы → таблицы
Описание структуры таблиц с детальной информацией о колонках
Безопасное выполнение SQL запросов с ограничениями
Анализ запросов через EXPLAIN без выполнения

Работа с DDL

Валидация DDL выражений перед выполнением
Анализ зависимостей между объектами базы данных
Безопасное выполнение DDL с предварительной проверкой
Обнаружение конфликтов имен объектов
Построение графа зависимостей для правильного порядка создания

Документация и анализ

Автогенерация документации схемы базы данных
Анализ качества DDL с рекомендациями
Обнаружение потенциальных проблем в структуре
Статистика использования подключений

🛠 Установка

Требования

Python 3.11+
Poetry для управления зависимостями
Доступ к Trino серверу

Установка зависимостей

# Клонирование репозитория
git clone <repository-url>
cd VkHackMCP

# Установка зависимостей через Poetry
poetry install

# Или через pip (если Poetry не используется)
pip install -r requirements.txt

Настройка окружения

Создайте файл .env на основе .env.example:

cp .env.example .env

Пример конфигурации .env:

# Настройки приложения
APP_NAME=trino-mcp
LOG_LEVEL=INFO

🎯 Запуск

Разработка

# Запуск MCP сервера для разработки
make run-server

# Или напрямую через Python
poetry run python -m src.api.server

Форматирование и проверка кода

# Форматирование кода
make format

# Проверка стиля кода
make lint

# Все проверки сразу
make check-all

Docker

# Сборка образа
docker build -t trino-mcp .

# Запуск контейнера
docker run -it trino-mcp

📖 Использование

MCP сервер предоставляет следующие инструменты:

Базовые инструменты

`connection_status`

Проверяет статус подключения к Trino серверу.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst"
}

`list_catalogs`

Возвращает список всех доступных каталогов.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst"
}

`list_schemas`

Получает список схем в указанном каталоге.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst",
  "catalog": "hive"
}

`describe_table`

Описывает структуру таблицы с детальной информацией о колонках.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst",
  "table": "users",
  "schema": "default",
  "catalog": "hive"
}

Инструменты для работы с DDL

`validate_ddl_statements`

Анализирует список DDL выражений без их выполнения.

{
  "ddl_list": [
    "CREATE TABLE users (id bigint, name varchar(255))",
    "CREATE VIEW active_users AS SELECT * FROM users WHERE active = true"
  ]
}

`execute_ddl_statements`

Выполняет DDL с предварительной валидацией.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst",
  "ddl_list": ["CREATE TABLE test (id bigint)"],
  "catalog": "hive",
  "schema": "default",
  "validate_first": true
}

`analyze_schema_dependencies`

Анализирует зависимости между объектами и рекомендует порядок создания.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst",
  "ddl_list": [
    "CREATE VIEW v1 AS SELECT * FROM table1",
    "CREATE TABLE table1 (id bigint)"
  ],
  "catalog": "hive",
  "schema": "default"
}

Инструменты анализа и документации

`generate_schema_documentation`

Создает документацию для схемы базы данных.

{
  "jdbc_url": "jdbc:trino://host:443?user=analyst",
  "catalog": "hive",
  "schema": "default",
  "include_ddl": ["CREATE TABLE additional (id bigint)"]
}

Пример workflow

# 1. Проверка подключения
connection_status(jdbc_url="jdbc:trino://host:443?user=analyst")

# 2. Анализ существующей структуры
list_catalogs(jdbc_url="...")
list_schemas(jdbc_url="...", catalog="hive")
list_tables(jdbc_url="...", schema="default", catalog="hive")

# 3. Валидация новых DDL
validate_ddl_statements(ddl_list=["CREATE TABLE ...", "CREATE VIEW ..."])

# 4. Анализ зависимостей
analyze_schema_dependencies(jdbc_url="...", ddl_list=["..."])

# 5. Выполнение DDL
execute_ddl_statements(jdbc_url="...", ddl_list=["..."], validate_first=True)

# 6. Генерация документации
generate_schema_documentation(jdbc_url="...", catalog="hive", schema="default")

📊 Мониторинг

Статистика подключений

{
  "tool": "get_connection_stats"
}