Django 中的线程同步

Question

有没有办法像Django中的Java同步一样阻塞关键区域？

Answer 1

您可以使用锁来确保一次只有一个线程可以访问特定的代码块。

为此，您只需创建一个 Lock 对象，然后在要同步的代码块之前获取锁。所有线程都必须有权访问同一个 Lock 对象才能正常工作。示例：

from threading import Lock, Thread

lock = Lock()

def do_something():
    lock.acquire()   # will block if another thread has lock
    try:
        # ... use lock
    finally:
        lock.release()

Thread(target=do_something).start()
Thread(target=do_something).start()

有关详细信息，请参阅 http://effbot.org/zone/thread-synchronization.htm。

Answer 2

我的方法是使用数据库的锁定功能。这也适用于多个服务器进程。

我将模型定义为：

from django.db import models

class ThreadSafe(models.Model):
    key = m.CharField(max_length=80, unique=True)

然后上下文管理器功能为：

from contextlib import contextmanager
from django.db.transaction import atomic

@contextmanager
def lock(key):
    pk = ThreadSafe.objects.get_or_create(key=key)[0].pk
    try:
        objs = ThreadSafe.objects.filter(pk=pk).select_for_update()
        with atomic():
            list(objs)
            yield None
    finally:
        pass

然后我只需执行以下操作即可获得线程/进程安全锁：

with lock("my_key"):
    do_scary_stuff_here()

这需要一个支持事务的数据库。

Answer 3

很棒的文章 Justin，只有一件事是使用 python 2.5 使这种方式变得更容易

在 Python 2.5 及更高版本中，您还可以使用 with 语句。当与锁一起使用时，该语句在进入块之前自动获取锁，并在离开块时释放锁：

from future import with_statement # 2.5 仅

使用锁：
...访问共享资源

Answer 4

如果您使用 PostgreSQL，则可以使用咨询锁。任何进程或线程都可以获取或释放此锁，假设它们都连接到同一个 PostgreSQL 数据库。 django-pglocks 采用这种方法。