Django get_or_create 因 multiprocessing.Pool 失败

Question

这是我的项目设置，使用标准 Django startproject 命令和单个应用程序：

Python 3.5.1
Django 1.9.7
PostgreSQL 9.5.3
Ubuntu 16.04

应用的 models.py 定义了 2 个模型：

from django.db import models

class A(models.Model):
    n = models.PositiveIntegerField(primary_key=True)

class B(models.Model):
    a = models.ForeignKey(A, blank=True, null=True)
    m = models.CharField(max_length=20, db_index=True)

    class Meta:
        unique_together = ('a', 'm')

    def __str__(self):
        return '%s' % self.m

这是我的名为 execute.py 的管理命令，用于创建 B 的实例：

from multiprocessing import Pool
from django import db
from django.core.management.base import BaseCommand
from .models import B

M = 'abcdef'

def create():
    obj, created = B.objects.get_or_create(m=M, defaults={'a': None})
    if created:
        print('obj=%s' % obj)

class Command(BaseCommand):
    def handle(self, *args, **kwargs):
        B.objects.filter(m=M).delete()
        db.connections.close_all()
        n = 4
        pool = Pool(processes=n)
        results = []
        for _ in range(n):
            result = pool.apply_async(create)
            results.append(result)
        pool.close()
        for result in results:
            result.get()
        pool.join()

运行 python manage.py execute 会导致创建 4 个 obj，而不仅仅是 1 个。但情况并非总是如此。有时，只创建了 3 个甚至 1 个对象：

obj=abcdef
obj=abcdef
obj=abcdef
obj=abcdef

我是否遗漏了一些东西来强制这里的唯一性？

Answer 1

documentation for get_or_create() 中解释了此行为：

假设正确使用、正确的数据库配置和正确的底层数据库行为，此方法是原子的。 但是，如果在数据库级别没有强制唯一性 [...]，这种方法很容易出现竞争条件，这可能会导致多行具有相同的参数同时插入。

B 模型的 m 字段没有 unique=True 约束，这会导致竞争条件。将字段定义更改为：

m = models.CharField(max_length=20, db_index=True, unique=True)

你应该没事的。