Airflow PostgresHook在执行插入语句后未提交并返回ID。
创始人
2024-08-01 11:01:57
0

在使用Airflow的PostgresHook执行插入语句后,如果未提交事务并返回ID,可以按照以下步骤来解决:

  1. 在插入语句之后,手动提交事务。可以通过调用PostgresHook对象的get_conn()方法获取数据库连接,然后调用连接对象的commit()方法来提交事务。
from airflow.hooks.postgres_hook import PostgresHook

# 创建PostgresHook对象
pg_hook = PostgresHook(postgres_conn_id="your_postgres_conn_id")

# 执行插入语句
insert_query = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
pg_hook.run(insert_query, parameters=("value1", "value2"))

# 手动提交事务
conn = pg_hook.get_conn()
conn.commit()
  1. 在调用run()方法时,设置autocommit参数为True,让PostgresHook自动提交事务。
from airflow.hooks.postgres_hook import PostgresHook

# 创建PostgresHook对象,并设置autocommit参数为True
pg_hook = PostgresHook(postgres_conn_id="your_postgres_conn_id", autocommit=True)

# 执行插入语句
insert_query = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
pg_hook.run(insert_query, parameters=("value1", "value2"))

# 不需要手动提交事务,因为autocommit参数已经设置为True

以上两种方法都可以解决在执行插入语句后未提交事务的问题。根据具体情况选择其中一种方法即可。

相关内容

热门资讯

安装apache-beam==... 出现此错误可能是因为用户的Python版本太低,而apache-beam==2.34.0需要更高的P...
避免在粘贴双引号时向VS 20... 在粘贴双引号时向VS 2022添加反斜杠的问题通常是由于编辑器的自动转义功能引起的。为了避免这个问题...
Android Recycle... 要在Android RecyclerView中实现滑动卡片效果,可以按照以下步骤进行操作:首先,在项...
omi系统和安卓系统哪个好,揭... OMI系统和安卓系统哪个好?这个问题就像是在问“苹果和橘子哪个更甜”,每个人都有自己的答案。今天,我...
原生ios和安卓系统,原生对比... 亲爱的读者们,你是否曾好奇过,为什么你的iPhone和安卓手机在操作体验上有着天壤之别?今天,就让我...
安装了Anaconda之后找不... 在安装Anaconda后,如果找不到Jupyter Notebook,可以尝试以下解决方法:检查环境...
Android - 无法确定任... 这个错误通常发生在Android项目中,表示编译Debug版本的Java代码时出现了依赖关系问题。下...
Android - NDK 预... 在Android NDK的构建过程中,LOCAL_SRC_FILES只能包含一个项目。如果需要在ND...
Akka生成Actor问题 在Akka框架中,可以使用ActorSystem对象生成Actor。但是,当我们在Actor类中尝试...
Agora-RTC-React... 出现这个错误原因是因为在 React 组件中使用,import AgoraRTC from “ago...