Django 分库教程：使用多个数据库提升性能

Django 支持多种数据库，包括'MySQL'、'PostgreSQL'、'SQLite'、'Oracle' 等。当需要处理大量数据时，单个数据库可能会遇到性能瓶颈，因此需要将数据分散到多个数据库中，这就是分库。

下面是一个简单的分库范例：

假设我们有一个电商网站，需要存储商品和订单信息。我们可以将商品信息存储在一个数据库中，将订单信息存储在另一个数据库中。

配置 settings.py 文件

首先，在 settings.py 文件中添加数据库配置信息。假设我们已经有了一个名为'products' 的数据库和一个名为'orders' 的数据库，那么我们可以这样配置：

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'products',
        'USER': 'your_username',
        'PASSWORD': 'your_password',
        'HOST': 'localhost',
        'PORT': '3306',
    },
    'orders': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'orders',
        'USER': 'your_username',
        'PASSWORD': 'your_password',
        'HOST': 'localhost',
        'PORT': '3306',
    }
}

这里我们定义了两个数据库：default 和 orders。default 是我们之前使用的数据库，存储商品信息。orders 是新的数据库，用于存储订单信息。

创建模型

接下来，我们需要创建模型。假设我们有两个模型：Product 和 Order，分别对应商品和订单。

class Product(models.Model):
    name = models.CharField(max_length=100)
    price = models.DecimalField(max_digits=5, decimal_places=2)
    description = models.TextField()

    class Meta:
        db_table = 'product'
        app_label = 'products'


class Order(models.Model):
    product_id = models.IntegerField()
    customer_name = models.CharField(max_length=100)
    customer_email = models.EmailField()
    quantity = models.IntegerField()

    class Meta:
        db_table = 'order'
        app_label = 'orders'

注意，每个模型都有一个 Meta 类，用于指定数据库表名和应用名称。在这里，我们将 Product 模型保存到'products' 数据库中，将 Order 模型保存到'orders' 数据库中。

创建路由

最后，我们需要创建路由，以便 Django 知道如何将模型映射到正确的数据库。在这里，我们将 Product 模型映射到'products' 数据库，将 Order 模型映射到'orders' 数据库。

class ProductRouter:
    def db_for_read(self, model, **hints):
        if model._meta.app_label == 'products':
            return 'default'
        return None

    def db_for_write(self, model, **hints):
        if model._meta.app_label == 'products':
            return 'default'
        return None

    def allow_relation(self, obj1, obj2, **hints):
        if obj1._meta.app_label == 'products' and obj2._meta.app_label == 'products':
            return True
        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        if app_label == 'products':
            return db == 'default'
        return None


class OrderRouter:
    def db_for_read(self, model, **hints):
        if model._meta.app_label == 'orders':
            return 'orders'
        return None

    def db_for_write(self, model, **hints):
        if model._meta.app_label == 'orders':
            return 'orders'
        return None

    def allow_relation(self, obj1, obj2, **hints):
        if obj1._meta.app_label == 'orders' and obj2._meta.app_label == 'orders':
            return True
        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        if app_label == 'orders':
            return db == 'orders'
        return None

这里我们创建了两个路由：ProductRouter 和 OrderRouter。每个路由都有四个方法：

db_for_read：返回读取数据所使用的数据库
db_for_write：返回写入数据所使用的数据库
allow_relation：判断两个对象是否可以关联
allow_migrate：判断模型是否可以在指定的数据库中进行迁移

在这里，我们将 Product 模型路由到'products' 数据库中，将 Order 模型路由到'orders' 数据库中。

注册路由

最后，在 settings.py 文件中注册路由。

DATABASE_ROUTERS = [
    'path.to.ProductRouter',
    'path.to.OrderRouter',
]

这里我们将 ProductRouter 和 OrderRouter 注册到 DATABASE_ROUTERS 中。

到此为止，我们已经完成了分库的设置。现在，我们可以使用模型来操作不同的数据库了。例如，我们可以这样查询所有的订单：

orders = Order.objects.using('orders').all()

总之，分库可以提高应用程序的性能和可伸缩性。Django 提供了灵活的机制来支持分库，可以根据具体的需求来进行配置。