Skip to main content
 首页 » 编程设计

python之cx-oracle中的并行执行

2025年05月04日7soundcode

我最近加入了一家新公司,并且是 python(他们首选的脚本语言)的新手,并且一直在使用 cx_oracle 创建一些 ETL 流程。到目前为止,我构建的脚本都是单线程作业,它们从 Oracle 源数据库中选择我需要的列子集,并将输出写入命名管道,外部进程正在等待读取该数据并将其插入目标.

在我访问某些行数在 5 亿 -20 亿行范围内的表之前,它一直运行良好。这项工作仍然有效,但需要花费很多时间才能完成。这些大型源表是分区的,所以我一直在尝试研究协调不同分区并行读取的方法,这样我就可以让两个或多个线程同时工作,每个线程写入一个单独的命名管道。

cx-oracle 中是否有一种优雅的方式来处理从同一个表的不同分区读取的多个线程?

这是我当前的(简单的)代码:

import cx_Oracle 
import csv 
 
# connect via SQL*Net string or by each segment in a separate argument 
connection = cx_Oracle.connect("user/password@TNS") 
 
 
csv.register_dialect('pipe_delimited', escapechar='\\' delimiter='|',quoting=csv.QUOTE_NONE) 
 
cursor = connection.cursor() 
f = open("<path_to_named_pipe>", "w") 
 
writer = csv.writer(f, dialect='pipe_delimited', lineterminator="\n") 
r = cursor.execute("""SELECT <column_list> from <SOURCE_TABLE>""") 
for row in cursor: 
        writer.writerow(row) 
f.close() 

我的一些源表有超过 1000 个分区,因此硬编码分区名称不是首选选项。我一直在考虑设置分区名称数组并遍历它们,但如果人们有其他想法,我很乐意听取他们的意见。

请您参考如下方法:

首先,您需要确保*cx_Oracle* 是线程安全的。因为它实现了 Python DB API Spec v2.0 ,您需要做的就是检查全局的 threadsafety 模块。 值 2 或 3 表示您可以打开到数据库的多个连接并同时运行多个查询。最好的方法是使用 threading模块,非常容易使用。 This是一篇关于如何开始使用它的简短而有趣的文章。

当然,不能保证流水线化查询会带来显着的性能提升(数据库引擎、I/O 等原因),但绝对值得一试。祝你好运!