23 lines
659 B
Python
23 lines
659 B
Python
from airflow import DAG
|
|
from airflow.operators.python import PythonOperator
|
|
from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook
|
|
from datetime import datetime
|
|
|
|
def test_clickhouse():
|
|
hook = ClickHouseHook(clickhouse_conn_id="my_clickhouse_provider")
|
|
# Простейший запрос
|
|
result = hook.get_records("SELECT 1")
|
|
print("ClickHouse result:", result)
|
|
|
|
with DAG(
|
|
dag_id="test_clickhouse_dag",
|
|
start_date=datetime(2024, 1, 1),
|
|
schedule=None,
|
|
catchup=False,
|
|
tags=["example"],
|
|
) as dag:
|
|
test_task = PythonOperator(
|
|
task_id="check_clickhouse",
|
|
python_callable=test_clickhouse
|
|
)
|