PostgreSQL 全文搜索权重/搜索词优先级

时间:2021-05-05 22:35:45

标签: python django postgresql full-text-search

我通过 Django 在 PostgreSQL 中使用全文搜索。

我想将权重与搜索词相关联。 我知道可以将不同的权重与不同的字段相关联,但我想对搜索词设置不同的权重。

示例:

from core.models import SkillName
vector = SearchVector(
    "name",
)
search = SearchQuery("Java") | SearchQuery("Spring")
search_result = (
    SkillName.objects.all()
        .annotate(search=vector)
        .filter(search=search)
        .annotate(rank=SearchRank(vector, search))
        .order_by("-rank")
)
for s in search_result.distinct():
    print(f"{s} rank: {s.rank}")

现在我希望“Java”比“Spring”更重要并相应地获得排名。 我想我可以进行 2 次不同的搜索并将排名乘以因子,但有没有更好的方法?

想要将不同的优先级与搜索词关联起来真的很奇怪吗?

生成的 SQL 以供参考,老实说,我不认为现在在 Django 中这是可能的,我们可能需要 PostgreSQL-guru 的帮助。

SELECT DISTINCT "core_skillname"."id",
                "core_skillname"."name",
                to_tsvector(COALESCE("core_skillname"."name", '')) AS "search",
                ts_rank(to_tsvector(COALESCE("core_skillname"."name", '')), (plainto_tsquery('Java') || plainto_tsquery('Spring'))) AS "rank"
FROM "core_skillname"
WHERE to_tsvector(COALESCE("core_skillname"."name", '')) @@ (plainto_tsquery('Java') || plainto_tsquery('Spring'))
ORDER BY "rank" DESC;```

1 个答案:

答案 0 :(得分:1)

应用带有权重的排名不需要两个查询,只需在同一个查询中使用两个子表达式即可。

SELECT DISTINCT "core_skillname"."id",
                "core_skillname"."name",
                to_tsvector(COALESCE("core_skillname"."name", '')) AS "search",
                ts_rank(to_tsvector(COALESCE("core_skillname"."name", '')), plainto_tsquery('Spring')) +
                ts_rank(to_tsvector(COALESCE("core_skillname"."name", '')), plainto_tsquery('Java')) * 1.5 AS "rank"
FROM "core_skillname"
WHERE to_tsvector(COALESCE("core_skillname"."name", '')) @@ (plainto_tsquery('Java') || plainto_tsquery('Spring'))
ORDER BY "rank" DESC;

既然这样很容易抓挠自己的痒,为什么还要发明其他机制来做到这一点?当权重是表的一部分,而不是查询的一部分时,你真的不能这样做,所以它自己的机制更有意义。