這次帶給大家Django多重資料庫使用步奏詳解，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
這樣app_label為auth的Model讀寫都在auth_db中完成，允許有關聯，migrate只在auth_db資料庫中可以運作。
其餘的：
import random
class PrimaryReplicaRouter(object):
def db_for_read(self, model, **hints):
"""
Reads go to a randomly-chosen replica.
"""
return random.choice(['replica1', 'replica2'])
def db_for_write(self, model, **hints):
"""
Writes always go to primary.
"""
return 'primary'
def allow_relation(self, obj1, obj2, **hints):
"""
Relations between objects are allowed if both objects are
in the primary/replica pool.
"""
db_list = ('primary', 'replica1', 'replica2')
if obj1._state.db in db_list and obj2._state.db in db_list:
return True
return None
def allow_migrate(self, db, app_label, model_name=None, **hints):
"""
All non-auth models end up in this pool.
"""
return True
這樣讀取在隨機在replica1和replica2中完成，寫入使用primary。
最後在settings中設定：
DATABASE_ROUTERS = ['path.to.AuthRouter', 'path.to.PrimaryReplicaRouter']
就可以了。
進行migrate操作時：
$ ./manage.py migrate $ ./manage.py migrate --database=users
migrate操作預設對default資料庫進行操作，並且要對其它資料庫進行操作，可以使用--database選項，後面為資料庫的別名。
與此對應的，dbshell，dumpdata，loaddata指令都有--database選項。
也可以手動的選擇路由：
查詢：
>>> # This will run on the 'default' database.
>>> Author.objects.all()
>>> # So will this.
>>> Author.objects.using('default').all()
>>> # This will run on the 'other' database.
>>> Author.objects.using('other').all()
儲存：
>>> my_object.save(using='legacy_users')
移動：
>>> p = Person(name='Fred') >>> p.save(using='first') # (statement 1) >>> p.save(using='second') # (statement 2)
以上的程式碼會產生問題，當p在first資料庫中第一次儲存時，會預設產生一個主鍵，這樣使用second資料庫儲存時，p已經有了主鍵，這個主鍵如果未被使用不會產生問題，但如果先前被使用了，就會覆蓋原先的資料。
有兩個解決方法;
1.儲存前清除主鍵：
>>> p = Person(name='Fred') >>> p.save(using='first') >>> p.pk = None # Clear the primary key. >>> p.save(using='second') # Write a completely new object.
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() # will delete from the `legacy_users` database
如果你想把一個物件從legacy_users資料庫轉移到new_users資料庫：
>>> user_obj.save(using='new_users') >>> user_obj.delete(using='legacy_users')
相信看了本文案例你已經掌握了方法，更多精彩請關注php中文網其它相關文章！
推薦閱讀：
#以上是Django多資料庫使用步奏詳解的詳細內容。更多資訊請關注PHP中文網其他相關文章！