PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

Django:IntegrityError durante Many To Many add()

È possibile riprodurre l'errore?

Sì, usiamo la famosa Publication e Article modelli da Documenti Django . Quindi, creiamo alcuni thread.

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

Sicuramente vedrai violazioni di vincoli di chiave univoci del tipo riportato nella segnalazione di bug . Se non li vedi, prova ad aumentare il numero di thread o iterazioni.

C'è una soluzione?

Sì. Usa through modelli e get_or_create . Ecco models.py adattato dall'esempio nei documenti di 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')

Ecco la nuova classe di threading che è una modifica di quella sopra.

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

Scoprirai che l'eccezione non viene più visualizzata. Sentiti libero di aumentare il numero di iterazioni. Sono arrivato solo a 1000 con get_or_create non ha generato l'eccezione. Tuttavia add() di solito generava un'eccezione entro 20 iterazioni.

Perché funziona?

Perché get_or_create è atomico.

Aggiornamento: Grazie @louis per aver sottolineato che il modello passante può infatti essere eliminato. Quindi get_or_create in MyThread2 può essere modificato come.

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