Mysql 为什么 SQLAlchemy count() 比原始查询慢得多

发布于02月08日

我在MySQL数据库中使用SQLAlchemy，我想计算表中的行数(大约300k).SQLAlchemy count函数的运行时间大约是直接在MySQL中编写同一个查询的50倍.我做错什么了吗？

# this takes over 3 seconds to return
session.query(Segment).count()

然而:

SELECT COUNT(*) FROM segments;
+----------+
| COUNT(*) |
+----------+
|   281992 |
+----------+
1 row in set (0.07 sec)

速度的差异随着表格的大小而增加(在100k行下几乎不明显).

Update

用session.query(Segment.id).count()而不是session.query(Segment).count()似乎可以达到这个目的，并使其加速.但我仍然不明白为什么最初的查询速度较慢.

推荐答案

不幸的是，MySQL对子查询的支持非常糟糕，这对我们产生了非常负面的影响.SQLAlchemy docs点指出，可以使用query(func.count(Segment.id))实现"优化"查询:

返回此查询将返回的行数.

这将生成此查询的SQL，如下所示:

SELECT count(1) AS count_1 FROM (
     SELECT <rest of query follows...> ) AS anon_1

要对要计数的特定列进行细粒度控制，请跳过

from sqlalchemy import func

# count User records, without
# using a subquery.
session.query(func.count(User.id))

# return count of user "id" grouped
# by "name"
session.query(func.count(User.id)).\
        group_by(User.name)

from sqlalchemy import distinct

# count distinct "name" values
session.query(func.count(distinct(User.name)))