Django连接多个数据库并实现读写分离

Posted 2022-12-28 wangyueping

当一个项目当中有大量的数据的时候，你所有的IO操作都在一个数据库中操作，会造成项目的性能的降低。如果你能对项目中的数据进行读写分离的话，那么将大大提高你项目的性能。而Django自带的机制也对此提供了支持。我们可以简单的操作一下。（当然数据的同步还是需要运维同志的协助）

修改配置文件

Django默认的是default,我们按照它的格式直接添加一个新的配置：

DATABASES = 
 'default': 
 'ENGINE': 'django.db.backends.sqlite3',
 'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
 ,
 'db2': 
 'ENGINE': 'django.db.backends.sqlite3',
 'NAME': os.path.join(BASE_DIR, 'db2.sqlite3'),
 

当然，如果你不想使用默认的sqlite3，想使用mysql还需要自己进行一下配置

DATABASES = 
 'default': 
 'ENGINE': 'django.db.backends.mysql',
 'HOST': '127.0.0.1', # 主服务器的运行ip
 'PORT': 3306, # 主服务器的运行port
 'USER': 'django', # 主服务器的用户名
 'PASSWORD': 'django', # 主服务器的密码
 'NAME': 'djangobase' # 数据表名
 ,
 'slave': 
 'ENGINE': 'django.db.backends.mysql', 
 'HOST': '127.0.0.1',
 'PORT': 8306,
 'USER': 'django_slave',
 'PASSWORD': 'django_slave',
 'NAME': 'djangobase_slave'
 
　　

在项目目录下的__init__文件中添加以下代码，将数据连接方式改为pymysql

import pymysql
pymysql.install_as_MySQLdb()

将数据配置修改完之后就可以在models.py文件当中创建表，接下里就可以进行数据库迁移了

python manage.py makemigrations # 在migrations文件夹下生成记录
python manage.py migrate --database default # 默认可以不写参数
python manage.py migrate --database db2 # 在从库再迁移一次，就可以在上面建立相同的表

手动读写分离

在遇到数据库相关操作的话，需要手动指定要使用的相应数据库，不需要进行多余的配置，但是，当你数据的读写操作过于频繁的时候，这个方法就会略显繁琐。

from django.shortcuts import render, HttpResponse
from app001 import models
# Create your views here.
def write(request):
 models.User.objects.using('default').create(name='张三', pwd='123', phone=1234)
 return HttpResponse('写成功')
def read(request):
 obj = models.User.objects.filter(id=1).using('db2').first()
 return HttpResponse('读成功')

自动读写分离

通过配置数据库路由，来自动实现读写分离，这样就不需要每次读写都手动指定数据库。

在项目的app中创建db_router.py文件,并在该文件中定义一个数据库路由类,用来进行读写分离，这个类最多提供四个方法，分别是：db_for_read、db_for_write、allow_relation、allow_migrate，以下只写了三个。

定义数据库路由类

class MasterSlaveDBRouter(object):
 """数据库主从读写分离路由"""
 
     def db_for_read(self, model, **hints):
     """读数据库"""
     return "slave"
 
     def db_for_write(self, model, **hints):
     """写数据库"""
     return "default"

     def allow_relation(self, obj1, obj2, **hints):
     """是否运行关联操作"""
     return True　　

配置Router

在settings.py中指定DATABASE_ROUTERS

DATABASE_ROUTERS = ["app001.db_router.MasterSlaveDBRouter"]

可以指定多个数据库路由，比如对于读操作，Django将会循环所有路由中的db_for_read()方法，直到其中一个有返回值，然后使用这个数据库进行当前操作。