MySQL 使用SQLAlchemy ORM库来进行bulk insert操作

在使用MySQL数据库的过程中，有时需要向表中插入大量数据，此时单个插入显然效率过低，而MySQL提供了bulk insert的功能。而为了更方便地进行bulk insert操作，可以使用SQLAlchemy ORM库来实现。

阅读更多：MySQL 教程

什么是bulk insert

bulk insert即批量插入，它将多个数据行作为一个整体进行操作，能够提高数据插入操作的效率。在MySQL中，bulk insert通常使用LOAD DATA INFILE命令来实现。

如何使用SQLAlchemy ORM进行bulk insert

首先需要导入SQLAlchemy库，并进行数据库的连接和表的定义。假设我们有一个名为Person的表，包含name和age两个字段，其定义如下：

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Person(Base):
    __tablename__ = 'person'

    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String)
    age = Column(Integer)

接下来，我们可以使用session.execute方法执行bulk insert操作。假设我们要向Person表中插入10000个人员信息，可以使用以下代码：

session = Session()

persons = []
for i in range(10000):
   person = Person(name=f'Person{i}', age=i)
   persons.append(person)

session.execute(Person.__table__.insert(), [{"name": person.name, "age": person.age} for person in persons])

session.commit()
session.close()

上述代码中，首先创建了一个session对象，然后使用for循环创建了10000个人员信息，并添加到一个列表中。接着，使用session.execute方法执行bulk insert操作。其中，传入的第一个参数是SQLAlchemy构建的插入语句，通过Person.__table__.insert()获取。第二个参数是需要插入的数据，此处使用了列表推导式，将persons列表中每个人员的name和age属性以字典的形式组装成了一个列表。最后，使用session.commit()提交事务，关闭session。