Django 验证后端多个数据库

Question

我正在重写一个遗留应用程序，该应用程序为每个客户提供一个数据库。每个客户都有自己的身份验证和用户集。因此，我需要一个自定义身份验证后端，因为 django 的身份验证设置为仅使用默认值。我编写了中间件，在每个请求时检查 url 并在那里提取信息以在请求上设置 database_name。

如果我在处理我的自定义身份验证后端期间可以访问请求，我可以轻松地执行数据库调用 user = User.objects.using(request.db).get(username=username) 但是，我认为没有简单的方法可以做到这一点。我在这里看到了答案:Access request.session from backend.get_user ，但这似乎不是线程安全的，所以我不想走那条路。

我能看到的唯一仍然使用 django-auth 的解决方案是为每个客户设置一个身份验证后端，将数据库名称设置为类属性。然后，我将创建一个自定义登录函数，将 request.session['_auth_user_backend'] 设置为客户特定的后端。因此，当对每个请求调用 get_user(user_id) 时，它使用知道从哪个数据库请求的客户后端。

如果可能，我希望避免为每个客户管理身份验证后端。有没有更好的方法来做到这一点？

Answer 1

由于 auth 后端没有调用 QuerySet 方法 using 您可以使用 database router与 thread local变量和一些中间件将变量设置为客户的数据库名称。中间件必须放在身份验证中间件之前。

线程局部变量是线程安全的。它创建一个线程局部全局变量。

如果您遵循请求的路径，它将执行以下操作:

1. 请求命中 django
2. 您的自定义中间件从 url 中获取数据库名称，并将其设置为线程局部全局变量。

3. django 身份验证中间件通过运行查询 User.object.get(id=user_id) 启动并设置用户。这将使用您的数据库路由器，它只会返回在之前的中间件中设置的线程局部全局变量。

4. 请求继续到 django 堆栈的其余部分。

例如，您有以下模块:

my_app/middleware.py

from threading import local

my_local_global = local()

class CustomerMiddleware(object):
    def process_request(self, request):
        my_local_global.database_name = get_database_name(request)

my_app/routers.py

from middleware import my_local_global

class MultiCustomerRouter(object):
    def db_for_read(self, model, **hints):
        return my_local_global.database_name

settings.py

...
MIDDLEWARE_CLASSES = (
 'django.middleware.common.CommonMiddleware',
 'django.contrib.sessions.middleware.SessionMiddleware',
 'django.middleware.csrf.CsrfViewMiddleware',
 'my_app.middleware.CustomerMiddleware',
 'django.contrib.auth.middleware.AuthenticationMiddleware',
 'django.contrib.messages.middleware.MessageMiddleware',
)

DATABASE_ROUTERS = ['my_app.routers.MultiCustomerRouter']
...