Python db-api：fetchone、fetchmany 和 fetchall

Question

我今天刚刚和一些同事讨论了 python 的 db-api fetchone、fetchmany 和 fetchall。

我确信每个用例的用例都取决于我正在使用的 db-api 的实现，但一般来说 fetchone、fetchmany 和 fetchall 的用例是什么？

换句话说，下面的内容是等价的吗？或者其中是否有一个比其他的更受青睐？如果是的话，在什么情况下？

cursor.execute("SELECT id, name FROM `table`")
for i in xrange(cursor.rowcount):
    id, name = cursor.fetchone()
    print id, name


cursor.execute("SELECT id, name FROM `table`")
result = cursor.fetchmany()
while result:
    for id, name in result:
        print id, name
    result = cursor.fetchmany()


cursor.execute("SELECT id, name FROM `table`")
for id, name in cursor.fetchall():
    print id, name

Answer 1

根据 官方 psycopg2 文档

fetchone()

获取查询结果集的下一行，返回单个元组，或者当没有更多数据可用时返回 None：

>>> cur.execute("SELECT * FROM test WHERE id = %s", (3,))
>>> cur.fetchone()

(3, 42, 'bar')

如果先前对execute*() 的调用未生成任何结果集或尚未发出任何调用，则会引发编程错误。

fetchmany([size=cursor.arraysize])

获取查询结果的下一组行，返回元组列表。当没有更多行可用时，将返回空列表。

每次调用要获取的行数由参数指定。如果未给出，则游标的数组大小决定要获取的行数。该方法应尝试获取大小参数指示的尽可能多的行。如果由于指定的行数不可用而无法实现此操作，则可能会返回更少的行：

>>> cur.execute("SELECT * FROM test;")
>>> cur.fetchmany(2)
[(1, 100, "abc'def"), (2, None, 'dada')]
>>> cur.fetchmany(2)
[(3, 42, 'bar')]
>>> cur.fetchmany(2)
[]

如果先前对execute*() 的调用未生成任何结果集或尚未发出任何调用，则会引发编程错误。

请注意，大小参数涉及性能考虑因素。为了获得最佳性能，通常最好使用 arraysize 属性。如果使用 size 参数，那么最好在一次 fetchmany() 调用到下一次调用中保留相同的值。

列表项

fetchall()

获取查询结果的所有（剩余）行，并将它们作为元组列表返回。如果没有更多记录可供获取，则返回空列表。

>>> cur.execute("SELECT * FROM test;")
>>> cur.fetchall()
[(1, 100, "abc'def"), (2, None, 'dada'), (3, 42, 'bar')]

如果先前对execute*()的调用没有产生任何结果集或者尚未发出任何调用，则会引发编程错误。

Answer 2

我认为这确实取决于实现，但是您可以通过查看 MySQLdb 源代码来了解差异。根据选项，mysqldb fetch* 将当前行集保留在内存或服务器端，因此 fetchmany 与 fetchone 在这里具有一定的灵活性，可以知道哪些内容保留在（python 的）内存中以及哪些内容保留在数据库服务器端。

PEP 249 没有提供太多细节，所以我猜这是根据数据库优化事物，而确切的语义是实现定义的。

Answer 3

这些是特定于实现的。

• fetchall

会从表中获取所有结果。当表尺寸较小时，此方法效果更好。如果表大小较大，在这种情况下 fetchall 将失败。

将使用大部分内存。

如果在网络上进行查询，则会导致出现一些问题。

• fetchmany

fetchmany 将仅获得所需数量的结果。您可以得出结果和过程。 fetchmany 实现的简单片段。

   while True:
    results = cursor.fetchmany(arraysize)
    if not results:
        break
    for result in results:
        yield result