Django REST Framework 高级搜索功能实现问题求助

你的问题出在默认的SearchQuery会将整个搜索字符串当作完整短语匹配，因此只有包含"how to"连续短语的帖子才会被返回。要实现"匹配任意单个词或完整短语"的需求，需要拆分搜索关键词并构建OR逻辑查询，同时保留短语匹配的权重优势。

修改后的代码 #

from django.contrib.postgres.search import SearchQuery, SearchVector, SearchRank, SearchHeadline

class PostsSearch(APIView):
    def get(self, request, format=None):
        search = request.GET.get('search', '').strip()
        if not search:
            return Response([])
        
        # 拆分搜索词为单个关键词，过滤空值
        keywords = [kw for kw in search.split() if kw]
        if not keywords:
            return Response([])
        
        # 构建单个关键词的OR查询
        keyword_queries = [SearchQuery(kw) for kw in keywords]
        combined_query = keyword_queries[0]
        for q in keyword_queries[1:]:
            combined_query |= q
        
        # 添加完整短语查询，让短语匹配的帖子排名更高
        phrase_query = SearchQuery(search)
        combined_query |= phrase_query
        
        # 包含所有需要搜索的字段：title、content、tags__name，设置权重优先级
        vector = (
            SearchVector('title', weight='A') +
            SearchVector('content', weight='B') +
            SearchVector('tags__name', weight='C')
        )
        
        # 查询并标注匹配关键词（可选，提升前端展示体验）
        posts = (
            Post.objects.prefetch_related('tags')
            .annotate(rank=SearchRank(vector, combined_query))
            .annotate(
                title_headline=SearchHeadline('title', combined_query),
                content_headline=SearchHeadline('content', combined_query)
            )
            .filter(rank__gte=0.0001)
            .order_by('-rank')
        )
        
        serializer = PostsSerializer(posts, many=True)
        return Response(serializer.data)

关键修改说明 #

• 拆分关键词构建OR逻辑：将搜索字符串拆分为单个词，每个词创建独立的SearchQuery，再用|（逻辑OR）组合，确保包含任意单个关键词的帖子都能被命中。
• 保留短语匹配权重：额外添加完整短语的查询，让包含"how to"连续短语的帖子获得更高的rank值，在结果中排名更靠前。
• 扩展搜索字段与权重：把content字段加入SearchVector，覆盖你提到的所有帖子字段；同时给不同字段设置权重（A> B> C），让标题匹配优先级高于内容，内容高于标签。
• 空搜索处理：增加对空搜索词的判断，避免程序报错。
• 可选关键词高亮：用SearchHeadline给匹配的关键词添加标记（默认是<b>标签），前端可直接渲染，提升用户体验。

额外优化建议 #

• 若数据量较大，建议使用SearchVectorField预先计算并存储搜索向量，避免每次请求重新计算，大幅提升查询性能。
• 可扩展支持排除词、精确匹配等复杂逻辑，只需在构建combined_query时添加对应的SearchQuery即可。

内容的提问来源于stack exchange，提问作者Sam Pearson