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

 

Django连接配置多个数据库

上一篇:跨端浪潮下前端开发者的机遇与挑战


下一篇:MySQL数据库自动补全命令的三种方式