Atomowy przyrost licznika w django

Question

Próbuję atomowo zwiększyć prosty licznik w Django. Mój kod wygląda następująco:

from models import Counter 
from django.db import transaction 

@transaction.commit_on_success 
def increment_counter(name): 
    counter = Counter.objects.get_or_create(name = name)[0] 
    counter.count += 1 
    counter.save() 

Jeśli dobrze zrozumiem Django, powinno to zawijać funkcję w transakcji i zwiększać ją atomowo. Ale to nie działa, aw aktualizacji licznika jest stan wyścigu. W jaki sposób ten kod można uczynić wątkiem bezpiecznym?

Answer 1

New in Django 1.1

Counter.objects.get_or_create(name = name) 
Counter.objects.filter(name = name).update(count = F('count')+1) 

lub używając an F expression:

counter = Counter.objects.get_or_create(name = name) 
counter.count = F('count') +1 
counter.save() 

Answer 2

Lub jeśli po prostu chcesz licznik, a nie trwałe obiektu można korzystać itertools licznik, który jest zaimplementowany w C. Gil zapewni potrzebne bezpieczeństwo.

--Sai

Answer 3

W Django 1.4 jest support for SELECT ... FOR UPDATE klauzule, używając blokady bazy danych, aby upewnić się, że żadne dane nie są jednocześnie dostęp przez pomyłkę.

Answer 4

ich prostota i budynek na użytkownika @ Oduvan odpowiedź:

counter, created = Counter.objects.get_or_create(name = name, 
               defaults={'count':1}) 
if not created: 
    counter.count = F('count') +1 
    counter.save() 

Zaletą jest to, że jeśli obiekt został stworzony w pierwszym piśmie, nie trzeba wykonywać żadnych dalszych aktualizacji.

Answer 5

Django 1,7

from django.db.models import F 

counter, created = Counter.objects.get_or_create(name = name) 
counter.count = F('count') +1 
counter.save() 

Answer 6

Jeśli nie trzeba znać wartość licznika gdy go ustawić górny odpowiedź jest zdecydowanie najlepszym:

counter = Counter.objects.get_or_create(name = name) 
counter.count = F('count') + 1 
counter.save() 

mówi to baza danych, aby dodać 1 do wartości count, która może doskonale działać bez blokowania innych operacji. Wadą jest to, że nie masz możliwości dowiedzenia się, co właśnie ustawiłeś. Jeśli dwa wątki jednocześnie uderzyłyby w tę funkcję, obaj zobaczyliby tę samą wartość i obaj powiedzieliby dbowi o dodaniu 1. Baza danych zakończy się dodaniem 2 zgodnie z oczekiwaniami, ale nie będzie wiadomo, która z nich poszła pierwsza.

Jeśli teraz dbasz o liczbę, możesz skorzystać z opcji select_for_update, do której odwołuje się Emil Stenstrom. Oto, jak to wygląda:

from models import Counter 
from django.db import transaction 

@transaction.atomic 
def increment_counter(name): 
    counter = (Counter.objects 
       .select_for_update() 
       .get_or_create(name=name)[0] 
    counter.count += 1 
    counter.save() 

To odczytuje bieżącą wartość i blokuje pasujące wiersze do końca transakcji. Teraz tylko jeden pracownik może czytać na raz. Więcej informacji na temat select_for_update można uzyskać pod numerem the docs.