我正在开发一个系统的一部分,其中进程限制为大约 350MB 的 RAM;我们使用 cx_Oracle 从外部系统下载文件进行处理。
外部系统将文件存储为 BLOB,我们可以通过以下方式获取它们:
# ... set up Oracle connection, then
cursor.execute(u"""SELECT filename, data, filesize
FROM FILEDATA
WHERE ID = :id""", id=the_one_you_wanted)
filename, lob, filesize = cursor.fetchone()
with open(filename, "w") as the_file:
the_file.write(lob.read())
lob.read()
当我们遇到大于 300-350MB 的文件时,显然会失败MemoryError
,所以我们尝试了这样的方法,而不是一次读取所有文件:
read_size = 0
chunk_size = lob.getchunksize() * 100
while read_size < filesize:
data = lob.read(chunk_size, read_size + 1)
read_size += len(data)
the_file.write(data)
不幸的是,我们仍然MemoryError
经过几次迭代。从所花费的时间lob.read()
和我们最终得到的内存不足的情况来看,看起来好像每次都lob.read()
从数据库中提取 (chunk_size + read_size) 个字节。也就是说,读取需要 O(n) 时间和 O(n) 内存,即使缓冲区小得多。
为了解决这个问题,我们尝试了类似的方法:
read_size = 0
while read_size < filesize:
q = u'''SELECT dbms_lob.substr(data, 2000, %s)
FROM FILEDATA WHERE ID = :id''' % (read_bytes + 1)
cursor.execute(q, id=filedataid[0])
row = cursor.fetchone()
read_bytes += len(row[0])
the_file.write(row[0])
这一次提取 2000 个字节(argh),并且需要很长时间(对于 1.5GB 文件来说大约需要两个小时)。为什么是 2000 字节?根据 Oracle 文档,dbms_lob.substr()
将其返回值存储在 RAW 中,限制为 2000 字节。
有什么方法可以将dbms_lob.substr()
结果存储在更大的数据对象中,一次读取几兆字节?如何使用 cx_Oracle 执行此操作?