Files
clickstream-ch-kafka-supers…/scripts/run_batch.sh
T

199 lines
8.6 KiB
Bash
Executable File

#!/usr/bin/env bash
#
# Скрипт batch-трансформации данных: STG → ODS → DDS → DM
#
# Назначение:
# Запускает SQL-скрипты из sql/ods, sql/dds и sql/dm для преобразования данных между слоями:
# 1. STG → ODS : Типизация и перенос ошибок в *_errors
# 2. ODS → DDS : Сборка сущностей из типизированных данных
# 3. DDS → DM : Обновление сводки по качеству данных (dq_summary)
#
# Как запускать:
# make transform
# или: bash scripts/run_batch.sh
#
# Требования:
# - ClickHouse запущен (make up)
# - STG содержит данные (make data выполнен)
#
# Стратегия:
# Сейчас: полная перезагрузка (TRUNCATE + INSERT) — для демо
# В продакшене: инкрементальная загрузка по watermark
#
set -euo pipefail
# Директория со скриптом
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
SQL_ROOT_DIR="${SCRIPT_DIR}/../sql"
ODS_TRANSFORM_SQL="${SQL_ROOT_DIR}/ods/20_stg_to_ods.sql"
DDS_TRANSFORM_SQL="${SQL_ROOT_DIR}/dds/30_ods_to_dds.sql"
DM_TRANSFORM_SQL="${SQL_ROOT_DIR}/dm/40_dds_to_dm.sql"
# Параметры подключения
COMPOSE_BIN="${COMPOSE_BIN:-docker compose}"
CLICKHOUSE_SERVICE="${CLICKHOUSE_SERVICE:-clickhouse}"
CLICKHOUSE_DB="${CLICKHOUSE_DB:-default}"
CLICKHOUSE_USER="${CLICKHOUSE_USER:-default}"
CLICKHOUSE_PASSWORD="${CLICKHOUSE_PASSWORD:-123456}"
# -----------------------------------------------------------------------------
# Проверка: ClickHouse запущен?
# -----------------------------------------------------------------------------
if ! ${COMPOSE_BIN} ps | grep -q "${CLICKHOUSE_SERVICE}"; then
echo "Ошибка: Сервис '${CLICKHOUSE_SERVICE}' не запущен."
echo "Запустите сначала: make up"
exit 1
fi
# -----------------------------------------------------------------------------
# Проверка: SQL-файлы batch существуют?
# -----------------------------------------------------------------------------
if [[ ! -f "${ODS_TRANSFORM_SQL}" ]]; then
echo "Ошибка: Не найден SQL-файл: ${ODS_TRANSFORM_SQL}"
exit 1
fi
if [[ ! -f "${DDS_TRANSFORM_SQL}" ]]; then
echo "Ошибка: Не найден SQL-файл: ${DDS_TRANSFORM_SQL}"
exit 1
fi
if [[ ! -f "${DM_TRANSFORM_SQL}" ]]; then
echo "Ошибка: Не найден SQL-файл: ${DM_TRANSFORM_SQL}"
exit 1
fi
# -----------------------------------------------------------------------------
# Проверка: в STG есть данные?
# -----------------------------------------------------------------------------
STG_COUNT=$(${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--query="
SELECT
(SELECT count() FROM stg.browser_raw)
+ (SELECT count() FROM stg.location_raw)
+ (SELECT count() FROM stg.device_raw)
+ (SELECT count() FROM stg.geo_raw) AS stg_rows_total
" 2>/dev/null || echo "0")
if [[ "${STG_COUNT}" == "0" ]]; then
echo "Предупреждение: Таблицы STG пусты."
echo "Сначала загрузите данные: make data"
exit 1
fi
echo "Найдено ${STG_COUNT} строк в STG (суммарно по 4 потокам)"
echo ""
# -----------------------------------------------------------------------------
# Шаг 1: STG → ODS (типизация и DQ)
# -----------------------------------------------------------------------------
echo "Шаг 1: Обновление ODS слоя (STG → ODS)..."
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--multiquery < "${ODS_TRANSFORM_SQL}"
echo " ✓ ODS обновлён"
# Показываем статистику ODS
echo ""
echo "Статистика ODS:"
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--query="
SELECT 'ods.browser_event' AS table, count() AS rows FROM ods.browser_event
UNION ALL SELECT 'ods.location_event', count() FROM ods.location_event
UNION ALL SELECT 'ods.device_by_click', count() FROM ods.device_by_click
UNION ALL SELECT 'ods.geo_by_click', count() FROM ods.geo_by_click
UNION ALL SELECT 'ods.browser_event_errors', count() FROM ods.browser_event_errors
UNION ALL SELECT 'ods.location_event_errors', count() FROM ods.location_event_errors
UNION ALL SELECT 'ods.device_by_click_errors', count() FROM ods.device_by_click_errors
UNION ALL SELECT 'ods.geo_by_click_errors', count() FROM ods.geo_by_click_errors
FORMAT PrettyCompact
"
# -----------------------------------------------------------------------------
# Шаг 2: ODS → DDS (сборка сущностей)
# -----------------------------------------------------------------------------
echo ""
echo "Шаг 2: Обновление DDS слоя (ODS → DDS)..."
echo " - Очистка текущих данных (TRUNCATE)..."
# Очищаем таблицы перед загрузкой (полная перезагрузка для демо)
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--query="TRUNCATE TABLE dds.click" 2>/dev/null || true
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--query="TRUNCATE TABLE dds.event" 2>/dev/null || true
echo " - Загрузка dds.click (device + geo)..."
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--multiquery < "${DDS_TRANSFORM_SQL}"
echo " ✓ DDS обновлён"
# Показываем статистику DDS
echo ""
echo "Статистика DDS:"
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--query="SELECT 'dds.click' AS table, count() AS rows FROM dds.click UNION ALL SELECT 'dds.event', count() FROM dds.event FORMAT PrettyCompact"
# -----------------------------------------------------------------------------
# Шаг 3: DDS → DM (сводка по качеству)
# -----------------------------------------------------------------------------
echo ""
echo "Шаг 3: Обновление DM слоя (DQ summary)..."
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--multiquery < "${DM_TRANSFORM_SQL}"
echo " ✓ DM обновлён"
# Показываем сводку по качеству
echo ""
echo "Сводка по качеству данных:"
${COMPOSE_BIN} exec -T "${CLICKHOUSE_SERVICE}" clickhouse-client \
--user="${CLICKHOUSE_USER}" \
--password="${CLICKHOUSE_PASSWORD}" \
--database="${CLICKHOUSE_DB}" \
--query="SELECT * FROM dm.dq_summary ORDER BY layer, table_name, check_name FORMAT PrettyCompact"
# -----------------------------------------------------------------------------
# Итог
# -----------------------------------------------------------------------------
echo ""
echo "========================================"
echo "Batch-трансформация завершена!"
echo "========================================"
echo ""
echo "Доступные витрины для анализа:"
echo " - dm.v_events_enriched : Полное обогащение событий"
echo " - dm.v_daily_traffic : Агрегация по дням"
echo " - dm.v_top_pages_daily : Топ страниц"
echo " - dm.v_dq_errors_daily : Ошибки качества"
echo " - dm.v_session_overview : Обзор сессий"
echo " - dm.v_utm_effectiveness : Эффективность UTM"
echo " - dm.dq_summary : Статистика по слоям"
echo ""
echo "Подключитесь к Superset: http://localhost:8088"