Django ORM 事務和查詢優化

一、事務操作

模塊 from django.db import transaction

1 開啟事務：with transaction.atomic()

from django.db import transaction
class MyView(View):
    def post(self, request):
        # 在with代碼塊中開啟事務，出了with自動結束事務
        with transaction.atomic()
        	pass

開啟事務：@transaction.atomic

class MyView(View):
    @transaction.atomic
    def post(self, request):
        # 整個post方法都在事務中
        pass

2 設置事務回滾點

sid = transaction.savepoint() # 可以設置多個

事務回滾

# 回滾整個事務  但是不能在自動提交的事務中使用
transaction.rollback() 

# 回滾到sid
transaction.savepoint_rollback(sid)  

3 事務提交

# 提交整個事務  但是不能在自動提交的事務中使用
transaction.commit() 

# 釋放保存點 sid ，自保存點被創建依賴執行的更改成為事務的一部分
transaction.savepoint_commit(sid)

# atomic的事務，只有在事務結束後，才會自動提交到資料庫，進行數據的持久化

4 設置事務隔離

配置文件

import psycopg2.extensions

# 根據需要設置隔離級別,如下配置是READ_COMMITTED
# 但是配置PostgreSQL，不能配置mysql，否則會報錯
DATABASES = {
    # ...
    'OPTIONS': {
        'isolation_level': psycopg2.extensions.ISOLATION_LEVEL_READ_COMMITTED,
    },
}

二、 ORM 惰性查詢

即儘量減少對資料庫的操作，減輕資料庫的壓力

1 ORM惰性查詢

ORM 內所有的 SQL 查詢語句，都是惰性操作，即當你需要使用查詢結果時，才會真正去執行 SQL 語句訪問資料庫，否則是不會執行查詢的

可使用代碼驗證：

2 配置settings文件，使得當執行 SQL 語句時，在終端列印 SQL 命令

LOGGING = {
    'version': 1,
    'disable_existing_loggers': False,
    'handlers': {
        'console': {
            'level': 'DEBUG',
            'class': 'logging.StreamHandler',
        },
    },
    'loggers': {
        'django.db.backends': {
            'handlers': ['console'],
            'propagate': True,
            'level': 'DEBUG',
        },
    }
}

3 驗證

# 只查詢，不使用查詢結果
models.User.objects.all()

# 將查詢結果賦值給變數res，並使用res
res = models.User.objects.all()
print(res)

# 分別執行，查看終端的列印結果

三、only與defer

only 查詢

• 指定查詢某欄位時，將全部結果一次性獲取出來，以後再訪問該欄位數據時，直接從記憶體中拿即可，不需要再操作資料庫，返回QuerySet[obj1, obj2]

defer 查詢

• 指的是一次性獲取指定欄位之外的所有欄位數據，封裝到對象中，以後再訪問這些欄位數據時，直接從記憶體中拿即可，不需要再操作資料庫

class UserTest(models.Model):
    id = models.AutoField(primary_key = True)
    name = models.CharField(max_length = 32)
    age = models.IntegerField()
    def __str__(self):
        return 'name: %s, age: %s' %(self.name, self.age)

models.UserTest.objects.create(name = '小慧',age = '18')
models.UserTest.objects.create(name = '小嘉',age = '19')
models.UserTest.objects.create(name = '小甜',age = '17')
models.UserTest.objects.create(name = '小妞',age = '16')

# all 查詢
# 訪問一次資料庫，列印一條sql命令，獲取所有的 UserTest 表中的數據
res = models.UserTest.objects.all()  
print(res)

# only 查詢
# 訪問一次資料庫，列印一條sql命令，僅僅獲取 name 的所有數據到記憶體
res = models.UserTest.objects.only("name")  
print(res)

for user_obj in res:
    print(user_obj.name)  # 直接從記憶體中獲取，不訪問資料庫，不列印 SQl 命令
    print(user_obj.age)   # 每執行一次，就訪問一次資料庫，列印一次 SQl 命令
    
# defer 查詢
# 訪問一次資料庫，列印一條 SQl 命令，獲取除了 name 之外的所有數據到記憶體
res = models.User.objects.defer("name")  
for user_obj in res:
    print(user_obj.name)  # 每執行一次，訪問一次資料庫，列印一條 SQl 命令
    print(user_obj.age)   # 直接從記憶體中獲取，不訪問資料庫，不列印 SQl 命令

註意：分別執行才能看到效果 註釋print

四、select_related與prefetch_related

都支持正反向跨表查詢，規則等同於基於對象的正反向查詢，正向：'外鍵'，反向：'表名小寫_set'

select_related：指定外鍵，連表查詢（inner join），獲取所有數據到記憶體，不能指定多對多關係的外鍵

res = models.Book.objects.select_related('publish')

prefetch_related：指定外鍵，先查詢本表所有數據，再根據外鍵查詢對應的數據表所有數據，相當於子查詢，可以指定多個外鍵，即多個子查詢

res = models.Book.objects.prefetch_related('publish')