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' можно не указывать, т.к. это значение по умолчанию
|
||
) |