feat(airflow): пробные DAG test_clickhouse и test_kafka вместо демонстрационного
Зачем. Демонстрационный DAG example_clickstream_hello ничего не проверял: он не обращался ни к ClickHouse, ни к Kafka, поэтому его зелёный результат ничего не говорил о стенде. Пробники проверяют связи по-настоящему — и тем же клиентом, каким будут ходить рабочие DAG. Что. - test_clickhouse: пишет строку в ReplicatedMergeTree на ноде 1 и читает её с ноды 2 через Distributed. Данные проходят путь «нода 2 → все шарды → шард ноды 1», то есть проверяется межшардовое чтение, а не одна нода. - test_kafka: пишет в постоянный топик сообщение с меткой прогона и вычитывает его обратно. - infra/airflow/Dockerfile: clickhouse-connect 1.6.0 и confluent-kafka 2.15.0 вшиты в образ, импорт проверяется на сборке — при запуске контейнера пакеты не доустанавливаются. - Проверки: scripts/stand-smoke.sh гоняет оба пробника через API Airflow, scripts/config-test.sh разбирает DAG без стенда, tests/stand-smoke-guards.sh проверяет красный путь, tests/dag-probes-unit.py — модульные проверки разбора. - README и ADR 0001 обновлены тем же изменением. - Удалён dags/example_clickstream_hello.py. Проверка. make config-test — пройдено 3, 3 и 6, ошибок 0. make clean; cp .env.example .env; make up — 116 с на чистых томах. make smoke — пройдено 25, ошибок 0; стенд занимает 2244,0 MiB. make smoke-cluster — все 8 проверок кластера. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -38,29 +38,34 @@ make smoke
|
||||
все локальные данные стенда, а следующий `make up` создаст их с новыми
|
||||
значениями.
|
||||
|
||||
`make up` собирает локальный образ Superset, поднимает весь стенд и ждёт
|
||||
здорового состояния долгоживущих контейнеров. Одноразовые `airflow-init` и
|
||||
`make up` собирает локальные образы Airflow и Superset, поднимает весь стенд и
|
||||
ждёт здорового состояния долгоживущих контейнеров. В образ Airflow добавлены
|
||||
закреплённые клиенты ClickHouse и Kafka. Одноразовые `airflow-init` и
|
||||
`superset-init` завершаются с кодом 0. Первый обновляет схему Airflow,
|
||||
подготавливает администратора и подключение к `clickhouse-01`. Второй обновляет
|
||||
Superset, создаёт администратора и импортирует подключение к `clickhouse-02`.
|
||||
|
||||
`make smoke` проверяет согласованность `.env.example` с Compose, зависимости
|
||||
машины, здоровье контейнеров, Kafka через порт машины, три цели Prometheus,
|
||||
источник Grafana, компоненты Airflow, ручной запуск примера DAG, метаданные и
|
||||
подключение Superset. В конце проверка ждёт 20 секунд покоя, печатает общую
|
||||
память контейнеров и падает при превышении 3,4 ГБ. Временный топик Kafka и
|
||||
проверочный запуск DAG удаляются.
|
||||
источник Grafana, компоненты Airflow, ручной запуск пробников `test_clickhouse`
|
||||
и `test_kafka`, метаданные и подключение Superset. Первый пробник создаёт
|
||||
таблицы на обеих нодах и читает через `Distributed` на ноде 2 строку из
|
||||
локальной таблицы ноды 1. Второй пишет в Kafka и читает свой маркер. В конце
|
||||
проверка ждёт 20 секунд покоя, печатает общую память контейнеров и падает при
|
||||
превышении 3,4 ГБ. Временный топик проверки с машины и запуски DAG удаляются;
|
||||
постоянный топик пробника сохраняется, а старые записи чистит Kafka.
|
||||
|
||||
`make smoke-cluster` запускает отдельную глубокую проверку ClickHouse: описание
|
||||
кластера, макросы, связь с keeper, `ReplicatedMergeTree`, `Distributed`, очередь
|
||||
распределённых DDL и очистку временных таблиц.
|
||||
|
||||
`make config-test` проверяет Compose, синтаксис Bash и Python и пробельные
|
||||
ошибки в diff без запуска стенда.
|
||||
`make config-test` проверяет Compose, синтаксис Bash и Python, малые проверки
|
||||
логики пробников и пробельные ошибки в diff без запуска стенда.
|
||||
|
||||
`make smoke-guards` сначала проверяет аварийную семантику кластерной проверки,
|
||||
а затем останавливает Prometheus и убеждается, что общая проверка называет его
|
||||
и завершается с ошибкой. В конце стенд восстанавливается.
|
||||
а затем удаляет служебную таблицу пробника только на второй ноде и
|
||||
останавливает Prometheus с Kafka. Общая проверка должна назвать Prometheus и
|
||||
оба пробника, после чего завершиться с ошибкой. В конце стенд восстанавливается.
|
||||
|
||||
Остановить контейнеры без удаления данных можно командой `make down`. Для
|
||||
полного сброса с удалением всех именованных томов используйте `make clean`.
|
||||
@@ -134,6 +139,10 @@ Airflow закреплён на 3.3.0. Состав обязательных п
|
||||
[архитектурой Airflow 3.3](https://airflow.apache.org/docs/apache-airflow/3.3.0/core-concepts/overview.html),
|
||||
[публичным интерфейсом](https://airflow.apache.org/docs/apache-airflow/3.3.0/public-airflow-interface.html)
|
||||
и [описанием здоровья](https://airflow.apache.org/docs/apache-airflow/3.3.0/administration-and-deployment/logging-monitoring/check-health.html).
|
||||
Для пробников проверены публичный `Connection.get` из `airflow.sdk` и клиент
|
||||
`clickhouse-connect`. Официальный провайдер Kafka сам использует
|
||||
`confluent-kafka`; отдельное подключение и обёртки провайдера здесь не нужны,
|
||||
поэтому прямой клиент оставляет образ и пример короче.
|
||||
Superset закреплён на 6.1.0; драйвер `clickhouse-connect`, форма
|
||||
`clickhousedb://` и драйвер Postgres сверены с
|
||||
[документацией подключений Superset](https://superset.apache.org/user-docs/6.1.0/databases/)
|
||||
|
||||
+5
-1
@@ -24,7 +24,11 @@ x-clickhouse-common: &clickhouse-common
|
||||
start_period: 10s
|
||||
|
||||
x-airflow-common: &airflow-common
|
||||
image: ${AIRFLOW_IMAGE:-apache/airflow:3.3.0}
|
||||
image: clickstream-airflow:local
|
||||
build:
|
||||
context: ./infra/airflow
|
||||
args:
|
||||
AIRFLOW_BASE_IMAGE: ${AIRFLOW_IMAGE:-apache/airflow:3.3.0}
|
||||
environment:
|
||||
AIRFLOW__CORE__EXECUTOR: LocalExecutor
|
||||
AIRFLOW__CORE__PARALLELISM: "4"
|
||||
|
||||
@@ -1,34 +0,0 @@
|
||||
"""Пример независимого DAG для проверки Airflow 3."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import datetime
|
||||
|
||||
from airflow.sdk import dag, task
|
||||
|
||||
|
||||
@dag(
|
||||
dag_id="example_clickstream_hello",
|
||||
schedule=None,
|
||||
start_date=datetime.datetime(2026, 1, 1, tzinfo=datetime.timezone.utc),
|
||||
catchup=False,
|
||||
tags=["пример"],
|
||||
doc_md=__doc__,
|
||||
)
|
||||
def example_clickstream_hello():
|
||||
@task
|
||||
def extract() -> dict[str, int]:
|
||||
return {"clicks": 3, "views": 10}
|
||||
|
||||
@task
|
||||
def calculate_ctr(counters: dict[str, int]) -> float:
|
||||
return round(counters["clicks"] / counters["views"], 3)
|
||||
|
||||
@task
|
||||
def show_result(ctr: float) -> None:
|
||||
print(f"CTR учебного примера: {ctr}")
|
||||
|
||||
show_result(calculate_ctr(extract()))
|
||||
|
||||
|
||||
example_clickstream_hello()
|
||||
@@ -0,0 +1,215 @@
|
||||
"""Сквозная проверка подключения Airflow к кластеру ClickHouse."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import datetime
|
||||
import uuid
|
||||
|
||||
from airflow.sdk import Connection, dag, get_current_context, task
|
||||
|
||||
CLUSTER = "clickstream_cluster"
|
||||
LOCAL_TABLE = "airflow_probe_local"
|
||||
DISTRIBUTED_TABLE = "airflow_probe_distributed"
|
||||
|
||||
|
||||
def _table_engines(client, *, query_node_2: bool) -> list[tuple[str, str]]:
|
||||
source = (
|
||||
"remote('clickhouse-02:9000', system.tables)"
|
||||
if query_node_2
|
||||
else "system.tables"
|
||||
)
|
||||
result = client.query(
|
||||
f"""
|
||||
SELECT name, engine
|
||||
FROM {source}
|
||||
WHERE database = 'default'
|
||||
AND name IN ('{LOCAL_TABLE}', '{DISTRIBUTED_TABLE}')
|
||||
ORDER BY name
|
||||
"""
|
||||
)
|
||||
return result.result_rows
|
||||
|
||||
|
||||
def _drop_tables(client) -> None:
|
||||
client.command(
|
||||
f"DROP TABLE IF EXISTS default.{DISTRIBUTED_TABLE} "
|
||||
f"ON CLUSTER {CLUSTER} SYNC"
|
||||
)
|
||||
client.command(
|
||||
f"DROP TABLE IF EXISTS default.{LOCAL_TABLE} ON CLUSTER {CLUSTER} SYNC"
|
||||
)
|
||||
|
||||
|
||||
def _assert_tables_absent(client) -> None:
|
||||
for query_node_2, node_name in ((False, "ноде 1"), (True, "ноде 2")):
|
||||
remaining = _table_engines(client, query_node_2=query_node_2)
|
||||
if remaining:
|
||||
raise RuntimeError(
|
||||
f"служебные таблицы остались на {node_name}: {remaining}"
|
||||
)
|
||||
|
||||
|
||||
def _assert_marker_path(
|
||||
*,
|
||||
local_rows: list[tuple[str, str]],
|
||||
distributed_rows: list[tuple[int, str, str]],
|
||||
node_2_hostname: str,
|
||||
marker: str,
|
||||
) -> None:
|
||||
if len(local_rows) != 1 or local_rows[0][1] != marker:
|
||||
raise RuntimeError(f"маркер не найден в локальной таблице: {marker}")
|
||||
node_1_hostname = local_rows[0][0]
|
||||
if node_1_hostname == node_2_hostname:
|
||||
raise RuntimeError("запись и чтение маркера должны выполняться с разных нод")
|
||||
expected_rows = [(1, node_1_hostname, marker)]
|
||||
if distributed_rows != expected_rows:
|
||||
raise RuntimeError(
|
||||
"нода 2 не прочитала маркер первого шарда через Distributed: "
|
||||
f"{marker}, получено {distributed_rows}"
|
||||
)
|
||||
|
||||
|
||||
def _cleanup_clickhouse_client(client, original_error: BaseException | None) -> None:
|
||||
cleanup_errors: list[Exception] = []
|
||||
try:
|
||||
_drop_tables(client)
|
||||
_assert_tables_absent(client)
|
||||
except Exception as error:
|
||||
cleanup_errors.append(error)
|
||||
try:
|
||||
client.close()
|
||||
except Exception as error:
|
||||
cleanup_errors.append(error)
|
||||
|
||||
if original_error is not None:
|
||||
for error in cleanup_errors:
|
||||
original_error.add_note(
|
||||
f"Дополнительная ошибка очистки ClickHouse: {error}"
|
||||
)
|
||||
return
|
||||
if len(cleanup_errors) == 1:
|
||||
raise cleanup_errors[0]
|
||||
if cleanup_errors:
|
||||
raise ExceptionGroup("ошибки очистки ClickHouse", cleanup_errors)
|
||||
|
||||
|
||||
@dag(
|
||||
dag_id="test_clickhouse",
|
||||
schedule=None,
|
||||
start_date=datetime.datetime(2026, 1, 1, tzinfo=datetime.timezone.utc),
|
||||
catchup=False,
|
||||
tags=["проверка"],
|
||||
doc_md=__doc__,
|
||||
)
|
||||
def test_clickhouse():
|
||||
@task
|
||||
def check_cluster_path() -> None:
|
||||
import clickhouse_connect
|
||||
|
||||
connection = Connection.get("clickhouse_default")
|
||||
client = clickhouse_connect.get_client(
|
||||
host=connection.host,
|
||||
port=connection.port,
|
||||
username=connection.login or "default",
|
||||
password=connection.password or "",
|
||||
database=connection.schema or "default",
|
||||
connect_timeout=5,
|
||||
send_receive_timeout=30,
|
||||
)
|
||||
marker = f"{get_current_context()['run_id']}:{uuid.uuid4()}"
|
||||
expected_tables = [
|
||||
(DISTRIBUTED_TABLE, "Distributed"),
|
||||
(LOCAL_TABLE, "ReplicatedMergeTree"),
|
||||
]
|
||||
probe_error = None
|
||||
|
||||
try:
|
||||
_drop_tables(client)
|
||||
_assert_tables_absent(client)
|
||||
client.command(
|
||||
f"""
|
||||
CREATE TABLE default.{LOCAL_TABLE} ON CLUSTER {CLUSTER}
|
||||
(
|
||||
marker String
|
||||
)
|
||||
ENGINE = ReplicatedMergeTree(
|
||||
'/clickhouse/tables/{{shard}}/{LOCAL_TABLE}',
|
||||
'{{replica}}'
|
||||
)
|
||||
ORDER BY marker
|
||||
"""
|
||||
)
|
||||
client.command(
|
||||
f"""
|
||||
CREATE TABLE default.{DISTRIBUTED_TABLE} ON CLUSTER {CLUSTER}
|
||||
AS default.{LOCAL_TABLE}
|
||||
ENGINE = Distributed(
|
||||
'{CLUSTER}',
|
||||
'default',
|
||||
'{LOCAL_TABLE}',
|
||||
cityHash64(marker)
|
||||
)
|
||||
"""
|
||||
)
|
||||
|
||||
for query_node_2, node_name in ((False, "ноде 1"), (True, "ноде 2")):
|
||||
actual_tables = _table_engines(
|
||||
client,
|
||||
query_node_2=query_node_2,
|
||||
)
|
||||
if actual_tables != expected_tables:
|
||||
raise RuntimeError(
|
||||
f"неверный набор таблиц на {node_name}: {actual_tables}"
|
||||
)
|
||||
|
||||
client.insert(
|
||||
f"default.{LOCAL_TABLE}",
|
||||
[[marker]],
|
||||
column_names=["marker"],
|
||||
)
|
||||
local_rows = client.query(
|
||||
f"""
|
||||
SELECT hostName(), marker
|
||||
FROM default.{LOCAL_TABLE}
|
||||
WHERE marker = {{marker:String}}
|
||||
""",
|
||||
parameters={"marker": marker},
|
||||
).result_rows
|
||||
node_2_rows = client.query(
|
||||
"""
|
||||
SELECT hostName()
|
||||
FROM remote('clickhouse-02:9000', system.one)
|
||||
"""
|
||||
).result_rows
|
||||
if len(node_2_rows) != 1:
|
||||
raise RuntimeError(
|
||||
f"не удалось определить имя ноды 2: {node_2_rows}"
|
||||
)
|
||||
distributed_rows = client.query(
|
||||
f"""
|
||||
SELECT _shard_num, hostName(), marker
|
||||
FROM remote(
|
||||
'clickhouse-02:9000',
|
||||
'default',
|
||||
'{DISTRIBUTED_TABLE}'
|
||||
)
|
||||
WHERE marker = {{marker:String}}
|
||||
""",
|
||||
parameters={"marker": marker},
|
||||
).result_rows
|
||||
_assert_marker_path(
|
||||
local_rows=local_rows,
|
||||
distributed_rows=distributed_rows,
|
||||
node_2_hostname=node_2_rows[0][0],
|
||||
marker=marker,
|
||||
)
|
||||
except BaseException as error:
|
||||
probe_error = error
|
||||
raise
|
||||
finally:
|
||||
_cleanup_clickhouse_client(client, probe_error)
|
||||
|
||||
check_cluster_path()
|
||||
|
||||
|
||||
test_clickhouse()
|
||||
@@ -0,0 +1,98 @@
|
||||
"""Проверка записи и чтения сообщения Airflow через Kafka."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import datetime
|
||||
import time
|
||||
import uuid
|
||||
|
||||
from airflow.sdk import dag, get_current_context, task
|
||||
|
||||
BROKER = "kafka:9092"
|
||||
TOPIC = "airflow_integration_probe"
|
||||
|
||||
|
||||
@dag(
|
||||
dag_id="test_kafka",
|
||||
schedule=None,
|
||||
start_date=datetime.datetime(2026, 1, 1, tzinfo=datetime.timezone.utc),
|
||||
catchup=False,
|
||||
tags=["проверка"],
|
||||
doc_md=__doc__,
|
||||
)
|
||||
def test_kafka():
|
||||
@task
|
||||
def check_round_trip() -> None:
|
||||
from confluent_kafka import Consumer, Producer, TopicPartition
|
||||
|
||||
marker = f"{get_current_context()['run_id']}:{uuid.uuid4()}"
|
||||
marker_bytes = marker.encode()
|
||||
group_id = f"airflow-probe-{uuid.uuid4()}"
|
||||
producer = None
|
||||
consumer = None
|
||||
|
||||
try:
|
||||
# Стенд опирается на автосоздание: срок хранения чистит записи, а
|
||||
# не топик; в бою автосоздание обычно выключают.
|
||||
delivery_errors: list[str] = []
|
||||
delivered_offsets: list[tuple[int, int]] = []
|
||||
|
||||
def on_delivery(error, message) -> None:
|
||||
if error is not None:
|
||||
delivery_errors.append(str(error))
|
||||
else:
|
||||
delivered_offsets.append((message.partition(), message.offset()))
|
||||
|
||||
producer = Producer(
|
||||
{
|
||||
"bootstrap.servers": BROKER,
|
||||
"client.id": "airflow-test-kafka",
|
||||
"message.timeout.ms": 10_000,
|
||||
"socket.timeout.ms": 5_000,
|
||||
}
|
||||
)
|
||||
producer.produce(
|
||||
TOPIC,
|
||||
key=marker_bytes,
|
||||
value=marker_bytes,
|
||||
on_delivery=on_delivery,
|
||||
)
|
||||
undelivered = producer.flush(10)
|
||||
if undelivered or delivery_errors or len(delivered_offsets) != 1:
|
||||
raise RuntimeError(
|
||||
"Kafka не подтвердила запись маркера: "
|
||||
f"не доставлено {undelivered}, ошибки {delivery_errors}, "
|
||||
f"смещения {delivered_offsets}"
|
||||
)
|
||||
|
||||
consumer = Consumer(
|
||||
{
|
||||
"bootstrap.servers": BROKER,
|
||||
"group.id": group_id,
|
||||
"enable.auto.commit": False,
|
||||
"session.timeout.ms": 6_000,
|
||||
"socket.timeout.ms": 5_000,
|
||||
}
|
||||
)
|
||||
partition, offset = delivered_offsets[0]
|
||||
consumer.assign([TopicPartition(TOPIC, partition, offset)])
|
||||
read_deadline = time.monotonic() + 20
|
||||
while time.monotonic() < read_deadline:
|
||||
message = consumer.poll(0.5)
|
||||
if message is None:
|
||||
continue
|
||||
if message.error():
|
||||
raise RuntimeError(f"Kafka вернула ошибку чтения: {message.error()}")
|
||||
if message.value() == marker_bytes:
|
||||
return
|
||||
raise RuntimeError(f"Kafka не вернула свой маркер за 20 секунд: {marker}")
|
||||
finally:
|
||||
if producer is not None:
|
||||
producer.flush(1)
|
||||
if consumer is not None:
|
||||
consumer.close()
|
||||
|
||||
check_round_trip()
|
||||
|
||||
|
||||
test_kafka()
|
||||
@@ -7,7 +7,7 @@
|
||||
Стенд использует Airflow 3.3.0 с LocalExecutor. Airflow разделён на
|
||||
одноразовый `airflow-init` и три долгоживущих процесса: API, планировщик и
|
||||
обработчик DAG. Triggerer не запускается: в каркасе нет отложенных задач.
|
||||
Пример DAG написан через публичный `airflow.sdk`.
|
||||
Два интеграционных пробника написаны через публичный `airflow.sdk`.
|
||||
|
||||
Для входа выбран SimpleAuthManager. У него нет команды создания пользователя,
|
||||
поэтому `airflow-init` записывает пароль администратора из окружения в
|
||||
@@ -25,9 +25,16 @@ JSON-файл отдельного тома `airflow_auth`. Новый имен
|
||||
|
||||
Airflow получает подготовленное общее подключение к `clickhouse-01`, а
|
||||
Superset — подключение `clickhousedb://` к `clickhouse-02`. Провайдер
|
||||
ClickHouse для Airflow пока не нужен. Разные ноды создают учебную ловушку:
|
||||
забытый `ON CLUSTER` проявится в Superset, даже если операция Airflow на первой
|
||||
ноде прошла успешно.
|
||||
ClickHouse для Airflow не нужен: локальный образ содержит прямой клиент.
|
||||
Пробник создаёт служебные таблицы `ON CLUSTER`, пишет на первой ноде и через
|
||||
`remote` читает `Distributed` на второй. Разные ноды сохраняют учебную
|
||||
ловушку: забытый `ON CLUSTER` проявится в Superset, даже если операция Airflow
|
||||
на первой ноде прошла успешно.
|
||||
|
||||
Клиент Kafka также добавлен прямо в образ Airflow. Официальный провайдер
|
||||
использует тот же `confluent-kafka`, а пробнику не нужны его подключение,
|
||||
операторы и обёртки. Топик пробника постоянный, сообщение каждого запуска
|
||||
отличается уникальным маркером.
|
||||
|
||||
Superset собирается от `apache/superset:6.1.0`. В образ добавлены
|
||||
`clickhouse-connect` для ClickHouse и `psycopg2-binary` для метаданных
|
||||
@@ -61,6 +68,11 @@ Prometheus читает встроенные точки метрик двух с
|
||||
Оттуда взяты `airflow.sdk`, обязательный отдельный обработчик DAG, возможность
|
||||
не запускать triggerer и проверка конкретных компонентов здоровья.
|
||||
|
||||
Для интеграционных пробников проверены `Connection.get` в публичном
|
||||
`airflow.sdk`, запросы через `clickhouse-connect` и состав официального
|
||||
провайдера Kafka. Выбран прямой `confluent-kafka`: провайдер строит свои
|
||||
подключения и обёртки поверх него, которые двум коротким пробникам не нужны.
|
||||
|
||||
Драйверы и строки подключения проверены по документации Superset 6.1.0:
|
||||
[подключения к базам](https://superset.apache.org/user-docs/6.1.0/databases/),
|
||||
[ClickHouse](https://superset.apache.org/user-docs/databases/supported/clickhouse/)
|
||||
|
||||
@@ -0,0 +1,13 @@
|
||||
ARG AIRFLOW_BASE_IMAGE
|
||||
FROM ${AIRFLOW_BASE_IMAGE}
|
||||
|
||||
USER root
|
||||
|
||||
# Официальный провайдер Kafka использует тот же confluent-kafka. Пробникам
|
||||
# не нужны его подключения и обёртки, поэтому в образ добавлен сам клиент.
|
||||
RUN uv pip install --python /home/airflow/.local/bin/python --no-cache \
|
||||
"clickhouse-connect==1.6.0" \
|
||||
"confluent-kafka==2.15.0" \
|
||||
&& python -c "import clickhouse_connect, confluent_kafka"
|
||||
|
||||
USER airflow
|
||||
@@ -19,6 +19,15 @@ jq -e '
|
||||
.services.superset.image == "clickstream-superset:local" and
|
||||
(.services.superset.build.args.SUPERSET_BASE_IMAGE | length > 0)
|
||||
' >/dev/null <<<"$config_json"
|
||||
grep -qx 'ARG AIRFLOW_BASE_IMAGE' "$ROOT_DIR/infra/airflow/Dockerfile"
|
||||
jq -e '
|
||||
.services["airflow-init"].image == "clickstream-airflow:local" and
|
||||
(.services["airflow-init"].build.args.AIRFLOW_BASE_IMAGE | length > 0) and
|
||||
all(
|
||||
.services[];
|
||||
((.environment // {}) | has("_PIP_ADDITIONAL_REQUIREMENTS") | not)
|
||||
)
|
||||
' >/dev/null <<<"$config_json"
|
||||
jq -e '
|
||||
.services["airflow-init"] as $service |
|
||||
($service.environment.AIRFLOW__CORE__SIMPLE_AUTH_MANAGER_PASSWORDS_FILE ==
|
||||
@@ -39,6 +48,20 @@ if [[ "${#shell_files[@]}" -eq 0 ]]; then
|
||||
fi
|
||||
bash -n "${shell_files[@]}"
|
||||
PYTHONPYCACHEPREFIX="$CACHE_DIR" uv run --no-project python -m compileall -q "$ROOT_DIR/dags"
|
||||
unit_status=0
|
||||
unit_output="$(
|
||||
PYTHONPYCACHEPREFIX="$CACHE_DIR" \
|
||||
uv run --no-project python "$ROOT_DIR/tests/dag-probes-unit.py" 2>&1
|
||||
)" || unit_status=$?
|
||||
printf '%s\n' "$unit_output"
|
||||
if [[ "$unit_status" -ne 0 ]]; then
|
||||
exit "$unit_status"
|
||||
fi
|
||||
if grep -Eq '^(Ran [0-9]+ tests|OK|FAILED)' <<<"$unit_output" ||
|
||||
! grep -Eq '^ИТОГ: пройдено [0-9]+, ошибок 0$' <<<"$unit_output"; then
|
||||
printf 'ОШИБКА: малые проверки пробников вывели итог не на русском языке.\n' >&2
|
||||
exit 1
|
||||
fi
|
||||
git -C "$ROOT_DIR" diff --check
|
||||
|
||||
printf 'ЗЕЛЁНО: Compose, Bash, Python и пробельные ошибки diff проверены.\n'
|
||||
|
||||
+86
-52
@@ -17,6 +17,7 @@ kafka_cleanup_image=''
|
||||
kafka_cleanup_port=''
|
||||
kafka_cleanup_topic=''
|
||||
airflow_cleanup_port=''
|
||||
airflow_cleanup_dag_id=''
|
||||
airflow_cleanup_original_paused=''
|
||||
airflow_cleanup_run_id=''
|
||||
airflow_cleanup_token=''
|
||||
@@ -232,7 +233,7 @@ cleanup_airflow_run() {
|
||||
encoded_run_id="$(jq -rn --arg value "$airflow_cleanup_run_id" '$value | @uri')"
|
||||
if curl -sf --max-time 10 -X DELETE \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/example_clickstream_hello/dagRuns/${encoded_run_id}" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${airflow_cleanup_dag_id}/dagRuns/${encoded_run_id}" \
|
||||
>/dev/null 2>&1; then
|
||||
airflow_cleanup_run_id=''
|
||||
return 0
|
||||
@@ -250,7 +251,7 @@ cleanup_airflow_pause() {
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
-H 'Content-Type: application/json' \
|
||||
-d "$payload" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/example_clickstream_hello" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${airflow_cleanup_dag_id}" \
|
||||
>/dev/null 2>&1; then
|
||||
airflow_cleanup_original_paused=''
|
||||
return 0
|
||||
@@ -383,18 +384,84 @@ check_grafana_datasource() {
|
||||
fi
|
||||
}
|
||||
|
||||
run_airflow_probe() {
|
||||
local dag
|
||||
local dag_id="$1"
|
||||
local encoded_run_id
|
||||
local response
|
||||
local run_state=''
|
||||
local unpaused
|
||||
local -i attempt
|
||||
|
||||
dag="$(curl -sf --max-time 10 \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${dag_id}" \
|
||||
2>/dev/null || true)"
|
||||
if ! jq -e --arg dag_id "$dag_id" '.dag_id == $dag_id' \
|
||||
>/dev/null 2>&1 <<<"$dag"; then
|
||||
fail "пробник ${dag_id} не виден через API Airflow"
|
||||
return 1
|
||||
fi
|
||||
|
||||
airflow_cleanup_dag_id="$dag_id"
|
||||
airflow_cleanup_original_paused="$(jq -r '.is_paused | tostring' <<<"$dag")"
|
||||
unpaused="$(curl -sf --max-time 10 -X PATCH \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
-H 'Content-Type: application/json' \
|
||||
-d '{"is_paused":false}' \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${dag_id}" \
|
||||
2>/dev/null || true)"
|
||||
if ! jq -e '.is_paused == false' >/dev/null 2>&1 <<<"$unpaused"; then
|
||||
fail "Airflow не смог включить пробник ${dag_id} перед ручным запуском"
|
||||
cleanup_airflow_pause || true
|
||||
return 1
|
||||
fi
|
||||
|
||||
response="$(curl -sf --max-time 10 -X POST \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
-H 'Content-Type: application/json' \
|
||||
-d '{"logical_date":null}' \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${dag_id}/dagRuns" \
|
||||
2>/dev/null || true)"
|
||||
airflow_cleanup_run_id="$(jq -r '.dag_run_id // empty' <<<"$response")"
|
||||
if [[ -z "$airflow_cleanup_run_id" ]]; then
|
||||
fail "Airflow не создал ручной запуск пробника ${dag_id}"
|
||||
cleanup_airflow_pause || true
|
||||
return 1
|
||||
fi
|
||||
|
||||
encoded_run_id="$(jq -rn --arg value "$airflow_cleanup_run_id" '$value | @uri')"
|
||||
for ((attempt = 1; attempt <= 60; attempt++)); do
|
||||
response="$(curl -sf --max-time 10 \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${dag_id}/dagRuns/${encoded_run_id}" \
|
||||
2>/dev/null || true)"
|
||||
run_state="$(jq -r '.state // empty' <<<"$response")"
|
||||
[[ "$run_state" == 'success' || "$run_state" == 'failed' ]] && break
|
||||
sleep 2
|
||||
done
|
||||
|
||||
if [[ "$run_state" == 'success' ]] && \
|
||||
cleanup_airflow_run && cleanup_airflow_pause; then
|
||||
pass "пробник ${dag_id} завершился успешно, запуск удалён, исходная пауза восстановлена"
|
||||
return 0
|
||||
fi
|
||||
|
||||
fail "пробник ${dag_id} не завершился чисто: состояние ${run_state:-неизвестно}"
|
||||
cleanup_airflow_run || true
|
||||
cleanup_airflow_pause || true
|
||||
return 1
|
||||
}
|
||||
|
||||
check_airflow() {
|
||||
local config
|
||||
local connection
|
||||
local dag
|
||||
local encoded_run_id
|
||||
local dag_id
|
||||
local health
|
||||
local password
|
||||
local response
|
||||
local run_state=''
|
||||
local unpaused
|
||||
local user
|
||||
local -i attempt
|
||||
|
||||
config="$(compose config --format json 2>/dev/null || true)"
|
||||
user="$(jq -r '.services["airflow-apiserver"].environment.AIRFLOW_ADMIN_USER // empty' <<<"$config")"
|
||||
@@ -419,29 +486,22 @@ check_airflow() {
|
||||
'{username: $username, password: $password}')" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/auth/token" 2>/dev/null || true)"
|
||||
airflow_cleanup_token="$(jq -r '.access_token // empty' <<<"$response")"
|
||||
if [[ -z "$airflow_cleanup_token" ]]; then
|
||||
fail 'Airflow не принял учётные данные администратора'
|
||||
return
|
||||
fi
|
||||
for dag_id in test_clickhouse test_kafka; do
|
||||
dag="$(curl -sf --max-time 10 \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/example_clickstream_hello" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/${dag_id}" \
|
||||
2>/dev/null || true)"
|
||||
if [[ -n "$airflow_cleanup_token" ]] && \
|
||||
jq -e '.dag_id == "example_clickstream_hello"' >/dev/null 2>&1 <<<"$dag"; then
|
||||
airflow_cleanup_original_paused="$(jq -r '.is_paused | tostring' <<<"$dag")"
|
||||
pass 'учётные данные администратора Airflow принимаются, пример DAG виден через API'
|
||||
else
|
||||
fail 'Airflow не принял учётные данные администратора или не показал пример DAG'
|
||||
return
|
||||
fi
|
||||
|
||||
unpaused="$(curl -sf --max-time 10 -X PATCH \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
-H 'Content-Type: application/json' \
|
||||
-d '{"is_paused":false}' \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/example_clickstream_hello" \
|
||||
2>/dev/null || true)"
|
||||
if ! jq -e '.is_paused == false' >/dev/null 2>&1 <<<"$unpaused"; then
|
||||
fail 'Airflow не смог включить пример DAG перед ручным запуском'
|
||||
if ! jq -e --arg dag_id "$dag_id" '.dag_id == $dag_id' \
|
||||
>/dev/null 2>&1 <<<"$dag"; then
|
||||
fail "Airflow не показал пробник ${dag_id}"
|
||||
return
|
||||
fi
|
||||
done
|
||||
pass 'учётные данные администратора Airflow принимаются, оба пробника видны через API'
|
||||
|
||||
connection="$(curl -sf --max-time 10 \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
@@ -463,34 +523,8 @@ check_airflow() {
|
||||
fail 'подключение Airflow не указывает на clickhouse-01 или нода недоступна из контейнера'
|
||||
fi
|
||||
|
||||
response="$(curl -sf --max-time 10 -X POST \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
-H 'Content-Type: application/json' \
|
||||
-d '{"logical_date":null}' \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/example_clickstream_hello/dagRuns" \
|
||||
2>/dev/null || true)"
|
||||
airflow_cleanup_run_id="$(jq -r '.dag_run_id // empty' <<<"$response")"
|
||||
if [[ -z "$airflow_cleanup_run_id" ]]; then
|
||||
fail 'Airflow не создал ручной запуск примера DAG'
|
||||
return
|
||||
fi
|
||||
|
||||
encoded_run_id="$(jq -rn --arg value "$airflow_cleanup_run_id" '$value | @uri')"
|
||||
for ((attempt = 1; attempt <= 60; attempt++)); do
|
||||
response="$(curl -sf --max-time 10 \
|
||||
-H "Authorization: Bearer ${airflow_cleanup_token}" \
|
||||
"http://127.0.0.1:${airflow_cleanup_port}/api/v2/dags/example_clickstream_hello/dagRuns/${encoded_run_id}" \
|
||||
2>/dev/null || true)"
|
||||
run_state="$(jq -r '.state // empty' <<<"$response")"
|
||||
[[ "$run_state" == 'success' || "$run_state" == 'failed' ]] && break
|
||||
sleep 2
|
||||
done
|
||||
|
||||
if [[ "$run_state" == 'success' ]] && cleanup_airflow_run && cleanup_airflow_pause; then
|
||||
pass 'ручной запуск примера DAG завершился успешно, удалён, исходная пауза восстановлена'
|
||||
else
|
||||
fail "ручной запуск примера DAG не завершился чисто: состояние ${run_state:-неизвестно}"
|
||||
fi
|
||||
run_airflow_probe test_clickhouse || true
|
||||
run_airflow_probe test_kafka || true
|
||||
}
|
||||
|
||||
check_superset() {
|
||||
|
||||
@@ -0,0 +1,184 @@
|
||||
"""Малые проверки логики пробника ClickHouse без запуска Airflow."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import importlib.util
|
||||
import sys
|
||||
import types
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def load_clickhouse_dag_module():
|
||||
airflow_module = types.ModuleType("airflow")
|
||||
sdk_module = types.ModuleType("airflow.sdk")
|
||||
|
||||
def dag(**_kwargs):
|
||||
def decorate(function):
|
||||
return function
|
||||
|
||||
return decorate
|
||||
|
||||
def task(function):
|
||||
def declare_task(*_args, **_kwargs):
|
||||
return None
|
||||
|
||||
return declare_task
|
||||
|
||||
sdk_module.Connection = object
|
||||
sdk_module.dag = dag
|
||||
sdk_module.get_current_context = lambda: {}
|
||||
sdk_module.task = task
|
||||
airflow_module.sdk = sdk_module
|
||||
sys.modules["airflow"] = airflow_module
|
||||
sys.modules["airflow.sdk"] = sdk_module
|
||||
|
||||
dag_path = Path(__file__).resolve().parents[1] / "dags" / "test_clickhouse.py"
|
||||
spec = importlib.util.spec_from_file_location("test_clickhouse_dag", dag_path)
|
||||
if spec is None or spec.loader is None:
|
||||
raise RuntimeError("не удалось загрузить модуль пробника ClickHouse")
|
||||
module = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(module)
|
||||
return module
|
||||
|
||||
|
||||
def load_kafka_dag_task():
|
||||
airflow_module = types.ModuleType("airflow")
|
||||
sdk_module = types.ModuleType("airflow.sdk")
|
||||
captured_tasks = {}
|
||||
|
||||
def dag(**_kwargs):
|
||||
def decorate(function):
|
||||
return function
|
||||
|
||||
return decorate
|
||||
|
||||
def task(function):
|
||||
captured_tasks[function.__name__] = function
|
||||
|
||||
def declare_task(*_args, **_kwargs):
|
||||
return None
|
||||
|
||||
return declare_task
|
||||
|
||||
sdk_module.dag = dag
|
||||
sdk_module.get_current_context = lambda: {"run_id": "unit-test"}
|
||||
sdk_module.task = task
|
||||
airflow_module.sdk = sdk_module
|
||||
sys.modules["airflow"] = airflow_module
|
||||
sys.modules["airflow.sdk"] = sdk_module
|
||||
|
||||
dag_path = Path(__file__).resolve().parents[1] / "dags" / "test_kafka.py"
|
||||
spec = importlib.util.spec_from_file_location("test_kafka_dag", dag_path)
|
||||
if spec is None or spec.loader is None:
|
||||
raise RuntimeError("не удалось загрузить модуль пробника Kafka")
|
||||
module = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(module)
|
||||
return captured_tasks["check_round_trip"]
|
||||
|
||||
|
||||
class FailingCleanupClient:
|
||||
def __init__(self) -> None:
|
||||
self.closed = False
|
||||
|
||||
def command(self, _sql: str) -> None:
|
||||
raise RuntimeError("очистка недоступна")
|
||||
|
||||
def close(self) -> None:
|
||||
self.closed = True
|
||||
|
||||
|
||||
class ClickHouseProbeTests(unittest.TestCase):
|
||||
@classmethod
|
||||
def setUpClass(cls) -> None:
|
||||
cls.module = load_clickhouse_dag_module()
|
||||
|
||||
def test_marker_path_requires_different_nodes_and_first_shard(self) -> None:
|
||||
marker = "свой-маркер"
|
||||
self.module._assert_marker_path(
|
||||
local_rows=[("clickhouse-01-host", marker)],
|
||||
distributed_rows=[(1, "clickhouse-01-host", marker)],
|
||||
node_2_hostname="clickhouse-02-host",
|
||||
marker=marker,
|
||||
)
|
||||
|
||||
with self.assertRaisesRegex(RuntimeError, "разных нод"):
|
||||
self.module._assert_marker_path(
|
||||
local_rows=[("clickhouse-02-host", marker)],
|
||||
distributed_rows=[(2, "clickhouse-02-host", marker)],
|
||||
node_2_hostname="clickhouse-02-host",
|
||||
marker=marker,
|
||||
)
|
||||
with self.assertRaisesRegex(RuntimeError, "первого шарда"):
|
||||
self.module._assert_marker_path(
|
||||
local_rows=[("clickhouse-01-host", marker)],
|
||||
distributed_rows=[(2, "clickhouse-01-host", marker)],
|
||||
node_2_hostname="clickhouse-02-host",
|
||||
marker=marker,
|
||||
)
|
||||
|
||||
def test_cleanup_keeps_original_error_as_primary(self) -> None:
|
||||
client = FailingCleanupClient()
|
||||
original_error = RuntimeError("маркер не найден")
|
||||
|
||||
self.module._cleanup_clickhouse_client(client, original_error)
|
||||
|
||||
self.assertTrue(client.closed)
|
||||
self.assertIn("очистка недоступна", "\n".join(original_error.__notes__))
|
||||
|
||||
|
||||
class KafkaProbeTests(unittest.TestCase):
|
||||
def test_producer_flushes_when_consumer_creation_fails(self) -> None:
|
||||
kafka_module = types.ModuleType("confluent_kafka")
|
||||
flush_timeouts = []
|
||||
|
||||
class Message:
|
||||
def partition(self) -> int:
|
||||
return 0
|
||||
|
||||
def offset(self) -> int:
|
||||
return 1
|
||||
|
||||
class Producer:
|
||||
def __init__(self, _config) -> None:
|
||||
pass
|
||||
|
||||
def produce(self, _topic, **kwargs) -> None:
|
||||
kwargs["on_delivery"](None, Message())
|
||||
|
||||
def flush(self, timeout: int) -> int:
|
||||
flush_timeouts.append(timeout)
|
||||
return 0
|
||||
|
||||
class Consumer:
|
||||
def __init__(self, _config) -> None:
|
||||
raise RuntimeError("чтение недоступно")
|
||||
|
||||
kafka_module.Consumer = Consumer
|
||||
kafka_module.Producer = Producer
|
||||
kafka_module.TopicPartition = object
|
||||
sys.modules["confluent_kafka"] = kafka_module
|
||||
check_round_trip = load_kafka_dag_task()
|
||||
|
||||
with self.assertRaisesRegex(RuntimeError, "чтение недоступно"):
|
||||
check_round_trip()
|
||||
|
||||
self.assertEqual(flush_timeouts, [10, 1])
|
||||
|
||||
|
||||
def run_tests() -> int:
|
||||
suite = unittest.defaultTestLoader.loadTestsFromModule(sys.modules[__name__])
|
||||
result = unittest.TestResult()
|
||||
suite.run(result)
|
||||
|
||||
problems = result.failures + result.errors
|
||||
for test, details in problems:
|
||||
print(f"ОШИБКА: {test.id()}", file=sys.stderr)
|
||||
print(details, file=sys.stderr)
|
||||
passed = result.testsRun - len(problems) - len(result.skipped)
|
||||
print(f"ИТОГ: пройдено {passed}, ошибок {len(problems)}")
|
||||
return 0 if result.wasSuccessful() else 1
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(run_tests())
|
||||
+11
-1
@@ -14,6 +14,16 @@ printf 'ЗЕЛЁНО: README объясняет сброс томов после
|
||||
grep -Fxq -- '- нода 2 — `http://127.0.0.1:28124`, нативный порт `29001`;' "$README"
|
||||
printf 'ЗЕЛЁНО: список портов остаётся единым списком.\n'
|
||||
|
||||
awk '
|
||||
previous == "`make config-test` проверяет Compose, синтаксис Bash и Python, малые проверки" &&
|
||||
$0 == "логики пробников и пробельные ошибки в diff без запуска стенда." {
|
||||
found = 1
|
||||
}
|
||||
{previous = $0}
|
||||
END {exit !found}
|
||||
' "$README"
|
||||
printf 'ЗЕЛЁНО: README перечисляет малые проверки пробников в составе config-test.\n'
|
||||
|
||||
awk '
|
||||
previous == "После изменения `infra/clickhouse/config.d/prometheus.xml` выполните" &&
|
||||
$0 == "`docker compose restart clickhouse-01 clickhouse-02`: обычный `make up` не" {
|
||||
@@ -35,4 +45,4 @@ elif [[ "$grep_status" -ne 1 ]]; then
|
||||
fi
|
||||
printf 'ЗЕЛЁНО: отчёт проверки использует русское обозначение ГБ.\n'
|
||||
|
||||
printf 'ИТОГ: пройдено 5, ошибок 0\n'
|
||||
printf 'ИТОГ: пройдено 6, ошибок 0\n'
|
||||
|
||||
+126
-4
@@ -4,13 +4,100 @@ set -uo pipefail
|
||||
readonly ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
||||
read -r -a COMPOSE_CMD <<<"${COMPOSE_BIN:-docker compose}"
|
||||
readonly LOG_FILE="$(mktemp)"
|
||||
readonly CLICKHOUSE_LOCAL_TABLE="airflow_probe_local"
|
||||
readonly CLICKHOUSE_DISTRIBUTED_TABLE="airflow_probe_distributed"
|
||||
restored=0
|
||||
passed=0
|
||||
red_smoke_pid=''
|
||||
|
||||
compose() {
|
||||
"${COMPOSE_CMD[@]}" --project-directory "$ROOT_DIR" "$@"
|
||||
}
|
||||
|
||||
published_port() {
|
||||
local binding
|
||||
local service="$1"
|
||||
local container_port="$2"
|
||||
binding="$(compose port "$service" "$container_port" 2>/dev/null || true)"
|
||||
printf '%s\n' "${binding##*:}"
|
||||
}
|
||||
|
||||
query_node_2() {
|
||||
local sql="$1"
|
||||
local port
|
||||
port="$(published_port clickhouse-02 8123)"
|
||||
curl -sf --max-time 2 --data-binary "$sql" "http://127.0.0.1:${port}/"
|
||||
}
|
||||
|
||||
break_clickhouse_probe() {
|
||||
local deadline
|
||||
local table_state
|
||||
|
||||
deadline=$((EPOCHSECONDS + 60))
|
||||
printf 'Ожидание служебных таблиц на ноде 2, предел 60 секунд...\n'
|
||||
while [[ "$EPOCHSECONDS" -lt "$deadline" ]]; do
|
||||
table_state="$(query_node_2 "
|
||||
SELECT
|
||||
countIf(name = '${CLICKHOUSE_LOCAL_TABLE}'),
|
||||
countIf(name = '${CLICKHOUSE_DISTRIBUTED_TABLE}')
|
||||
FROM system.tables
|
||||
WHERE database = 'default'
|
||||
AND name IN (
|
||||
'${CLICKHOUSE_LOCAL_TABLE}',
|
||||
'${CLICKHOUSE_DISTRIBUTED_TABLE}'
|
||||
)
|
||||
FORMAT TSV
|
||||
" 2>/dev/null || true)"
|
||||
if [[ "$table_state" == $'1\t1' ]]; then
|
||||
query_node_2 \
|
||||
"DROP TABLE IF EXISTS default.${CLICKHOUSE_LOCAL_TABLE} SYNC" \
|
||||
>/dev/null
|
||||
table_state="$(query_node_2 "
|
||||
SELECT
|
||||
countIf(name = '${CLICKHOUSE_LOCAL_TABLE}'),
|
||||
countIf(name = '${CLICKHOUSE_DISTRIBUTED_TABLE}')
|
||||
FROM system.tables
|
||||
WHERE database = 'default'
|
||||
AND name IN (
|
||||
'${CLICKHOUSE_LOCAL_TABLE}',
|
||||
'${CLICKHOUSE_DISTRIBUTED_TABLE}'
|
||||
)
|
||||
FORMAT TSV
|
||||
" 2>/dev/null || true)"
|
||||
if [[ "$table_state" == $'0\t1' ]]; then
|
||||
printf 'Служебная локальная таблица удалена только на ноде 2.\n'
|
||||
return
|
||||
fi
|
||||
return 1
|
||||
fi
|
||||
sleep 0.05
|
||||
done
|
||||
return 1
|
||||
}
|
||||
|
||||
clickhouse_break_is_reported() {
|
||||
compose exec -T airflow-scheduler bash -ceu '
|
||||
latest="$(
|
||||
find /opt/airflow/logs/dag_id=test_clickhouse \
|
||||
-type f -name "attempt=1.log" -printf "%T@ %p\n" |
|
||||
sort -nr |
|
||||
head -n 1
|
||||
)"
|
||||
latest="${latest#* }"
|
||||
test -n "$latest"
|
||||
grep -Eq \
|
||||
"неверный набор таблиц на ноде 2|Unknown table expression identifier '\''default.airflow_probe_local'\''" \
|
||||
"$latest"
|
||||
'
|
||||
}
|
||||
|
||||
probe_failure_is_reported() {
|
||||
local dag_id="$1"
|
||||
grep -Eq \
|
||||
"ОШИБКА: (пробник ${dag_id}|Airflow не показал пробник ${dag_id})" \
|
||||
"$LOG_FILE"
|
||||
}
|
||||
|
||||
restore_stand() {
|
||||
make --no-print-directory -C "$ROOT_DIR" COMPOSE="${COMPOSE_BIN:-docker compose}" up >/dev/null
|
||||
}
|
||||
@@ -18,6 +105,10 @@ restore_stand() {
|
||||
on_exit() {
|
||||
local status=$?
|
||||
trap - EXIT INT TERM
|
||||
if [[ -n "$red_smoke_pid" ]]; then
|
||||
kill -TERM -- "-$red_smoke_pid" >/dev/null 2>&1 || true
|
||||
wait "$red_smoke_pid" >/dev/null 2>&1 || true
|
||||
fi
|
||||
rm -f "$LOG_FILE"
|
||||
if [[ "$restored" -eq 0 ]] && ! restore_stand; then
|
||||
printf 'ОШИБКА: не удалось восстановить стенд после проверки.\n' >&2
|
||||
@@ -38,14 +129,45 @@ else
|
||||
fi
|
||||
|
||||
compose stop prometheus >/dev/null
|
||||
make --no-print-directory -C "$ROOT_DIR" COMPOSE="${COMPOSE_BIN:-docker compose}" smoke >"$LOG_FILE" 2>&1
|
||||
setsid make --no-print-directory -C "$ROOT_DIR" \
|
||||
COMPOSE="${COMPOSE_BIN:-docker compose}" smoke >"$LOG_FILE" 2>&1 &
|
||||
red_smoke_pid=$!
|
||||
if ! break_clickhouse_probe; then
|
||||
printf 'ОШИБКА: не удалось удалить служебную таблицу только на ноде 2.\n' >&2
|
||||
exit 1
|
||||
fi
|
||||
compose stop kafka >/dev/null
|
||||
wait "$red_smoke_pid"
|
||||
status=$?
|
||||
red_smoke_pid=''
|
||||
|
||||
if [[ "$status" -ne 0 ]] && grep -q 'ОШИБКА: сервис prometheus' "$LOG_FILE"; then
|
||||
red_path_ok=1
|
||||
if [[ "$status" -eq 0 ]]; then
|
||||
printf 'ОШИБКА: make smoke остался зелёным после внесённых поломок.\n' >&2
|
||||
red_path_ok=0
|
||||
fi
|
||||
if ! grep -q 'ОШИБКА: сервис prometheus' "$LOG_FILE"; then
|
||||
printf 'ОШИБКА: make smoke не назвал остановленный prometheus.\n' >&2
|
||||
red_path_ok=0
|
||||
fi
|
||||
if ! probe_failure_is_reported test_clickhouse; then
|
||||
printf 'ОШИБКА: make smoke не назвал пробник test_clickhouse.\n' >&2
|
||||
red_path_ok=0
|
||||
fi
|
||||
if ! probe_failure_is_reported test_kafka; then
|
||||
printf 'ОШИБКА: make smoke не назвал пробник test_kafka.\n' >&2
|
||||
red_path_ok=0
|
||||
fi
|
||||
if ! clickhouse_break_is_reported; then
|
||||
printf 'ОШИБКА: журнал test_clickhouse не связал отказ с удалённой таблицей на ноде 2.\n' >&2
|
||||
red_path_ok=0
|
||||
fi
|
||||
|
||||
if [[ "$red_path_ok" -eq 1 ]]; then
|
||||
passed=$((passed + 1))
|
||||
printf 'ЗЕЛЁНО: остановленный prometheus делает make smoke красным и назван в отчёте.\n'
|
||||
printf 'ЗЕЛЁНО: удаление таблицы на ноде 2 и остановка prometheus с kafka делают make smoke красным; оба пробника названы в отчёте.\n'
|
||||
else
|
||||
printf 'ОШИБКА: make smoke не обнаружил остановленный prometheus; код=%s.\n' "$status" >&2
|
||||
printf 'ОШИБКА: проверка красного пути завершилась с кодом make smoke %s.\n' "$status" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
|
||||
Reference in New Issue
Block a user