26 lines
913 B
Python
26 lines
913 B
Python
from airflow import DAG
|
|
from airflow_clickhouse_plugin.operators.clickhouse import ClickHouseOperator
|
|
from airflow.utils.dates import days_ago
|
|
|
|
# Аргументы DAG по умолчанию
|
|
default_args = {
|
|
'owner': 'airflow',
|
|
'depends_on_past': False,
|
|
'start_date': days_ago(1),
|
|
}
|
|
|
|
# Создаем DAG
|
|
with DAG(
|
|
dag_id='test_clickhouse_connection',
|
|
default_args=default_args,
|
|
schedule_interval=None, # Запуск только вручную
|
|
tags=['test'],
|
|
) as dag:
|
|
|
|
# Задача с использованием ClickHouseOperator
|
|
test_query = ClickHouseOperator(
|
|
task_id='run_simple_query',
|
|
# Простой запрос для проверки соединения
|
|
sql='SELECT 1 AS test_value',
|
|
# clickhouse_conn_id='clickhouse_default' можно не указывать, т.к. это значение по умолчанию
|
|
) |