pg_trgm模块提供函数和操作符,用于基于三字符组匹配确定字母数字文本的相似度, 同时还提供支持快速搜索相似字符串的索引操作符类。
三字符组是一组从字符串中取出的三个连续字符。我们可以通过统计两个字符串共享的三字符组数量来度量它们的相似度。 这个简单的思想在度量许多自然语言中词的相似度时都非常有效。
从字符串中提取三字符组时,pg_trgm会忽略非词字符(即非字母数字字符)。 在确定字符串所包含的三字符组集合时,认为每个词前面都有两个空格,后面都有一个空格。 例如,字符串“cat”的三字符组集合是 “ c”、 “ ca”、 “cat”和 “at ”。 字符串“foo|bar”的三字符组集合是 “ f”、 “ fo”、 “foo”、 “oo ”、 “ b”、 “ ba”、 “bar”和 “ar ”。
pg_trgm模块提供的函数列在Table F.24中,操作符列在Table F.25中。
Table F.24. pg_trgm函数
考虑以下示例:
# SELECT word_similarity('word', 'two words');
word_similarity
-----------------
0.8
(1 row)
在第一个字符串中,三字符组集合为{" w"," wo","wor","ord","rd "}。在第二个字符串中,有序三字符组集合为{" t"," tw","two","wo "," w"," wo","wor","ord","rds","ds "}。第二个字符串的有序三字符组集合中最相似的连续区段是{" w"," wo","wor","ord"},相似度为0.8。
这个函数返回的值大致可以理解为第一个字符串与第二个字符串任意子串之间的最大相似度。 不过,该函数不会在这个区段的边界处添加填充。 因此,除了词边界不匹配的情况外,第二个字符串中额外存在的字符数不会被考虑在内。
同时,strict_word_similarity(text, text)会在第二个字符串中选择一个由完整单词组成的连续区段。在上面的示例中,strict_word_similarity(text, text)会选择仅包含单个单词的区段,这个单词是'words',其三字符组集合为{" w"," wo","wor","ord","rds","ds "}。
# SELECT strict_word_similarity('word', 'two words'), similarity('word', 'words');
strict_word_similarity | similarity
------------------------+------------
0.571429 | 0.571429
(1 row)
因此,strict_word_similarity(text, text)适合查找与整个词的相似度, 而word_similarity(text, text)更适合查找与词的一部分的相似度。
Table F.25. pg_trgm操作符
| 操作符 | 返回值 | 描述 |
|---|---|---|
text % text |
boolean |
如果参数之间的相似度大于pg_trgm.similarity_threshold设置的当前相似度阈值,则返回true。 |
text <% text |
boolean |
如果第一个参数中的三字符组集合与第二个参数中的有序三字符组集合某个连续区段之间的相似度大于 pg_trgm.word_similarity_threshold参数设置的当前词相似度阈值, 则返回true。 |
text %> text |
boolean |
<%操作符的交换子。 |
text <<% text |
boolean |
如果第二个参数中存在一个与词边界一致的有序三字符组集合连续区段,且它与第一个参数三字符组集合的相似度大于 pg_trgm.strict_word_similarity_threshold参数设置的当前严格词相似度阈值, 则返回true。 |
text %>> text |
boolean |
<<%操作符的交换子。 |
text <-> text |
real |
返回参数之间的“距离”,即 1 减去similarity()的值。 |
text <<-> text |
real |
返回参数之间的“距离”,即 1 减去word_similarity()的值。 |
text <->> text |
real |
<<->操作符的交换子。 |
text <<<-> text |
real |
返回参数之间的“距离”,即 1 减去strict_word_similarity()的值。 |
text <->>> text |
real |
<<<->操作符的交换子。 |
pg_trgm模块提供 GiST 和 GIN 索引操作符类,允许你为文本列创建索引,以实现非常快速的相似度搜索。这些索引类型支持上述相似度操作符,还支持对LIKE、ILIKE、~和~*查询执行基于三字符组的索引搜索。(这些索引不支持等值操作符或简单比较操作符,因此你可能还需要一个常规 B-树索引。)
示例:
CREATE TABLE test_trgm (t text); CREATE INDEX trgm_idx ON test_trgm USING GIST (t gist_trgm_ops);
或者
CREATE INDEX trgm_idx ON test_trgm USING GIN (t gin_trgm_ops);
此时,你已经在t列上有了一个可用于相似度搜索的索引。典型查询如下:
SELECT t, similarity(t, 'word') AS sml FROM test_trgm WHERE t % 'word' ORDER BY sml DESC, t;
这将返回文本列中所有与以下词足够相似的值:word,按从最佳匹配到最差匹配的顺序排序。即使在非常大的数据集上,索引也会让这一操作保持高效。
上述查询的一种变体是:
SELECT t, t <-> 'word' AS dist
FROM test_trgm
ORDER BY dist LIMIT 10;
GiST 索引可以相当高效地实现这一点,但 GIN 索引不能。当只需要少量最接近的匹配项时,它通常会优于第一种写法。
还可以使用t列上的索引进行单词相似度或严格单词相似度查询。典型查询如下:
SELECT t, word_similarity('word', t) AS sml
FROM test_trgm
WHERE 'word' <% t
ORDER BY sml DESC, t;
以及
SELECT t, strict_word_similarity('word', t) AS sml
FROM test_trgm
WHERE 'word' <<% t
ORDER BY sml DESC, t;
这会返回文本列中所有满足以下条件的值:在其对应的有序三字符组集合中,存在一个连续区段,与以下词的三字符组集合足够相似:word,按从最佳匹配到最差匹配的顺序排序。即使在非常大的数据集上,索引也会让这一操作保持高效。
上述查询的可能变体还有:
SELECT t, 'word' <<-> t AS dist
FROM test_trgm
ORDER BY dist LIMIT 10;
以及
SELECT t, 'word' <<<-> t AS dist
FROM test_trgm
ORDER BY dist LIMIT 10;
GiST 索引可以非常高效地实现这一查询,但 GIN 索引不能。
从PostgreSQL9.1 起,这些索引类型还支持以下操作的索引搜索:LIKE以及ILIKE,例如:
SELECT * FROM test_trgm WHERE t LIKE '%foo%bar';
索引搜索的工作方式是从搜索字符串中提取三字符组,然后在索引中查找这些三字符组。搜索字符串中包含的三字符组越多,索引搜索就越有效。与基于 B-树的搜索不同,搜索字符串不需要在左端锚定。
从PostgreSQL9.3 起,这些索引类型还支持正则表达式匹配的索引搜索(~以及~*操作符),例如:
SELECT * FROM test_trgm WHERE t ~ '(foo|bar)';
索引搜索的工作方式是从正则表达式中提取三字符组,然后在索引中查找这些三字符组。能从正则表达式中提取出的三字符组越多,索引搜索就越有效。与基于 B-树的搜索不同,搜索字符串不需要在左端锚定。
对于LIKE和正则表达式搜索,都要记住:无法提取出三字符组的模式会退化为全索引扫描。
GiST 和 GIN 索引之间如何取舍,取决于二者各自的相对性能特征;相关讨论见其他章节。
与全文索引结合使用时,三字符组匹配是非常有用的工具。 尤其是,它有助于识别那些因拼写错误而无法被全文搜索机制直接匹配的输入词。
第一步是生成一个辅助表,其中包含文档中的全部唯一词:
CREATE TABLE words AS SELECT word FROM
ts_stat('SELECT to_tsvector(''simple'', bodytext) FROM documents');
其中documents是一个表,包含我们希望搜索的文本字段bodytext。 之所以对to_tsvector函数使用simple配置,而不是使用特定语言的配置, 是因为我们需要原始的(未经词干提取的)词列表。
接下来,在词列上创建一个三字符组索引:
CREATE INDEX words_idx ON words USING GIN (word gin_trgm_ops);
现在,可以使用与前面示例类似的SELECT查询,为用户搜索词中拼错的单词提供拼写建议。 一个有用的附加测试是要求选出的词长度也与该拼错单词相近。
由于words表是作为一张独立的静态表生成的,因此需要定期重新生成, 以便与文档集合保持大致同步。 通常没有必要让它始终保持精确同步。
GiST 开发站点 http://www.sai.msu.su/~megera/postgres/gist/
Tsearch2 开发站点 http://www.sai.msu.su/~megera/postgres/gist/tsearch/V2/
Oleg Bartunov <oleg@sai.msu.su>,俄罗斯莫斯科,莫斯科大学
Teodor Sigaev <teodor@sigaev.ru>,俄罗斯莫斯科,Delta-Soft Ltd.
Alexander Korotkov <a.korotkov@postgrespro.ru>,俄罗斯莫斯科,Postgres Professional
文档:Christopher Kings-Lynne
该模块由俄罗斯莫斯科的 Delta-Soft Ltd. 赞助。