使用 sqlalchemy 从 mysql 获取大量数据的最佳方法是什么？

Question

我想处理存储在 MySQL 中的超过 1000 万条数据。 所以我写这个是为了将 sql 分成几个部分，然后将数据连接到后面的过程中。 如果count < 2 millions ，则效果很好。 但是当count上升时，sqlalchemy 消耗的时间会更长。

def fetch_from_sql(_sql_pat, count):
    """
    :param _sql_pat: SELECT id, data FROM a.b LIMIT {},{};
    :param count: how many data you want to fetch from mysql
    :return: generator
    """
    def gen_connect(sql):
        __engine = create_engine(db_config['SQLALCHEMY_DATABASE_URI'])
        with __engine.connect() as c:
            for row in c.execute(sql)
                yield row

    def gen_range(limit, step):
        if step > limit:
            yield 0, limit
        else:
            R = range(0, limit + 1, step)
            for idx, v in enumerate(R):
                if idx == 0:
                    yield v, step
                elif limit - v >= step:
                    yield v + 1, step
                else:
                    yield v + 1, limit - v

    sqls = [_sql_pat.format(start, step) for start, step in gen_range(count, 100000)]
    sources = (gen_connect(sql) for sql in sqls)
    for s in sources:
        for item in s:
            yield item
        gc.collect()

问题是为什么 sqlalchemy 花费越来越多的时间（我记录了时间并在下面发布），以及处理这种情况的最佳方法是什么？

Dumped 10000 items, at 2016-10-08 11:55:33
Dumped 1000000 items, at 2016-10-08 11:59:23
Dumped 2000000 items, at 2016-10-08 12:05:07
Dumped 3000000 items, at 2016-10-08 13:54:05

Answer 1

这是因为您使用的是LIMIT / OFFSET ，因此当您指定偏移量 3000000 时，例如，数据库必须跳过 3000000 条记录。

执行此操作的正确方法是按某些索引列（例如主键id列）进行ORDER BY ，然后执行WHERE id > :last_fetched_id 。