PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Django :erreur d'intégrité lors de l'ajout de plusieurs à plusieurs ()

L'erreur peut-elle être reproduite ?

Oui, utilisons la fameuse Publication et Article modèles de docs Django . Ensuite, créons quelques discussions.

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()

Vous êtes sûr de voir des violations de contrainte de clé unique du type signalé dans le rapport de bogue . Si vous ne les voyez pas, essayez d'augmenter le nombre de threads ou d'itérations.

Existe-t-il une solution ?

Oui. Utilisez through modèles et get_or_create . Voici le models.py adapté de l'exemple dans les docs 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')

Voici la nouvelle classe de threading qui est une modification de celle ci-dessus.

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

Vous constaterez que l'exception n'apparaît plus. N'hésitez pas à augmenter le nombre d'itérations. Je suis seulement allé jusqu'à 1000 avec get_or_create il n'a pas levé l'exception. Cependant add() lançait généralement une exception avec en 20 itérations.

Pourquoi ça marche ?

Parce que get_or_create est atomique.

Mise à jour : Merci @louis d'avoir souligné que le modèle traversant peut en fait être éliminé. Utilisez donc le get_or_create dans MyThread2 peut être changé comme.

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