Django配置數據庫讀寫分離

對網站的數據庫作讀寫分離（Read/Write Splitting）可以提高性能，在Django中對此提供了支持，下面我們來簡單看一下。注意，還需要運維人員作數據庫的讀寫分離和數據同步。

配置數據庫

我們知道在Django項目的settings中，可以配置數據庫，除了默認的數據庫，我在下面又加了一個db2。因爲是演示，我這裏用的是默認的SQLite，如果希望用MySQL，看這裏 。

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'),
    },
}

創建models並執行數據庫遷移

這裏我簡單創建一張產品表

from django.db import models


class Products(models.Model):
    """產品表"""
    prod_name = models.CharField(max_length=30)
    prod_price = models.DecimalField(max_digits=6, decimal_places=2)

創建完成後，執行數據庫遷移操作：

python manage.py makemigrations  # 在migrations文件夾下生成記錄，遷移前檢查
python manage.py migrate  # 創建表

在migrations文件夾下生成記錄，並在遷移前檢查是否有問題，默認值檢查defualt數據庫，但是可以在後面的數據庫路由類(Router)中通過allow_migrate()方法來指定是否檢查其它的數據庫。

其實第二步遷移默認有參數python manage.py migrate --database default ，在默認數據庫上創建表。因此完成以上遷移後，執行python manage.py --database db2，再遷移一次，就可以在db2上創建相同的表。這樣在項目根目錄下，就有了兩個表結構一樣的數據庫，分別是db.sqlite3和db2.sqlite3。

讀寫分離

手動讀寫分離

在使用數據庫時，通過.using(db_name)來手動指定要使用的數據庫

from django.shortcuts import HttpResponse
from . import models


def write(request):
    models.Products.objects.using('default').create(prod_name='熊貓公仔', prod_price=12.99)
    return HttpResponse('寫入成功')


def read(request):
    obj = models.Products.objects.filter(id=1).using('db2').first()
    return HttpResponse(obj.prod_name)

自動讀寫分離

通過配置數據庫路由，來自動實現，這樣就不需要每次讀寫都手動指定數據庫了。數據庫路由中提供了四個方法。這裏這裏主要用其中的兩個：def db_for_read()決定讀操作的數據庫，def db_for_write()決定寫操作的數據庫。

定義Router類

新建myrouter.py腳本，定義Router類：

class Router:
    def db_for_read(self, model, **hints):
        return 'db2'

    def db_for_write(self, model, **hints):
        return 'default'

配置Router

settings.py中指定DATABASE_ROUTERS

DATABASE_ROUTERS = ['myrouter.Router',]  

可以指定多個數據庫路由，比如對於讀操作，Django將會循環所有路由中的db_for_read()方法，直到其中一個有返回值，然後使用這個數據庫進行當前操作。

一主多從方案

網站的讀的性能通常更重要，因此，可以多配置幾個數據庫，並在讀取時，隨機選取，比如：

class Router:
    def db_for_read(self, model, **hints):
        """
        讀取時隨機選擇一個數據庫
        """
        import random
        return random.choice(['db2', 'db3', 'db4'])

    def db_for_write(self, model, **hints):
        """
        寫入時選擇主庫
        """
        return 'default'

分庫分表

在大型web項目中，常常會創建多個app來處理不同的業務，如果希望實現app之間的數據庫分離，比如app01走數據庫db1，app02走數據庫

class Router:
    def db_for_read(self, model, **hints):
        if model._meta.app_label == 'app01':
            return 'db1'
        if model._meta.app_label == 'app02':
            return 'db2'

    def db_for_write(self, model, **hints):
       if model._meta.app_label == 'app01':
            return 'db1'
       if model._meta.app_label == 'app02':
            return 'db2'