python - Django Rest Framework 分页计数极慢

Question

我在 Django Rest 框架中打开了分页，它看起来非常慢。 Count 看起来像是罪魁祸首，由于表中有数百万行，每次返回都需要数百毫秒。

我正在使用 postgresql 作为数据库。有没有办法不计算行数并仍然使用分页？如果我手动过滤查询集，则在启用此功能之前性能很好。

Answer 1

问题是，用于计数的查询与用于获取数据的查询可能很复杂。这样比较浪费。 PageNumberPagination 在内部使用 Django 自己的 Paginator。

为了使计数查询更简单，请重写 DRF 使用的分页器类:

from django.core.paginator import Paginator
from django.utils.functional import cached_property
from rest_framework.pagination import PageNumberPagination

class FasterDjangoPaginator(Paginator):
    @cached_property
    def count(self):
        # only select 'id' for counting, much cheaper
        return self.object_list.values('id').count()


class FasterPageNumberPagination(PageNumberPagination):
    django_paginator_class = FasterDjangoPaginator