缓慢保存到 Django 数据库

Question

我创建了一个自定义的manage.py 命令，如下所示：

from django.contrib.auth.models import User
from django.core.management.base import BaseCommand
from photos.models import Person


class Command(BaseCommand):
    help = 'Pre-populate database with initial data'

    def _create_people(self, user):
        for i in range(0, 100):
            person = Person(first_name='FN', surname='SN', added_by=user)
            person.save()

    def handle(self, *args, **options):
        user = User.objects.get(username="user1")
        self._create_people(user)

我已经计时了 handle() 的执行时间，如果我不执行 person.save() 大约需要 0.02 秒，如果我保存的话，每个 Person 大约需要 0.1 秒。数据库是 sqlite，我相信它应该更快。什么可以解释如此糟糕的性能以及如何改进它？

Answer 1

分析：

def _create_people(self, user):
   for i in range(0, 100):
     person = Person(first_name='FN', surname='SN', added_by=user)
     # Hits the database for each save.
     person.save()

这个函数会命中数据库100次，每次都自动提交，这就是导致性能低下的原因（不考虑sqlite与MySQL或PostgreSQL相比的低性能）。

改进：

您需要的是bulk_create，它将使用类构造函数创建的对象数组作为输入。所以一个可能的解决方案如下：

def _create_people(self, user):
   person_data = {"first_name": "FN", "surname":"SN", "added_by": user}
   person_list = [Person(**person_data) for i in range(100)]
   Person.objects.bulk_create(person_list)

默认情况下，bulk_create 无论有多少对象都会命中数据库一次，SQLite 除外（在 SQLite 中，每个查询大约 999 个。）。 可以通过指定单个查询中可以创建多少个对象 参数batch_size。

注意：

• save()不会被调用，相关信号也不会被发送。
• 不适用于 m2m 关系。