Python MysqlDB 使用cursor.rowcount 和 SSDictCursor 返回错误的计数

Question

我有以下代码

cur = db.cursor(cursors.SSDictCursor)
cur.execute("SELECT * FROM large_table")
result_count = cur.rowcount
print result_count

这会打印数字18446744073709551615，这显然是错误的。如果我删除cursors.SSDictCursor，则会显示正确的数字。谁能告诉我如何在保留 SSDictCursor 的同时获取返回的记录数？

Answer 1

要获取 SSDictCursor 或 SSCursor 返回的记录数，您唯一的选择是：

1. 获取整个结果并使用 len() 进行计数，这违背了使用 SSDictCursor 或 SSCursor 的初衷；

2. 在迭代时自己计算行数，这意味着直到结束时您才会知道计数（不太实用）；或者，

3. 运行一个额外的单独的 COUNT(*) 查询。

我强烈推荐第三种选择。如果您所做的只是SELECT COUNT(*) FROM table;，那么速度会非常快。对于一些更复杂的查询来说，它会更慢，但是通过适当的索引，它对于大多数用途来说仍然足够快。

---

顺便说一句，您看到的返回值在某种程度上是正确的；至少，就 MySQL C API 而言。

根据 PEP 249 中定义的 Python DB API，如果接口无法确定最后一次操作的rowcount。 @glglgl 解释了为什么在他们的答案中无法确定行数：

在内部，SSDictCursor使用mysql_use_result()，它允许服务器在获取完成之前开始传输数据。

换句话说，服务器不知道它最终要获取多少行。当您执行查询时，MySQLdb 存储 mysql_affected_rows() 在游标的 rowcount 属性中。由于计数不确定，因此此函数返回 -1 作为无符号 long long 整数 (my_ulonglong)，标准库的 ctypes 模块中提供的数字类型：

>>> from ctypes import c_ulonglong
>>> n = c_ulonglong(-1)
>>> n.value
18446744073709551615L

A快速而肮脏的替代方案对于ctypes，当您知道您将始终处理 64 位无符号整数时，是：

>>> -1 & 0xFFFFFFFFFFFFFFFF
18446744073709551615L

如果 MySQLdb 检查此返回值并给您，那就太好了您期望看到的有符号整数，但不幸的是它没有。

Answer 2

对于SSDictCursor，该值只能读取。确定光标何时用完。

在内部，SSDictCursor 使用 mysql_use_result() 允许服务器在获取完成之前开始传输数据。