airflow - MssqlHook Airflow 连接

Question

我是使用 Airflow 的新手，我需要做的是使用 MssqlHook，但我不知道如何使用。我应该在构造函数中给出哪些元素？

我有一个名为 connection_test 的 Airflow 连接。

我不完全理解类中的属性:

class MsSqlHook(DbApiHook):
    """
    Interact with Microsoft SQL Server.
    """

    conn_name_attr = 'mssql_conn_id'
    default_conn_name = 'mssql_default'
    supports_autocommit = True

我有以下代码:

sqlhook=MsSqlHook(connection_test)
sqlhook.get_conn()

当我这样做时，错误是 未知原因连接失败 .

我应该怎么做才能使其与 Airflow 连接一起工作？

我需要的是为 MsSqlHook 调用函数 .get_conn()。

Answer 1

请参阅 Airflow 的标准示例。

https://github.com/gtoonstra/etl-with-airflow/blob/master/examples/mssql-example/dags/mssql_bcp_example.py

例如。:

t1 = MsSqlImportOperator(task_id='import_data',
                         table_name='test.test',
                         generate_synth_data=generate_synth_data,
                         mssql_conn_id='mssql',
                         dag=dag)

编辑

hook = MsSqlHook(mssql_conn_id="my_mssql_conn")
hook.run(sql)

您需要提供 Connections 中定义的连接。此外，如果使用 Hooks 查看相应的 Operators 通常会产生一些有关使用的信息。此代码来自 MSSQLOperator。