我正在考虑将 Airflow 从 ECS 迁移到 MWAA,显然效果很好。但是从 CICD 的角度来看,加载连接有一些限制。 根据文档,我们无法根据他们在此处所说的内容使用命令行设置连接:https://docs.aws.amazon.com/mwaa/latest/userguide/access-airflow-ui.html#call-mwaa-apis-cli
我知道我们可以使用 UI 来实现,但这不是我们的意图。有谁知道如何处理这个问题,或者是否有任何方法可以自动处理?
非常感谢 哈维
答案 0 :(得分:0)
您可以在 DAG 中添加一个步骤,如果连接尚不存在于环境中,它将以编程方式添加连接:
def add_connection_callable(**kwargs):
connection: Connection = Connection(
conn_id="foo",
conn_type="HTTP",
host=kwargs['host'],
port=kwargs['port']
)
session: Session = settings.Session
db_connection: Connection = session.query(Connection) \
.filter(Connection.conn_id == "foo") \
.first()
if db_connection is None:
logging.info("Adding connection \"foo\"..")
session.add(connection)
session.commit()
else:
logging.info("Connection \"foo\" already exists.")
add_connection: PythonOperator = PythonOperator(
task_id="add_connection",
python_callable=add_connection_callable,
op_kwargs={
"host": "{{ dag_run.conf['Host'] }}",
"port": "{{ dag_run.conf['Port'] }}"
},
dag=dag)