如何限制GeoDjango+Postgis+django-rest-framework查询的返回条目数量

解决方案 #

方案1：使用DRF原生分页功能（最推荐） #

DRF的分页机制会在过滤逻辑执行完成后自动对结果做切片，不需要修改InBBoxFilter的源码，适配性最好：

1. 首先定义分页类，可根据前端承载能力调整每页最大返回量：

from rest_framework.pagination import PageNumberPagination

class MapMarkerPagination(PageNumberPagination):
    # 默认每页返回数量
    page_size = 1000
    # 允许前端通过参数自定义每页数量
    page_size_query_param = 'page_size'
    # 全局限制最大返回数量，避免恶意请求拉取全量数据
    max_page_size = 2000

2. 在视图集中引入分页类即可：

class MarkerViewSet(viewsets.ReadOnlyModelViewSet):
    bbox_filter_field = "geom"
    filter_backends = (filters.InBBoxFilter,)
    queryset = Tablename.objects.all()
    serializer_class = MarkerSerializer
    # 新增分页配置
    pagination_class = MapMarkerPagination

方案2：无需分页，固定限制最大返回条数 #

如果前端地图场景不需要分页，只需要保证单次请求最多返回N条数据，可以重写视图的list方法，在过滤完成后手动对结果集切片：

from rest_framework.response import Response

class MarkerViewSet(viewsets.ReadOnlyModelViewSet):
    bbox_filter_field = "geom"
    filter_backends = (filters.InBBoxFilter,)
    queryset = Tablename.objects.all()
    serializer_class = MarkerSerializer
    # 自定义单次请求最大返回条数
    MAX_RETURN_COUNT = 1500

    def list(self, request, *args, **kwargs):
        queryset = self.filter_queryset(self.get_queryset())
        # 过滤完成后切片，避免返回过多数据
        queryset = queryset[:self.MAX_RETURN_COUNT]
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)

大数据库场景性能优化建议 #

• 给空间字段加空间索引：在模型的geom字段定义中添加spatial_index=True参数，执行迁移后数据库会创建专用空间索引，可将BBox查询速度提升数十倍，是GIS场景的必备优化。
• 精简序列化字段：只返回前端渲染需要的字段，去掉无用的大文本、冗余字段，降低序列化内存占用和传输带宽消耗。
• 高频查询缓存：对相同BBox范围的查询结果做短时间缓存，降低重复查询的数据库压力。
• 超大规模数据可改用矢量瓦片方案：如果点位超过千万级，可改用矢量瓦片接口返回数据，按需返回当前视野瓦片范围内的简化点位，大幅降低前后端性能开销。

内容的提问来源于stack exchange，提问作者Chris Dixon