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")