sql >> Base de Datos >  >> RDS >> PostgreSQL

Django:IntegrityError durante la adición de muchos a muchos ()

¿Se puede reproducir el error?

Sí, usemos la famosa Publication y Article modelos de Documentos de Django . Luego, creemos algunos hilos.

import threading
import random

def populate():

    for i in range(100):
        Article.objects.create(headline = 'headline{0}'.format(i))
        Publication.objects.create(title = 'title{0}'.format(i))

    print 'created objects'


class MyThread(threading.Thread):

    def run(self):
        for q in range(1,100):
            for i in range(1,5):
                pub = Publication.objects.all()[random.randint(1,2)]
                for j in range(1,5):
                    article = Article.objects.all()[random.randint(1,15)]
                    pub.article_set.add(article)

            print self.name


Article.objects.all().delete()
Publication.objects.all().delete()
populate()
thrd1 = MyThread()
thrd2 = MyThread()
thrd3 = MyThread()

thrd1.start()
thrd2.start()
thrd3.start()

Seguro que verá infracciones de restricciones de clave únicas del tipo informado en el informe de error . Si no los ve, intente aumentar el número de subprocesos o iteraciones.

¿Hay alguna solución alternativa?

Sí. Usar through modelos y get_or_create . Aquí está el models.py adaptado del ejemplo en los documentos de Django.

class Publication(models.Model):
    title = models.CharField(max_length=30)

    def __str__(self):              # __unicode__ on Python 2
        return self.title

    class Meta:
        ordering = ('title',)

class Article(models.Model):
    headline = models.CharField(max_length=100)
    publications = models.ManyToManyField(Publication, through='ArticlePublication')

    def __str__(self):              # __unicode__ on Python 2
        return self.headline

    class Meta:
        ordering = ('headline',)

class ArticlePublication(models.Model):
    article = models.ForeignKey('Article', on_delete=models.CASCADE)
    publication = models.ForeignKey('Publication', on_delete=models.CASCADE)
    class Meta:
        unique_together = ('article','publication')

Aquí está la nueva clase de subprocesos que es una modificación de la anterior.

class MyThread2(threading.Thread):

    def run(self):
        for q in range(1,100):
            for i in range(1,5):
                pub = Publication.objects.all()[random.randint(1,2)]
                for j in range(1,5):
                    article = Article.objects.all()[random.randint(1,15)]
                    ap , c = ArticlePublication.objects.get_or_create(article=article, publication=pub)
            print 'Get  or create', self.name

Encontrará que la excepción ya no aparece. Siéntase libre de aumentar el número de iteraciones. Solo subí a 1000 con get_or_create no arrojó la excepción. Sin embargo, add() por lo general lanzaba una excepción en 20 iteraciones.

¿Por qué funciona esto?

Porque get_or_create es atómico.

Actualización: Gracias @louis por señalar que, de hecho, el modelo pasante puede eliminarse. Por lo tanto, get_or_create en MyThread2 se puede cambiar como.

ap , c = article.publications.through.objects.get_or_create(
            article=article, publication=pub)