airflow_dag_jdbc_operator_create_table_quickstart.py

1import datetime
2
3from airflow import DAG
4from airflow.providers.jdbc.operators.jdbc import JdbcOperator
5
6with DAG(
7    dag_id="example_jdbc_operator",
8    schedule=None,
9    start_date=datetime.datetime(2021, 1, 1),
10    catchup=False,
11    tags=["example"],
12) as dag:
13
14    # [START howto_operator_jdbc]
15    create_table_mysql_task = JdbcOperator(
16        task_id="create_table_mysql",
17        sql="""
18            CREATE TABLE IF NOT EXISTS countries (
19                country_id INT AUTO_INCREMENT PRIMARY KEY,
20                name VARCHAR(255) NOT NULL
21            );
22        """,
23        jdbc_conn_id="jdbc_default",
24    )
25    # [END howto_operator_jdbc]