348 lines
14 KiB
Bash
Executable File
348 lines
14 KiB
Bash
Executable File
#!/bin/bash
|
||
|
||
# Скрипт для тестирования baseline миграции на чистой БД
|
||
# Создает тестовую БД, применяет baseline, и сравнивает схему с production
|
||
|
||
set -e
|
||
|
||
# Цвета для вывода
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[0;34m'
|
||
NC='\033[0m' # No Color
|
||
|
||
# Получаем переменные окружения
|
||
DB_HOST=${DB_HOST:-localhost}
|
||
DB_PORT=${DB_PORT:-5432}
|
||
DB_USER=${DB_USER:-playeng}
|
||
DB_PASSWORD=${DB_PASSWORD:-playeng}
|
||
DB_NAME=${DB_NAME:-playeng}
|
||
|
||
TEST_DB_NAME="playeng_baseline_test_$$"
|
||
MIGRATIONS_PATH="migrations"
|
||
TMP_DIR=$(mktemp -d)
|
||
|
||
echo "=== Тестирование baseline миграции на чистой БД ==="
|
||
echo ""
|
||
|
||
# Добавляем ~/go/bin в PATH если migrate не найден
|
||
if ! command -v migrate &> /dev/null; then
|
||
export PATH="$HOME/go/bin:$PATH"
|
||
fi
|
||
|
||
# Проверяем наличие необходимых инструментов
|
||
if ! command -v migrate &> /dev/null; then
|
||
echo -e "${RED}Ошибка: migrate не найден. Установите golang-migrate:${NC}"
|
||
echo " brew install golang-migrate"
|
||
echo " или"
|
||
echo " go install -tags 'postgres' github.com/golang-migrate/migrate/v4/cmd/migrate@latest"
|
||
exit 1
|
||
fi
|
||
|
||
# Определяем способ выполнения PostgreSQL команд
|
||
PG_DUMP_CMD=""
|
||
PG_PSQL_CMD=""
|
||
POSTGRES_CONTAINER=""
|
||
if command -v pg_dump &> /dev/null; then
|
||
PG_DUMP_CMD="pg_dump"
|
||
PG_PSQL_CMD="psql"
|
||
else
|
||
# Пытаемся найти PostgreSQL контейнер
|
||
if command -v docker &> /dev/null; then
|
||
POSTGRES_CONTAINER=$(docker ps --format "{{.Names}}" 2>/dev/null | grep -iE "(postgres|db)" | head -1)
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
PG_DUMP_CMD="docker exec $POSTGRES_CONTAINER pg_dump"
|
||
PG_PSQL_CMD="docker exec -i $POSTGRES_CONTAINER psql"
|
||
echo -e "${BLUE}Используется PostgreSQL из Docker контейнера: $POSTGRES_CONTAINER${NC}"
|
||
fi
|
||
fi
|
||
fi
|
||
|
||
HAS_PG_DUMP=false
|
||
if [ -n "$PG_DUMP_CMD" ]; then
|
||
HAS_PG_DUMP=true
|
||
else
|
||
echo -e "${YELLOW}Предупреждение: pg_dump не найден. Сравнение схем будет пропущено.${NC}"
|
||
echo " Для полного тестирования установите PostgreSQL client tools"
|
||
fi
|
||
|
||
# Проверяем наличие директории миграций
|
||
if [ ! -d "$MIGRATIONS_PATH" ]; then
|
||
echo -e "${RED}Ошибка: Директория миграций не найдена: $MIGRATIONS_PATH${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# Проверяем наличие baseline миграции
|
||
if [ ! -f "$MIGRATIONS_PATH/000001_baseline.up.sql" ]; then
|
||
echo -e "${RED}Ошибка: Baseline миграция не найдена: $MIGRATIONS_PATH/000001_baseline.up.sql${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo "Параметры подключения:"
|
||
echo " Host: $DB_HOST"
|
||
echo " Port: $DB_PORT"
|
||
echo " User: $DB_USER"
|
||
echo " Test DB: $TEST_DB_NAME"
|
||
echo ""
|
||
|
||
# Проверяем подключение к БД
|
||
echo "1. Проверка подключения к БД..."
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
# Используем Docker
|
||
echo "SELECT 1;" | $PG_PSQL_CMD -U $DB_USER -d postgres > /dev/null 2>&1
|
||
elif [ -n "$PG_PSQL_CMD" ]; then
|
||
# Используем локальный psql
|
||
PGPASSWORD=$DB_PASSWORD $PG_PSQL_CMD \
|
||
-h $DB_HOST \
|
||
-p $DB_PORT \
|
||
-U $DB_USER \
|
||
-d postgres \
|
||
-c "SELECT 1;" > /dev/null 2>&1
|
||
else
|
||
# Пытаемся через стандартный psql
|
||
PGPASSWORD=$DB_PASSWORD psql \
|
||
-h $DB_HOST \
|
||
-p $DB_PORT \
|
||
-U $DB_USER \
|
||
-d postgres \
|
||
-c "SELECT 1;" > /dev/null 2>&1
|
||
fi
|
||
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}Ошибка: Не удалось подключиться к БД${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${GREEN}✓ Подключение успешно${NC}"
|
||
echo ""
|
||
|
||
# Создаем тестовую БД
|
||
echo "2. Создание тестовой БД..."
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
echo "CREATE DATABASE $TEST_DB_NAME;" | $PG_PSQL_CMD -U $DB_USER -d postgres > /dev/null 2>&1
|
||
elif [ -n "$PG_PSQL_CMD" ]; then
|
||
PGPASSWORD=$DB_PASSWORD $PG_PSQL_CMD \
|
||
-h $DB_HOST \
|
||
-p $DB_PORT \
|
||
-U $DB_USER \
|
||
-d postgres \
|
||
-c "CREATE DATABASE $TEST_DB_NAME;" > /dev/null 2>&1
|
||
else
|
||
PGPASSWORD=$DB_PASSWORD psql \
|
||
-h $DB_HOST \
|
||
-p $DB_PORT \
|
||
-U $DB_USER \
|
||
-d postgres \
|
||
-c "CREATE DATABASE $TEST_DB_NAME;" > /dev/null 2>&1
|
||
fi
|
||
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}Ошибка: Не удалось создать тестовую БД${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${GREEN}✓ Тестовая БД создана: $TEST_DB_NAME${NC}"
|
||
echo ""
|
||
|
||
# Ждем немного, чтобы БД точно создалась
|
||
sleep 1
|
||
|
||
# Проверяем, что БД создана
|
||
echo "3. Проверка существования тестовой БД..."
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
if echo "SELECT 1 FROM pg_database WHERE datname='$TEST_DB_NAME';" | $PG_PSQL_CMD -U $DB_USER -d postgres -t | grep -q 1; then
|
||
echo -e "${GREEN}✓ БД подтверждена${NC}"
|
||
else
|
||
echo -e "${RED}Ошибка: БД не найдена после создания${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
echo ""
|
||
|
||
# Применяем baseline миграцию
|
||
echo "4. Применение baseline миграции..."
|
||
cd "$(dirname "$0")" || exit 1
|
||
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
# Для Docker контейнеров используем psql напрямую, так как migrate может иметь проблемы с подключением
|
||
echo -e "${BLUE}Применение миграции через psql (Docker)...${NC}"
|
||
if [ -f "$MIGRATIONS_PATH/000001_baseline.up.sql" ]; then
|
||
if cat "$MIGRATIONS_PATH/000001_baseline.up.sql" | $PG_PSQL_CMD -U $DB_USER -d $TEST_DB_NAME > /dev/null 2>&1; then
|
||
echo -e "${GREEN}✓ Миграция применена через psql${NC}"
|
||
# Создаем таблицу schema_migrations вручную для migrate
|
||
echo "CREATE TABLE IF NOT EXISTS schema_migrations (version bigint NOT NULL PRIMARY KEY, dirty boolean NOT NULL);" | $PG_PSQL_CMD -U $DB_USER -d $TEST_DB_NAME > /dev/null 2>&1
|
||
echo "INSERT INTO schema_migrations (version, dirty) VALUES (1, false) ON CONFLICT (version) DO UPDATE SET dirty = false;" | $PG_PSQL_CMD -U $DB_USER -d $TEST_DB_NAME > /dev/null 2>&1
|
||
MIGRATE_SUCCESS=false # Устанавливаем в false, чтобы использовать psql для проверки версии
|
||
else
|
||
echo -e "${RED}Ошибка: Не удалось применить миграцию через psql${NC}"
|
||
exit 1
|
||
fi
|
||
else
|
||
echo -e "${RED}Ошибка: Файл миграции не найден${NC}"
|
||
exit 1
|
||
fi
|
||
DATABASE_URL="postgres://$DB_USER:$DB_PASSWORD@localhost:$DB_PORT/$TEST_DB_NAME?sslmode=disable"
|
||
else
|
||
DATABASE_URL="postgres://$DB_USER:$DB_PASSWORD@$DB_HOST:$DB_PORT/$TEST_DB_NAME?sslmode=disable"
|
||
if ! migrate -path "$MIGRATIONS_PATH" -database "$DATABASE_URL" up; then
|
||
echo -e "${RED}Ошибка: Не удалось применить baseline миграцию${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
echo -e "${GREEN}✓ Baseline миграция применена${NC}"
|
||
echo ""
|
||
|
||
# Проверяем версию миграции
|
||
echo "5. Проверка версии миграции..."
|
||
if [ -n "$POSTGRES_CONTAINER" ] && [ "${MIGRATE_SUCCESS:-false}" = "false" ]; then
|
||
# Проверяем версию через psql
|
||
VERSION=$(echo "SELECT version FROM schema_migrations;" | $PG_PSQL_CMD -U $DB_USER -d $TEST_DB_NAME -t 2>/dev/null | tr -d ' ' | head -1)
|
||
if [ -n "$VERSION" ] && [ "$VERSION" != "" ]; then
|
||
echo " Версия: $VERSION"
|
||
if [ "$VERSION" = "1" ]; then
|
||
echo -e "${GREEN}✓ Версия миграции корректна${NC}"
|
||
else
|
||
echo -e "${YELLOW}⚠ Неожиданная версия миграции: $VERSION${NC}"
|
||
fi
|
||
else
|
||
echo -e "${YELLOW}⚠ Не удалось определить версию миграции${NC}"
|
||
fi
|
||
else
|
||
# Используем migrate для проверки версии
|
||
VERSION=$(migrate -path "$MIGRATIONS_PATH" -database "$DATABASE_URL" version 2>&1)
|
||
echo " Версия: $VERSION"
|
||
|
||
if echo "$VERSION" | grep -qE "^1"; then
|
||
echo -e "${GREEN}✓ Версия миграции корректна${NC}"
|
||
else
|
||
echo -e "${YELLOW}⚠ Неожиданная версия миграции${NC}"
|
||
fi
|
||
fi
|
||
echo ""
|
||
|
||
# Экспортируем схему из тестовой БД (если pg_dump доступен)
|
||
if [ "$HAS_PG_DUMP" = true ]; then
|
||
echo "6. Экспорт схемы из тестовой БД..."
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
$PG_DUMP_CMD -U $DB_USER -d $TEST_DB_NAME --schema-only --no-owner --no-privileges > "$TMP_DIR/baseline_schema.sql"
|
||
else
|
||
PGPASSWORD=$DB_PASSWORD $PG_DUMP_CMD \
|
||
-h $DB_HOST \
|
||
-p $DB_PORT \
|
||
-U $DB_USER \
|
||
-d $TEST_DB_NAME \
|
||
--schema-only \
|
||
--no-owner \
|
||
--no-privileges \
|
||
-f "$TMP_DIR/baseline_schema.sql"
|
||
fi
|
||
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}Ошибка: Не удалось экспортировать схему${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${GREEN}✓ Схема экспортирована${NC}"
|
||
echo ""
|
||
|
||
# Пытаемся экспортировать схему из production БД для сравнения
|
||
echo "7. Экспорт схемы из production БД для сравнения..."
|
||
if [ -n "$POSTGRES_CONTAINER" ]; then
|
||
if $PG_DUMP_CMD -U $DB_USER -d $DB_NAME --schema-only --no-owner --no-privileges > "$TMP_DIR/production_schema.sql" 2>/dev/null; then
|
||
PROD_EXPORT_SUCCESS=true
|
||
else
|
||
PROD_EXPORT_SUCCESS=false
|
||
fi
|
||
else
|
||
if PGPASSWORD=$DB_PASSWORD $PG_DUMP_CMD \
|
||
-h $DB_HOST \
|
||
-p $DB_PORT \
|
||
-U $DB_USER \
|
||
-d $DB_NAME \
|
||
--schema-only \
|
||
--no-owner \
|
||
--no-privileges \
|
||
-f "$TMP_DIR/production_schema.sql" 2>/dev/null; then
|
||
PROD_EXPORT_SUCCESS=true
|
||
else
|
||
PROD_EXPORT_SUCCESS=false
|
||
fi
|
||
fi
|
||
|
||
if [ "$PROD_EXPORT_SUCCESS" = true ]; then
|
||
|
||
echo -e "${GREEN}✓ Схема production экспортирована${NC}"
|
||
echo ""
|
||
|
||
# Сравниваем схемы
|
||
echo "8. Сравнение схем..."
|
||
|
||
# Подсчитываем объекты
|
||
echo -e "${BLUE}Таблицы:${NC}"
|
||
BASELINE_TABLES=$(grep -c "CREATE TABLE" "$TMP_DIR/baseline_schema.sql" || echo "0")
|
||
PROD_TABLES=$(grep -c "CREATE TABLE" "$TMP_DIR/production_schema.sql" || echo "0")
|
||
echo " Baseline: $BASELINE_TABLES"
|
||
echo " Production: $PROD_TABLES"
|
||
|
||
if [ "$BASELINE_TABLES" -eq "$PROD_TABLES" ]; then
|
||
echo -e " ${GREEN}✓ Количество таблиц совпадает${NC}"
|
||
else
|
||
echo -e " ${YELLOW}⚠ Количество таблиц не совпадает${NC}"
|
||
fi
|
||
|
||
echo ""
|
||
echo -e "${BLUE}Индексы:${NC}"
|
||
BASELINE_INDEXES=$(grep -c "CREATE.*INDEX" "$TMP_DIR/baseline_schema.sql" || echo "0")
|
||
PROD_INDEXES=$(grep -c "CREATE.*INDEX" "$TMP_DIR/production_schema.sql" || echo "0")
|
||
echo " Baseline: $BASELINE_INDEXES"
|
||
echo " Production: $PROD_INDEXES"
|
||
|
||
if [ "$BASELINE_INDEXES" -eq "$PROD_INDEXES" ]; then
|
||
echo -e " ${GREEN}✓ Количество индексов совпадает${NC}"
|
||
else
|
||
echo -e " ${YELLOW}⚠ Количество индексов не совпадает${NC}"
|
||
fi
|
||
|
||
echo ""
|
||
echo -e "${BLUE}Materialized Views:${NC}"
|
||
BASELINE_MV=$(grep -c "CREATE MATERIALIZED VIEW" "$TMP_DIR/baseline_schema.sql" || echo "0")
|
||
PROD_MV=$(grep -c "CREATE MATERIALIZED VIEW" "$TMP_DIR/production_schema.sql" || echo "0")
|
||
echo " Baseline: $BASELINE_MV"
|
||
echo " Production: $PROD_MV"
|
||
|
||
if [ "$BASELINE_MV" -eq "$PROD_MV" ]; then
|
||
echo -e " ${GREEN}✓ Количество materialized views совпадает${NC}"
|
||
else
|
||
echo -e " ${YELLOW}⚠ Количество materialized views не совпадает${NC}"
|
||
fi
|
||
|
||
echo ""
|
||
echo "Для детального сравнения выполните:"
|
||
echo " diff $TMP_DIR/baseline_schema.sql $TMP_DIR/production_schema.sql"
|
||
echo ""
|
||
echo "Или используйте:"
|
||
echo " diff -u $TMP_DIR/baseline_schema.sql $TMP_DIR/production_schema.sql | less"
|
||
|
||
else
|
||
echo -e "${YELLOW}⚠ Не удалось экспортировать схему production БД${NC}"
|
||
echo " Продолжаем без сравнения"
|
||
echo ""
|
||
echo "Схема baseline сохранена в: $TMP_DIR/baseline_schema.sql"
|
||
fi
|
||
else
|
||
echo "6. Пропуск экспорта схемы (pg_dump недоступен)"
|
||
echo ""
|
||
echo -e "${YELLOW}Для полного тестирования установите PostgreSQL client tools:${NC}"
|
||
echo " macOS: brew install postgresql"
|
||
echo " или используйте Docker контейнер с PostgreSQL"
|
||
echo ""
|
||
fi
|
||
|
||
echo ""
|
||
echo "=== Тестирование завершено ==="
|
||
echo ""
|
||
echo -e "${GREEN}✓ Baseline миграция успешно применена к чистой БД${NC}"
|
||
echo ""
|