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
classMyUser(models.Model): ... classMeta: app_label='users'
3.写DatabaseRouters
DatabaseRouter用来确定一个Model使用哪一个数据库，主要定义以下四个方法：
db_for_read(model,**hints)
规定model使用哪一个数据库读取。
db_for_write(model,**hints)
规定model使用哪一个数据库写入。
allow_relation(obj1,obj2,**hints)
确定obj1和obj2之间是否可以产生关联，主要用于foreignkey和manytomany操作。
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：
classAuthRouter(object): """ Aroutertocontrolalldatabaseoperationsonmodelsinthe authapplication. """ defdb_for_read(self,model,**hints): """ Attemptstoreadauthmodelsgotoauth_db. """ ifmodel._meta.app_label=='auth': return'auth_db' returnNone defdb_for_write(self,model,**hints): """ Attemptstowriteauthmodelsgotoauth_db. """ ifmodel._meta.app_label=='auth': return'auth_db' returnNone defallow_relation(self,obj1,obj2,**hints): """ Allowrelationsifamodelintheauthappisinvolved. """ ifobj1._meta.app_label=='auth'or\ obj2._meta.app_label=='auth': returnTrue returnNone defallow_migrate(self,db,app_label,model_name=None,**hints): """ Makesuretheauthapponlyappearsinthe'auth_db' database. """ ifapp_label=='auth': returndb=='auth_db' returnNone
这样app_label为auth的Model读写都在auth_db中完成，允许有关联，migrate只在auth_db数据库中可以运行。
其余的：
importrandom
classPrimaryReplicaRouter(object):
defdb_for_read(self,model,**hints):
"""
Readsgotoarandomly-chosenreplica.
"""
returnrandom.choice(['replica1','replica2'])
defdb_for_write(self,model,**hints):
"""
Writesalwaysgotoprimary.
"""
return'primary'
defallow_relation(self,obj1,obj2,**hints):
"""
Relationsbetweenobjectsareallowedifbothobjectsare
intheprimary/replicapool.
"""
db_list=('primary','replica1','replica2')
ifobj1._state.dbindb_listandobj2._state.dbindb_list:
returnTrue
returnNone
defallow_migrate(self,db,app_label,model_name=None,**hints):
"""
Allnon-authmodelsendupinthispool.
"""
returnTrue
这样读取在随机在replica1和replica2中完成，写入使用primary。
最后在settings中设定：
DATABASE_ROUTERS=['path.to.AuthRouter','path.to.PrimaryReplicaRouter']
就可以了。
进行migrate操作时：
$./manage.pymigrate $./manage.pymigrate--database=users
migrate操作默认对default数据库进行操作，要对其它数据库进行操作，可以使用--database选项，后面为数据库的别名。
与此相应的，dbshell，dumpdata，loaddata命令都有--database选项。
也可以手动的选择路由：
查询：
>>>#Thiswillrunonthe'default'database.
>>>Author.objects.all()
>>>#Sowillthis.
>>>Author.objects.using('default').all()
>>>#Thiswillrunonthe'other'database.
>>>Author.objects.using('other').all()
保存：
>>>my_object.save(using='legacy_users')
移动：
>>>p=Person(name='Fred') >>>p.save(using='first')#(statement1) >>>p.save(using='second')#(statement2)
以上的代码会产生问题，当p在first数据库中第一次保存时，会默认生成一个主键，这样使用second数据库保存时，p已经有了主键，这个主键如果未被使用不会产生问题，但如果先前被使用了，就会覆盖原先的数据。
有两个解决方法;
1.保存前清除主键：
>>>p=Person(name='Fred') >>>p.save(using='first') >>>p.pk=None#Cleartheprimarykey. >>>p.save(using='second')#Writeacompletelynewobject.
2.使用force_insert
>>>p=Person(name='Fred') >>>p.save(using='first') >>>p.save(using='second',force_insert=True)
删除：
从哪个数据库取得的对象，从哪删除
>>>u=User.objects.using('legacy_users').get(username='fred')
>>>u.delete()#willdeletefromthe`legacy_users`database
如果想把一个对象从legacy_users数据库转移到new_users数据库：
>>>user_obj.save(using='new_users') >>>user_obj.delete(using='legacy_users')
总结
以上所述是小编给大家介绍的Django使用多数据库的方法，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对毛票票网站的支持！