Django使用多数据库的方法

有些项目可能涉及到使用多个数据库的情况，方法很简单。

1.在settings中设定DATABASE

比如要使用两个数据库：

DATABASES = {  'default': {    'NAME': 'app_data',    'ENGINE': 'django.db.backends.postgresql',    'USER': 'postgres_user',    'PASSWORD': 's3krit'  },  'users': {    'NAME': 'user_data',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_user',    'PASSWORD': 'priv4te'  }}

这样就确定了2个数据库，别名一个为default，一个为user。数据库的别名可以任意确定。

default的别名比较特殊，一个Model在路由中没有特别选择时，默认使用default数据库。

当然，default也可以设置为空：

DATABASES = {  'default': {},  'users': {    'NAME': 'user_data',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_user',    'PASSWORD': 'superS3cret'  },  'customers': {    'NAME': 'customer_data',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_cust',    'PASSWORD': 'veryPriv@ate'  }}

这样，因为没有了默认的数据库，就需要为所有的Model，包括使用的第三方库中的Model做好数据库路由选择。

2.为需要做出数据库选择的Model规定app_label

class MyUser(models.Model):  ...  class Meta:    app_label = 'users'

3.写Database Routers

Database Router用来确定一个Model使用哪一个数据库，主要定义以下四个方法：

db_for_read(model, **hints)

规定model使用哪一个数据库读取。

db_for_write(model, **hints)

规定model使用哪一个数据库写入。

allow_relation(obj1, obj2, **hints)

确定obj1和obj2之间是否可以产生关联， 主要用于foreign key和 many to many操作。

allow_migrate(db, app_label, model_name=None, **hints)

确定migrate操作是否可以在别名为db的数据库上运行。

一个完整的例子：

数据库设定：

DATABASES = {  'default': {},  'auth_db': {    'NAME': 'auth_db',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_user',    'PASSWORD': 'swordfish',  },  'primary': {    'NAME': 'primary',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_user',    'PASSWORD': 'spam',  },  'replica1': {    'NAME': 'replica1',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_user',    'PASSWORD': 'eggs',  },  'replica2': {    'NAME': 'replica2',    'ENGINE': 'django.db.backends.mysql',    'USER': 'mysql_user',    'PASSWORD': 'bacon',  },}

如果想要达到如下效果：

app_label为auth的Model读写都在auth_db中完成，其余的Model写入在primary中完成，读取随机在replica1和replica2中完成。

auth：

class AuthRouter(object):  """  A router to control all database operations on models in the  auth application.  """  def db_for_read(self, model, **hints):    """    Attempts to read auth models go to auth_db.    """    if model._meta.app_label == 'auth':      return 'auth_db'    return None  def db_for_write(self, model, **hints):    """    Attempts to write auth models go to auth_db.    """    if model._meta.app_label == 'auth':      return 'auth_db'    return None  def allow_relation(self, obj1, obj2, **hints):    """    Allow relations if a model in the auth app is involved.    """    if obj1._meta.app_label == 'auth' or /      obj2._meta.app_label == 'auth':      return True    return None  def allow_migrate(self, db, app_label, model_name=None, **hints):    """    Make sure the auth app only appears in the 'auth_db'    database.    """    if app_label == 'auth':      return db == 'auth_db'    return None