261 lines
11 KiB
Python
261 lines
11 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Примеры использования новой DataSource API для управления источниками данных.
|
|
"""
|
|
|
|
import requests
|
|
import json
|
|
from typing import Dict, List
|
|
|
|
BASE_URL = "http://localhost:8000/api/v1"
|
|
|
|
# Цвета для вывода
|
|
GREEN = '\033[92m'
|
|
RED = '\033[91m'
|
|
BLUE = '\033[94m'
|
|
YELLOW = '\033[93m'
|
|
RESET = '\033[0m'
|
|
|
|
|
|
def print_section(title: str):
|
|
"""Красиво вывести заголовок секции"""
|
|
print(f"\n{BLUE}{'='*60}{RESET}")
|
|
print(f"{BLUE}{title:^60}{RESET}")
|
|
print(f"{BLUE}{'='*60}{RESET}\n")
|
|
|
|
|
|
def print_response(response: requests.Response, title: str = "Response"):
|
|
"""Вывести HTTP response в красивом формате"""
|
|
print(f"{YELLOW}{title}:{RESET}")
|
|
try:
|
|
print(json.dumps(response.json(), indent=2))
|
|
except:
|
|
print(response.text)
|
|
print()
|
|
|
|
|
|
def example_1_create_data_sources():
|
|
"""Пример 1: Создание источников данных"""
|
|
print_section("Пример 1: Создание источников данных")
|
|
|
|
# 1. Создать MSSQL источник
|
|
print(f"{BLUE}1. Создание MSSQL источника...{RESET}")
|
|
mssql_source = {
|
|
"name": "MSSQL Production",
|
|
"source_type": "mssql",
|
|
"host": "mssql-server.example.com",
|
|
"port": 1433,
|
|
"database": "SalesDatabase",
|
|
"username": "sa",
|
|
"password": "StrongPassword123!",
|
|
"default_schema": "dbo",
|
|
"description": "Production MSSQL server for sales data"
|
|
}
|
|
|
|
response = requests.post(f"{BASE_URL}/data-sources", json=mssql_source)
|
|
print_response(response, "Created MSSQL Source")
|
|
mssql_id = response.json()['id'] if response.status_code == 200 else None
|
|
|
|
# 2. Создать PostgreSQL источник
|
|
print(f"{BLUE}2. Создание PostgreSQL источника...{RESET}")
|
|
pgsql_source = {
|
|
"name": "PostgreSQL Analytics",
|
|
"source_type": "pgsql",
|
|
"host": "postgres-analytics.example.com",
|
|
"port": 5432,
|
|
"database": "analytics_db",
|
|
"username": "analyst",
|
|
"password": "AnalyticsPass456!",
|
|
"default_schema": "public",
|
|
"description": "Analytics database for reporting"
|
|
}
|
|
|
|
response = requests.post(f"{BASE_URL}/data-sources", json=pgsql_source)
|
|
print_response(response, "Created PostgreSQL Source")
|
|
pgsql_id = response.json()['id'] if response.status_code == 200 else None
|
|
|
|
return mssql_id, pgsql_id
|
|
|
|
|
|
def example_2_list_and_test_sources(mssql_id: int, pgsql_id: int):
|
|
"""Пример 2: Получить список источников и протестировать подключения"""
|
|
print_section("Пример 2: Список источников и тестирование подключений")
|
|
|
|
# 1. Получить полный список
|
|
print(f"{BLUE}1. Получение списка всех источников...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/data-sources")
|
|
print_response(response, "All Data Sources")
|
|
|
|
# 2. Получить только активные
|
|
print(f"{BLUE}2. Получение списка активных источников...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/data-sources?active_only=true")
|
|
print_response(response, "Active Data Sources")
|
|
|
|
# 3. Получить конкретный источник
|
|
if mssql_id:
|
|
print(f"{BLUE}3. Получение конкретного источника (ID: {mssql_id})...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/data-sources/{mssql_id}")
|
|
print_response(response, f"Data Source {mssql_id}")
|
|
|
|
# 4. Протестировать подключение
|
|
if mssql_id:
|
|
print(f"{BLUE}4. Тестирование подключения к MSSQL (ID: {mssql_id})...{RESET}")
|
|
response = requests.post(f"{BASE_URL}/data-sources/{mssql_id}/test")
|
|
print_response(response, "MSSQL Connection Test Result")
|
|
|
|
|
|
def example_3_create_migrations(mssql_id: int, pgsql_id: int):
|
|
"""Пример 3: Создание миграций (расписание репликации)"""
|
|
print_section("Пример 3: Создание миграций таблиц")
|
|
|
|
# 1. Создать миграцию для таблицы Orders из MSSQL
|
|
print(f"{BLUE}1. Создание миграции таблицы Orders из MSSQL...{RESET}")
|
|
migration_1 = {
|
|
"table_name": "Orders",
|
|
"source_id": mssql_id,
|
|
"source_schema": "dbo",
|
|
"target_schema": "public",
|
|
"cron_schedule": "0 2 * * *" # Каждый день в 2:00 AM
|
|
}
|
|
|
|
response = requests.post(f"{BASE_URL}/migration-tables", json=migration_1)
|
|
print_response(response, "Created Migration for Orders")
|
|
migration_1_id = response.json()['id'] if response.status_code == 200 else None
|
|
|
|
# 2. Создать миграцию для таблицы Products из PostgreSQL
|
|
if pgsql_id:
|
|
print(f"{BLUE}2. Создание миграции таблицы Products из PostgreSQL...{RESET}")
|
|
migration_2 = {
|
|
"table_name": "Products",
|
|
"source_id": pgsql_id,
|
|
"source_schema": "public",
|
|
"target_schema": "public",
|
|
"cron_schedule": "0 4 * * *" # Каждый день в 4:00 AM
|
|
}
|
|
|
|
response = requests.post(f"{BASE_URL}/migration-tables", json=migration_2)
|
|
print_response(response, "Created Migration for Products")
|
|
|
|
return migration_1_id
|
|
|
|
|
|
def example_4_list_migrations(migration_1_id: int):
|
|
"""Пример 4: Получить список миграций"""
|
|
print_section("Пример 4: Получение списка миграций")
|
|
|
|
# 1. Получить все миграции
|
|
print(f"{BLUE}1. Получение всех миграций...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/migration-tables")
|
|
print_response(response, "All Migrations")
|
|
|
|
# 2. Получить только активные
|
|
print(f"{BLUE}2. Получение активных миграций...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/migration-tables?active_only=true")
|
|
print_response(response, "Active Migrations")
|
|
|
|
# 3. Получить конкретную миграцию
|
|
if migration_1_id:
|
|
print(f"{BLUE}3. Получение конкретной миграции (ID: {migration_1_id})...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/migration-tables/{migration_1_id}")
|
|
print_response(response, f"Migration {migration_1_id}")
|
|
|
|
|
|
def example_5_update_data_source(mssql_id: int):
|
|
"""Пример 5: Обновить источник данных"""
|
|
print_section("Пример 5: Обновление источника данных")
|
|
|
|
if not mssql_id:
|
|
print(f"{RED}No MSSQL source ID available{RESET}")
|
|
return
|
|
|
|
print(f"{BLUE}Обновление узла MSSQL источника...{RESET}")
|
|
update_data = {
|
|
"host": "new-mssql-server.example.com",
|
|
"port": 1433
|
|
}
|
|
|
|
response = requests.put(f"{BASE_URL}/data-sources/{mssql_id}", json=update_data)
|
|
print_response(response, "Updated Data Source")
|
|
|
|
|
|
def example_6_update_migration(migration_1_id: int):
|
|
"""Пример 6: Обновить миграцию"""
|
|
print_section("Пример 6: Обновление миграции")
|
|
|
|
if not migration_1_id:
|
|
print(f"{RED}No migration ID available{RESET}")
|
|
return
|
|
|
|
print(f"{BLUE}Обновление расписания миграции...{RESET}")
|
|
update_data = {
|
|
"cron_schedule": "0 3 * * *" # Изменить на 3:00 AM
|
|
}
|
|
|
|
response = requests.put(f"{BASE_URL}/migration-tables/{migration_1_id}", json=update_data)
|
|
print_response(response, "Updated Migration")
|
|
|
|
|
|
def example_7_check_health():
|
|
"""Пример 7: Проверить здоровье сервиса"""
|
|
print_section("Пример 7: Проверка здоровья сервиса")
|
|
|
|
response = requests.get(f"{BASE_URL}/health")
|
|
print_response(response, "Service Health Check")
|
|
|
|
|
|
def example_8_get_replication_jobs():
|
|
"""Пример 8: Получить историю задач репликации"""
|
|
print_section("Пример 8: История задач репликации")
|
|
|
|
print(f"{BLUE}1. Получение последних 10 задач репликации...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/replication-jobs?limit=10&offset=0")
|
|
print_response(response, "Recent Replication Jobs")
|
|
|
|
print(f"{BLUE}2. Получение только успешных задач...{RESET}")
|
|
response = requests.get(f"{BASE_URL}/replication-jobs?status=success&limit=10")
|
|
print_response(response, "Successful Replication Jobs")
|
|
|
|
|
|
def main():
|
|
"""Главная функция - запустить все примеры"""
|
|
|
|
print(f"\n{GREEN}{'='*60}{RESET}")
|
|
print(f"{GREEN}{'Data Replication Service - API Examples':^60}{RESET}")
|
|
print(f"{GREEN}{'='*60}{RESET}\n")
|
|
|
|
try:
|
|
# Проверить подключение к сервису
|
|
response = requests.get(f"{BASE_URL.replace('/api/v1', '')}/")
|
|
if response.status_code != 200:
|
|
print(f"{RED}✗ Error: Cannot connect to service at {BASE_URL}{RESET}")
|
|
print(f"{YELLOW}Make sure the service is running on http://localhost:8000{RESET}")
|
|
return
|
|
|
|
print(f"{GREEN}✓ Connected to Data Replication Service{RESET}\n")
|
|
|
|
# Запустить примеры
|
|
mssql_id, pgsql_id = example_1_create_data_sources()
|
|
example_2_list_and_test_sources(mssql_id, pgsql_id)
|
|
migration_1_id = example_3_create_migrations(mssql_id, pgsql_id)
|
|
example_4_list_migrations(migration_1_id)
|
|
example_5_update_data_source(mssql_id)
|
|
example_6_update_migration(migration_1_id)
|
|
example_7_check_health()
|
|
example_8_get_replication_jobs()
|
|
|
|
print(f"\n{GREEN}{'='*60}{RESET}")
|
|
print(f"{GREEN}{'✓ All examples completed successfully!':^60}{RESET}")
|
|
print(f"{GREEN}{'='*60}{RESET}\n")
|
|
|
|
print(f"{YELLOW}📚 Documentation: See DATA_SOURCES.md for detailed API reference{RESET}\n")
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
print(f"{RED}✗ Connection error: Cannot connect to {BASE_URL}{RESET}")
|
|
print(f"{YELLOW}Make sure the service is running: python main.py{RESET}")
|
|
except Exception as e:
|
|
print(f"{RED}✗ Error: {e}{RESET}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|