使用DRF Viewset分页时,如何避免order_by('?')导致重复对象
解决Django REST Framework Viewset分页出现重复对象的问题
问题根源是你在queryset中使用了order_by('?'),这个操作会让每次请求数据库时都重新执行随机排序。分页时,每一页的查询都是独立的随机排序,自然会出现不同页面重复展示同一个对象,甚至遗漏部分对象的情况。
以下是几种可行的解决方法:
方法一:使用固定种子的稳定随机排序(推荐)
通过固定随机种子,让同一用户或请求上下文的分页查询使用相同的随机顺序,避免每次请求重新打乱。
import random from django.db.models.functions import Random from rest_framework import viewsets class MyViewSet(viewsets.ModelViewSet): serializer_class = MySerializer pagination_class = StandardPagination def get_queryset(self): # 生成固定随机种子:可基于用户ID(登录状态)或固定值,确保分页顺序一致 seed = self.request.user.id if self.request.user.is_authenticated else 42 random.seed(seed) # 用带种子的Random函数实现稳定随机排序(PostgreSQL原生支持,MySQL需注意兼容性) return Model.objects.all().order_by(Random(seed))
方法二:缓存打乱后的ID列表(适合中小数据量)
先一次性获取所有对象ID并打乱,将结果存入缓存,分页时直接从缓存的ID列表中截取对应页码的ID,再查询数据库。这种方式能保证全部分页的顺序完全一致。
import random from django.core.cache import cache from rest_framework import viewsets class MyViewSet(viewsets.ModelViewSet): serializer_class = MySerializer pagination_class = StandardPagination def get_queryset(self): cache_key = "model_random_ids" random_ids = cache.get(cache_key) if not random_ids: # 获取所有对象ID并打乱顺序 all_ids = list(Model.objects.values_list('id', flat=True)) random.shuffle(all_ids) # 缓存1小时(可根据需求调整过期时间) cache.set(cache_key, random_ids, 3600) random_ids = all_ids # 通过分页器获取当前页对应的ID范围 paginator = self.pagination_class() page_ids = paginator.paginate_queryset(random_ids, self.request) # 根据ID查询对象并返回 return Model.objects.filter(id__in=page_ids).order_by('id')
方法三:内存中打乱查询集(仅适合小数据量)
如果数据量很小,可直接把所有对象加载到内存中打乱,后续分页基于这个已打乱的列表操作。注意:数据量大时会导致内存占用过高,不推荐。
import random from rest_framework import viewsets class MyViewSet(viewsets.ModelViewSet): serializer_class = MySerializer pagination_class = StandardPagination def get_queryset(self): queryset = list(Model.objects.all()) random.shuffle(queryset) return queryset
内容的提问来源于stack exchange,提问作者user19100277
相关产品推荐
相关产品推荐

