Django multidb：写入多个数据库

Question

使用 Django multidb，编写 路由器< /a> 运行主/从基础设施。但是是否可以编写一个可以写入多个数据库的路由器？我的用例是项目的集合，所有项目都在同一域上运行。为了避免用户在每个站点上注册/登录，我想同步 contrib.auth 和 contrib.sessions 表。 Django multidb 是否可以实现这一点，或者我应该研究数据库系统（在我的例子中为 MySQL）的复制功能？

Answer 1

我认为你会更好地实现 SSO 或 OAuth 服务

，但如果你想在两个数据库之间同步你的表用户，并且如果你使用你自己的 UserModel，你可以做这样的事情，

class MyUser(models.Model):
    name = models.CharField(max_length=100)
    user = models.ForeignKey(User, unique=True)


    def save(self, ...): # ALL the signature
        super(MyUser, self).save(using='database_1')
        super(MyUser, self).save(using='database_2')

你也可以使用这样的装饰器，比如您还可以使用它来同步其他表：

def save_multi_db(model_class):

    def save_wrapper(save_func): 
        def new_save(self, *args, **kws):
            super(model_class, self).save(using='database_1')
            super(model_class, self).save(using='database_1')
        return new_save

    func = getattr(model_class, 'save')
    setattr(model_class, 'save', save_wrapper(func)) 

    return save_wrapper

# and use it like this:

@save_multi_db
class MyUser(models.Model):
      ....

希望这会有所帮助:)

Answer 2

我现在正在研究 Django 分片架构。

我查看了 Django 路由器，但决定推出自己的路由器。

关于您的问题的一些想法：

1. 一种想法是使用一个数据库，然后通过在保存后使用 Django 信号复制适当的数据。

类似 -

import settings.databases as dbs_list

def post_save_function(UserModel):
     for db in dbs_list:
          UserModel.save(using=db,force_insert=True)

保存 User 对象（至少在单数据库模型上）似乎通过 django.contrib 内部发生的各种魔法在幕后保存会话、身份验证等数据，因此您可能不必这样做并找出所有这些数据库表的名称和类型。

为了支持这项工作的可能性，我发誓我最近在某个地方读到（可能是在 Alex Gaynor 的一篇博客文章中），如果一个对象有外键，Django 将尝试使用该对象所在的同一个数据库（出于明显的原因，根据 Django 通常的操作方式）。

1. 另一个想法：

从您引用的 Django multiDB 页面上的示例中我想知道类似以下内容是否可行：

他们的示例代码：

def allow_syncdb(self, db, model):
        "Explicitly put all models on all databases."
        return True

可能的修改：

def allowed_syncdb(self, db, model)：

    if isinstance(model,User):
         return True

    elif isinstance(model,Session):
         return True

    else:
         ''' something appropriate --
             whatever your sharding schema is for other objects '''

再看一遍，这段代码可能作为“db_for_write”函数更有用。但你明白了。

毫无疑问，您必须添加其他类型的模型才能使这项工作正常进行（所有身份验证内容，内容广泛）。

祝你好运！希望这在某种程度上有帮助。

我对您的发现和评论感兴趣！

新山

Answer 3

首先，我认为您需要更多的 SSO 框架，例如 在这篇文章中

我尝试了mouad的答案，但我无法让类装饰器工作......在我看来，这个解决方案不允许在模型中自定义 save() 。

更适合我的需要，我定义了一个自定义泛型类并简单地覆盖 save() 函数。

class MultiDbModel(models.Model):
    class Meta:
        abstract = True

    def save(self, *args, **kwargs):
        for dbname in settings.DATABASES:
            super(MultiDbModel, self).save(using=dbname)

进而：

class MyObject(MultiDbModel):
    [...]
    def save(self, *args, **kwargs):
        [custom save]
        super(MyObject, self).save(args, kwargs)

Answer 4

这是我的 multidb_model.py ，它似乎也可以处理外键。如果存在错误，将更新它，但是一旦您按照 Stepahne 的答案从此类继承，这将处理外键保存。

import logging

from django.db import models

DEFAULT_DB = 'default'  # main postgres host
MIRROR_COPY_DB = 'pg_mirror'  # a copy original db, i.e. if you want to move your data and keep it in sync


class MultiPgDbModel(models.Model):
    class Meta:
        abstract = True

    def save(self, *args, **kwarg):
        super(MultiPgDbModel, self).save(using=DEFAULT_DB)
        try:
            fields = [field for field in self._meta.fields if field.get_internal_type() == 'ForeignKey']

            for field in fields:
                getattr(self, field.name).save(using=MIRROR_COPY_DB)
            super(MultiPgDbModel, self).save(using=MIRROR_COPY_DB)
        except Exception as e:
            logging.exception(f"MultiPgDbModel.save unexpected error when saving to pg_mirror, object id "
                              f"{self.pk} of model {self._meta.model}. {e}")