多租户Django应用:修改每个请求的数据库连接

我已经做了类似的事情，最接近第一点，但不是使用中间件来设置默认连接，而是使用Django数据库路由器。这允许应用程序逻辑在每个请求需要时使用多个数据库。为每个查询选择合适的数据库取决于应用程序逻辑，这是这种方法的一大缺点。

在此设置下，所有数据库都在settings.DATABASES中列出，包括可能在客户之间共享的数据库。每个特定于客户的模型都放在一个Django应用程序中，该应用程序有一个特定的应用标签。

。下面的类定义了一个存在于所有客户数据库中的模型。

class MyModel(Model):
    ....
    class Meta:
        app_label = 'customer_records'
        managed = False

一个数据库路由器被放置在settings.DATABASE_ROUTERS链中，以路由app_label的数据库请求，如下所示(不是完整的示例):

class AppLabelRouter(object):
    def get_customer_db(self, model):
        # Route models belonging to 'myapp' to the 'shared_db' database, irrespective
        # of customer.
        if model._meta.app_label == 'myapp':
            return 'shared_db'
        if model._meta.app_label == 'customer_records':
            customer_db = thread_local_data.current_customer_db()
            if customer_db is not None:
                return customer_db
            raise Exception("No customer database selected")
        return None
    def db_for_read(self, model, **hints):
        return self.get_customer_db(model, **hints)
    def db_for_write(self, model, **hints):
        return self.get_customer_db(model, **hints)

该路由器的特殊之处在于thread_local_data.current_customer_db()呼叫。在使用路由器之前，调用者/应用程序必须在thread_local_data中设置当前客户db。Python上下文管理器可用于此目的，以推送/弹出当前客户数据库。

配置好所有这些后，应用程序代码看起来像这样，其中UseCustomerDatabase是一个上下文管理器，将当前客户数据库名称推送/弹出到thread_local_data中，以便thread_local_data.current_customer_db()在最终到达路由器时返回正确的数据库名称:

class MyView(DetailView):
    def get_object(self):
        db_name = determine_customer_db_to_use(self.request) 
        with UseCustomerDatabase(db_name):
            return MyModel.object.get(pk=1)

这已经是相当复杂的设置了。它是有效的，但我将试着总结一下我所看到的优点和缺点:

• 数据库选择是灵活的。它允许在一个查询中使用多个数据库，客户特定的和共享的数据库都可以在一个请求中使用。
• 数据库选择是明确的(不确定这是优点还是缺点)。如果您尝试运行一个查询，命中客户数据库，但应用程序没有选择一个，将出现一个异常，指示编程错误。
• 使用数据库路由器允许不同的数据库存在于不同的主机上，而不是依靠USE db;语句猜测所有数据库都可以通过单个连接访问。