【Django】连接使用多个数据库

匿名 (未验证) 提交于 2019-12-02 23:36:01

第一步: 配置你要使用的数据库


setting.py 配置文件中的 DATABASES 字典中指定。
注意:默认的 default 数据库别名不可删除,如果不使用,可留空 {}

# 先定义好数据库别名 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 

注:
这里的 db_for_readdb_for_write 方法中的逻辑是根据app进行分库分表的,即不同的app使用不同的数据库。
如果想要读写分离的逻辑,可见文章补充部分。

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  # 在setting中定义的路由表 DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING  # 提前定义好的数据库别名 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  # 在setting中定义的路由表 DATABASE_MAPPING = settings.DATABASE_APPS_MAPPING  # 从库列表 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' 

当然你也可以不配置路由,直接使用第四步骤中的示例来实现读写分离。

文章来源: https://blog.csdn.net/qq_41964425/article/details/88742390
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!