Hoạt động nguyên tử ở Django?

Question

Tôi đang cố gắng để thực hiện (những gì tôi nghĩ là) một mô hình dữ liệu khá đơn giản đối với một bộ đếm:

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

Khi ai đó đi qua, nó sẽ tìm kiếm một hàng đó phù hợp với visitType và visitDate; nếu hàng này không tồn tại, nó sẽ được tạo với counter = 0.

Sau đó, chúng tôi tăng bộ đếm và lưu.

Mối quan tâm của tôi là quá trình này hoàn toàn là cuộc đua. Hai yêu cầu có thể đồng thời kiểm tra xem thực thể có ở đó không và cả hai đều có thể tạo ra nó. Giữa việc đọc bộ đếm và lưu kết quả, một yêu cầu khác có thể đi qua và tăng nó (dẫn đến số lượng mất).

Cho đến nay tôi chưa thực sự tìm thấy một cách tốt về điều này, hoặc trong tài liệu Django hoặc trong hướng dẫn (trên thực tế, có vẻ như hướng dẫn có điều kiện chủng tộc trong phần Bình chọn của nó).

Làm cách nào để thực hiện điều này một cách an toàn?

Answer 1

Đây là một chút hack. SQL thô sẽ làm cho mã của bạn kém di động hơn, nhưng nó sẽ loại bỏ tình trạng của cuộc đua khi tăng số lượt truy cập. Về lý thuyết, điều này sẽ tăng bộ đếm bất kỳ lúc nào bạn thực hiện truy vấn. Tôi đã không thử nghiệm này, vì vậy bạn nên chắc chắn rằng danh sách được nội suy trong truy vấn đúng cách.

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

Hai gợi ý:

Thêm một unique_together để mô hình của bạn, và quấn tạo trong xử lý ngoại lệ để đón các bản sao:

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

Sau này, bạn stlll có thể có một tình trạng chủng tộc nhỏ trên cập nhật của bộ đếm. Nếu bạn nhận được đủ lưu lượng truy cập để được quan tâm về điều đó, tôi sẽ đề nghị xem xét giao dịch để kiểm soát cơ sở dữ liệu hạt mịn hơn. Tôi không nghĩ rằng ORM có hỗ trợ trực tiếp cho khóa/đồng bộ hóa. Tài liệu giao dịch có sẵn here.

Answer 3

Tại sao không sử dụng cơ sở dữ liệu làm lớp đồng thời? Thêm khóa chính hoặc ràng buộc duy nhất bảng vào visitType và visitDate. Nếu tôi không nhầm, django không hỗ trợ chính xác điều này trong lớp Mô hình cơ sở dữ liệu của họ hoặc ít nhất tôi đã không nhìn thấy một ví dụ.

Một khi bạn đã thêm các hạn chế/chìa khóa để bảng, sau đó tất cả các bạn phải làm là:

1. kiểm tra nếu hàng là ở đó. nếu có, hãy lấy nó.
2. chèn hàng. nếu không có lỗi, bạn sẽ ổn và có thể tiếp tục.
3. nếu có lỗi (tức là điều kiện chủng tộc), tìm nạp lại hàng. nếu không có hàng, thì đó là lỗi chính hãng. Nếu không, bạn ổn.

Thật khó chịu khi thực hiện theo cách này, nhưng dường như đủ nhanh và sẽ bao gồm hầu hết các trường hợp.

Answer 4

Bạn nên sử dụng giao dịch cơ sở dữ liệu để tránh loại tình trạng chủng tộc này. Giao dịch cho phép bạn thực hiện toàn bộ hoạt động tạo, đọc, tăng và lưu bộ đếm trên cơ sở "tất cả hoặc không có gì". Nếu bất cứ điều gì sai, nó sẽ quay trở lại toàn bộ điều và bạn có thể thử lại.

Kiểm tra Django docs. Có một mặt hàng trung gian giao dịch hoặc bạn có thể sử dụng trang trí xung quanh lượt xem hoặc phương pháp để tạo giao dịch.

Answer 5

Nếu bạn thực sự muốn bộ đếm chính xác, bạn có thể sử dụng giao dịch nhưng số tiền đồng thời yêu cầu sẽ thực sự kéo ứng dụng và cơ sở dữ liệu của bạn xuống dưới bất kỳ tải trọng đáng kể nào.Thay vào đó, hãy nghĩ đến việc tiếp cận với một cách tiếp cận phong cách nhắn tin hơn và chỉ giữ các bản ghi bán phá giá vào một bảng cho mỗi lần truy cập mà bạn muốn tăng bộ đếm. Sau đó, khi bạn muốn tổng số lượt truy cập thực hiện đếm trên bảng lượt truy cập. Bạn cũng có thể có một quy trình nền chạy bất kỳ số lần nào trong một ngày có tổng số lượt truy cập và sau đó lưu trữ nó trong bảng cha. Để tiết kiệm không gian, nó cũng sẽ xóa bất kỳ bản ghi nào từ bảng truy cập con mà nó tóm tắt. Bạn sẽ cắt giảm chi phí đồng thời của bạn một số tiền rất lớn nếu bạn không có nhiều đại lý cạnh tranh cho các nguồn tài nguyên tương tự (bộ đếm).

Answer 6

Bạn có thể sử dụng bản vá từ http://code.djangoproject.com/ticket/2705 để khóa cấp cơ sở dữ liệu hỗ trợ.

Với vá mã này sẽ là nguyên tử:

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

Answer 7

Tính đến Django 1.1 bạn có thể sử dụng biểu thức của 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() 

Để biết thêm chi tiết hãy xem tài liệu:

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