Django连接配置多个数据库

一、配置你要使用的数据库

# 先定义好数据库别名
DB01 = 'db01'  # 第一个数据库别名
DB02 = 'db02'  # 第二个数据库别名

DATABASES = {
    'default': {},  # 不可删除，留空 {}
    DB01: {  # 第一个数据库
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'db01',
        'HOST': 'localhost',
        'POST': '3306',
        'USER': 'u1',
        'PASSWORD': 'user@u1',
    },
    DB02: {  # 第二个数据库
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'db02',
        'HOST': 'localhost',
        'POST': '3306',
        'USER': 'u2',
        'PASSWORD': 'user@u2',
    }
}

二、定义数据库路由方法类

使用多个数据库时，最简单的方法就是设置数据库路由方案，以保证对象对原始数据库的“粘性”，默认所有的查询都会返回到 default 数据库中。

数据库路由类方法详述：

1. db_for_read(model, **hints)：应用于读取类型对象的数据库模型，如果数据库提供附加信息会在hints字典中提供，最后如果没有则返回None。
2. db_for_write(model, **hints)：应用于写入类型对象的数据库模型，hints字典提供附加信息，如果没有则返回None。
3. allow_relation(obj1, obj2, **hints)：外键操作，判断两个对象之间是否是应该允许关系，是返回True,否则返回False，如果路由允许返回None。
4. allow_migrate(db, app_label, model_name=None, **hints)：db确定是否允许在具有别名的数据库上运行迁移操作，操作运行返回True，否则返回False，或者返回None，如果路由器没有意见。

位置参数解释：

• app_label：正在迁移的应用程序的标签。
• model_name：多个迁移操作设置模型的值，如：model._meta.app_label

------------↑

实现数据库路由的具体步骤如下

1、在 settings.py 文件中配置app与数据库匹配路由表，采用字典形式，app名对应数据库别名：

DB01 = 'db01'  # 第一个数据库别名
DB02 = 'db02'  # 第二个数据库别名

# 数据库路由表
DATABASE_APPS_MAPPING = {
    'app1': DB01,
    'app2': DB02,
}

2、在与 setting.py 同级的目录下创建路由表，app应用会根据上面指定的路由选择数据库。

# db_router文件内容如下

from django.conf import settings

DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING  # 在setting中定义的路由表


class DatabaseAppsRouter(object):
    def db_for_read(self, model, **hints):
        if model._meta.app_label in DATABASE_MAPPING:
            return DATABASE_MAPPING[model._meta.app_label]
        return None

    def db_for_write(self, model, **hints):
        if model._meta.app_label in DATABASE_MAPPING:
            return DATABASE_MAPPING[model._meta.app_label]
        return None

    def allow_relation(self, obj1, obj2, **hints):
        db_obj1 = DATABASE_MAPPING.get(obj1._meta.app_label)
        db_obj2 = DATABASE_MAPPING.get(obj2._meta.app_label)
        if db_obj1 and db_obj2:
            if db_obj1 == db_obj2:
                return True
            else:
                return False
        return None

    def allow_syncdb(self, db, model):
        if db in DATABASE_MAPPING.values():
            return DATABASE_MAPPING.get(model._meta.app_label) == db
        elif model._meta.app_label in DATABASE_MAPPING:
            return False
        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        if db in DATABASE_MAPPING.values():
            return DATABASE_MAPPING.get(app_label) == db
        elif app_label in DATABASE_MAPPING:
            return False
        return None

3、在 settings.py 文件中配置 DATABASE_ROUTERS 来指定路由表：

# 指定数据库路由表
DATABASE_ROUTERS = ['blog.db_router.DatabaseAppsRouter']
# blog：项目名称
# db_router：路由文件名称
# DatabaseAppRouter：路由文件中的路由类名称

三、指定数据库进行数据迁移

根据检测到的变化，创建新的迁移文件：

python manage.py makemigratins

将模型和迁移数据同步到指定的数据：

# 将应用程序下的模型根据路由迁移到对应的数据库中
python manage.py migrate --database=db01
python manage.py migrate --database=db02

可通过指定 app_label 来指定模型对应的数据库（例如在app1下指定某个模型将迁移到app2对应的数据库）。

class XX(models.Model):
    # xx...
    class Meta:
        app_label = 'app2'  # 如果指定，将在app02对应的数据库下创建数据表

四、使用指定的数据库来执行操作

配置完第二步骤中的数据库路由后，便可自动实现在某个app下使用某个数据库。例如，在 app1 中对数据库进行操作，将自动使用 db01 数据库；在 app2 中对数据库进行操作，将自动使用 db02 数据库。

当然你也可以手动指定数据库：

• 使用查询语句时，通过 using(db_alias) 来指定要操作的数据库。
• 保存时，使用 obj.save(using=db_alias) 来指定要操作的数据库。

from django.conf import settings

# 提前定义好的数据库别名
DB01 = settings.DB01  # 'db01'
Db02 = settings.DB02  # 'db02'

# 在 'db01' 数据库中添加数据
models.Md1.objects.using(DB01).create(username='u1')

# 在 'db02' 数据库中查询数据
models.Md1.objects.using(DB02).all()


# save方法
obj = models.Md1(username='u2')

# 保存到 'db01' 数据库
obj.save(using=DB01)

# 从 'db01' 数据库中删除数据
obj.delete(using=DB01)


# 清除查询到的数据的主键
# obj.pk = None

五、实现读写分离的路由方法

一主一从方案：

from django.conf import settings

# 提前定义好的数据库别名
DB01 = settings.DB01  # 'db01'
DB02 = settings.DB02  # 'db02'


class DatabaseAppsRouter(object):
    def db_for_read(self, model, **hints):
        """读操作"""
        return DB02  # 通过 'db02' 数据库

    def db_for_write(self, model, **hints):
        """写操作"""
        return DB01  # 通过 'db01' 数据库

一主多从方案：

import random
from django.conf import settings

# 从库列表
FOLLOW_DB_LIST = ['db02', 'db03', 'db04']


class DatabaseAppsRouter(object):
    def db_for_read(self, model, **hints):
        """读操作"""
        return random.choice(FOLLOW_DB_LIST)

    def db_for_write(self, model, **hints):
        """写操作"""
        return 'db01'