Select и prefetch related: работа с БД для конкретной логики

В Django ORM есть много способов отойти от стандартных запросов, которые генерируются при запросах вида Book.objects.filter(category_id=category_id), и сделать работу с БД оптимальной для конкретной логики.

Среди подобных методов отдельное место занимают select_related и prefetch_related. Во-первых, в некоторых случаях они позволяют десятком символов снизить нагрузку на БД на порядок. Во-вторых, при неправильном их использовании можно убить приложение. А в-третьих, на собеседованиях на позицию начинающего веб-разработчика почти никто не может рассказать о разнице между ними.

Рассмотрим пример

class Book(Model):
    category = ForeignKey(Category)
    authors = ManyToManyField(Author)
…
book = Book.objects.get(pk=book_id)
print(book.category.name)
print([a.full_name for a in book.authors.all()])
print([a.pk for a in book.authors.all()])

Такой код сгенерирует четыре запроса: вытащит книгу, потом категорию, потом авторов, чтобы вывести их имена, потом опять авторов, чтобы вывести их номера. Было бы классно снизить количество запросов: авторов вытаскивать один раз, а категорию вытаскивать одним запросом с книгой.

Давайте начнём с категории

Тут нам пригодится select_related:

Book.objects.select_related(‘category’).get(pk=book_id)

за один запрос вытащит и книгу и категорию и тогда book.category.name уже не сгенерирует запроса.

Круто: мы сэкономили один запрос и не получили никаких штрафов. Хочется проделать нечто подобное с авторами, но увы: select_related(‘authors’) сделать не получится.

Всё дело в том, что тут отношение «многие ко многим», а в этом случае нельзя просто взять, сделать join и за один запрос вытащить и книгу, и всех авторов.

Тут-то и пригодится prefetch_related

Вот так:

Book.objects.prefetch_related(‘authors’).get(pk=book_id)

Здесь произойдёт два запроса: один вытащит книгу, а другой – всех авторов. Объединит их уже ORM на стороне Python. Теперь

print([a.full_name for a in book.authors.all()])

не сгенерирует ни одного запроса: всё уже есть.

Как видите, select_related и prefetch_related – мощные команды, которые работают совсем по-разному. Используйте их к месту, радуйте своих DBA и спите спокойно!

Есть вопрос? Напишите в комментариях!