Django中的原子操作？

Question

我想實現（我認爲是）一個櫃檯一個很簡單的數據模型：

class VisitorDayTypeCounter(models.Model): 
    visitType = models.CharField(max_length=60) 
    visitDate = models.DateField('Visit Date') 
    counter = models.IntegerField() 

當有人經過，它會尋找一個排的visitType和visitDate匹配;如果該行不存在，則會使用counter = 0創建它。

然後我們增加計數器並保存。

我擔心的是這個過程完全是一場比賽。兩個請求可以同時檢查，看看實體是否在那裏，並且他們都可以創建它。在讀取計數器並保存結果之間，可能會發出另一個請求並將其增加（導致計數丟失）。

到目前爲止，我還沒有找到一個很好的解決方法，無論是在Django文檔還是在教程中（實際上，它看起來像教程在它的投票部分有一個競爭條件）。

我該如何安全地做到這一點？

Answer 1

這是一個黑客。原始的SQL將使你的代碼更加便於移植，但它會擺脫計數器增加的競爭條件。理論上，每當你做查詢時，這應該增加計數器。我沒有測試過這個，所以你應該確保列表在查詢中正確插入。

class VisitorDayTypeCounterManager(models.Manager): 
    def get_query_set(self): 
     qs = super(VisitorDayTypeCounterManager, self).get_query_set() 

     from django.db import connection 
     cursor = connection.cursor() 

     pk_list = qs.values_list('id', flat=True) 
     cursor.execute('UPDATE table_name SET counter = counter + 1 WHERE id IN %s', [pk_list]) 

     return qs 

class VisitorDayTypeCounter(models.Model): 
    ... 

    objects = VisitorDayTypeCounterManager() 

Answer 2

兩個建議：

添加unique_together到模型，幷包裹創建一個異常處理程序來捕獲重複：

class VisitorDayTypeCounter(models.Model): 
    visitType = models.CharField(max_length=60) 
    visitDate = models.DateField('Visit Date') 
    counter = models.IntegerField() 
    class Meta: 
     unique_together = (('visitType', 'visitDate')) 

在此之後，你可以stlll對未成年人的比賽狀態計數器的更新。如果您獲得足夠的流量來關注這個問題，我會建議查看更細粒度數據庫控制的交易。我不認爲ORM直接支持鎖定/同步。交易文件可用here。

Answer 3

爲什麼不使用數據庫作爲併發層？將表的主鍵或唯一約束添加到visitType和visitDate。如果我沒有弄錯，django在他們的數據庫模型類中並不完全支持這個，或者至少我沒有看到過一個例子。

一旦你添加的約束/鍵的表，那麼所有你需要做的是：如果

1. 檢查行是存在的。如果是，取回它。
2. 插入該行。如果沒有錯誤，你很好，可以繼續前進。
3. 如果有錯誤（即競爭條件），請重新獲取該行。如果沒有行，那麼這是一個真正的錯誤。否則，你沒事。

這樣做很討厭，但它看起來足夠快，可以覆蓋大多數情況。

Answer 4

您應該使用數據庫事務來避免這種競爭條件。通過事務處理，您可以執行在「全部或全部」基礎上創建，讀取，遞增和保存計數器的整個操作。如果出現任何問題，它會回滾整個事情，你可以再試一次。

查看Django docs.有一個事務中間件，或者你可以在視圖或方法周圍使用裝飾器來創建事務。

Answer 5

如果您確實希望計數器準確無誤，您可以使用事務，但所需的併發量將真正拖累您的應用程序和數據庫在任何重要負載下下降。相反，考慮採用更多消息傳遞風格的方法，並將每次訪問的計數記錄存入表中，以便增加計數器。然後，當您希望總訪問次數在訪問表上進行計數時。您也可以有一個後臺進程，每天運行任意次數，將訪問總和，然後將其存儲在父表中。爲了節省空間，它也會刪除它總結的子訪問表中的任何記錄。如果您沒有多個代理爭奪相同資源（計數器），您將減少併發成本。

Answer 6

您可以使用http://code.djangoproject.com/ticket/2705的補丁來支持數據庫級鎖定。

隨着補丁此代碼將是原子：

visitors = VisitorDayTypeCounter.objects.get(day=curday).for_update() 
visitors.counter += 1 
visitors.save() 

Answer 7

由於Django的1.1，你可以使用ORM的F（）表達式。

from django.db.models import F 
product = Product.objects.get(name='Venezuelan Beaver Cheese') 
product.number_sold = F('number_sold') + 1 
product.save() 

欲瞭解更多詳細信息，請參閱文檔：

https://docs.djangoproject.com/en/1.8/ref/models/instances/#updating-attributes-based-on-existing-fields

https://docs.djangoproject.com/en/1.8/ref/models/expressions/#django.db.models.F