基于
Lucene
检索引擎我们开发了自己的全文检索系统,承担起后台
PB
级、万亿条数据记录的检索工作,这里向大家分享下
Lucene
底层原理研究和一些优化经验。
从两个方面介绍:
1. Lucene
简介和索引原理
2. Lucene
优化经验总结
该部分从三方面展开: Lucene 简介、索引原理、 Lucene 索引实现。
Lucene
最初由鼎鼎大名
Doug Cutting
开发,
2000
年开源,现在也是开源全文检索方案的不二选择,它的特点概述起来就是:全
Java
实现、开源、高性能、功能完整、易拓展,功能完整体现在对分词的支持、各种查询方式(前缀、模糊、正则等)、打分高亮、列式存储(
DocValues
)等等。
而且
Lucene
虽已发展
10
余年,但仍保持着一个活跃的开发度,以适应着日益增长的数据分析需求,最新的
6.0
版本里引入
block k-d trees
,全面提升了数字类型和地理位置信息的检索性能,另基于
Lucene
的
Solr
和
ElasticSearch
分布式检索分析系统也发展地如火如荼,
ElasticSearch
也在我们项目中有所应用。
结合代码说明一下四个步骤:
Document doc=new Document( new StringField("name", "Donald Trump", Field.Store.YES)); //构建索引文档
iw.addDocument(doc); //做索引库
IndexReader reader = DirectoryReader.open(FSDirectory.open(new File(index)));
IndexSearcher searcher = new IndexSearcher(reader); //打开索引
Query query = parser.parse("name:trump");//解析查询
TopDocs results =searcher.search(query, 100);//检索并取回前100个文档号
for(ScoreDoc hit:results.hits)
{
Document doc=searcher .doc(hit.doc)//真正取文档
}
使用起来很简单,但只有知道这背后的原理,才能更好地用好 Lucene ,后面将介绍通用检索原理和 Lucene 的实现细节。
全文检索技术由来已久,绝大多数都基于倒排索引来做,曾经也有过一些其他方案如文件指纹。倒排索引,顾名思义,它相反于一篇文章包含了哪些词,它从词出发,记载了这个词在哪些文档中出现过,由两部分组成 —— 词典和倒排表。
其中词典结构尤为重要,有很多种词典结构,各有各的优缺点,最简单如排序数组,通过二分查找来检索数据,更快的有哈希表,磁盘查找有 B 树、 B+ 树,但一个能支持 TB 级数据的倒排索引结构需要在时间和空间上有个平衡,下图列了一些常见词典的优缺点:
其中可用的有:
B+
树、跳跃表、
FST
B+
树:
mysql
的
InnoDB B+
数结构
优点:外存索引、可更新
缺点:空间大、速度不够快
跳跃表:
优点:结构简单、跳跃间隔、级数可控,Lucene3.0之前使用的也是跳跃表结构,后换成了FST,但跳跃表在Lucene其他地方还有应用如倒排表合并和文档号索引。
缺点:模糊查询支持不好
FST
Lucene
现在使用的索引结构
优点:内存占用率低,压缩率一般在3倍~20倍之间、模糊查询支持好、查询快
缺点:结构复杂、输入要求有序、更新不易
Lucene里有个FST的实现,从对外接口上看,它跟Map结构很相似,有查找,有迭代:
String inputs={"abc","abd","acf","acg"}; //keys
long outputs={1,3,5,7}; //values
FST<Long> fst=new FST<>();
for(int i=0;i<inputs.length;i++)
{
fst.add(inputs[i],outputs[i])
}
//get
Long value=fst.get("abd"); //得到3
//迭代
BytesRefFSTEnum<Long> iterator=new BytesRefFSTEnum<>(fst);
while(iterator.next!=null){...}
100万数据性能测试:
· 1
|
数据结构 |
HashMap |
TreeMap |
FST |
|
构建时间 (ms) |
185 |
500 |
1512 |
|
查询所有 key(ms) |
106 |
218 |
890 |
可以看出,
FST
性能基本跟
HaspMap
差距不大,但
FST
有个不可比拟的优势就是占用内存小,只有
HashMap10
分之一左右,这对大数据规模检索是至关重要的,毕竟速度再快放不进内存也是没用的。
因此一个合格的词典结构要求有:
1.
查询速度。
2.
内存占用。
3.
内存
+
磁盘结合。
后面我们将解析
Lucene
索引结构,重点从
Lucene
的
FST
实现特点来阐述这三点。
Lucene
现在采用的数据结构为
FST
,它的特点就是:
1
、词查找复杂度为
O(len(str))
2
、共享前缀、节省空间
3
、内存存放前缀索引、磁盘存放后缀词块
这跟我们前面说到的词典结构三要素是一致的:
1.
查询速度。
2.
内存占用。
3.
内存
+
磁盘结合。我们往索引库里插入四个单词
abd
、
abe
、
acf
、
acg,
看看它的索引文件内容。
tip
部分,每列一个
FST
索引,所以会有多个
FST
,每个
FST
存放前缀和后缀块指针,这里前缀就为
a
、
ab
、
ac
。
tim
里
面存放后缀块和词的其他
域名交易
信息如倒排表指针
、
TFDF
等,
doc
文件里就为每个单词的倒排表。
所以它的检索过程分为三个步骤:
1.
内存加载
tip
文件,通过
FST
匹配前缀找到后缀词块位置。
2.
根据词块位置,读取磁盘中
tim
文件中后缀块并找到后缀和相应的倒排表位置信息。
3.
根据倒排表位置去
doc
文件中加载倒排表。
这里就会有两个问题,第一就是前缀如何计算,第二就是后缀如何写磁盘并通过
FST
定位,下面将描述下
Lucene
构建
FST
过程
:
已知
FST
要求输入有序,所以
Lucene
会将解析出来的文档单词预先排序,然后构建
FST
,我们假设输入为
abd,abd,acf,acg
,那么整个构建过程如下:
2. 插入abe时,计算出前缀ab,但此时不知道后续还不会有其他以ab为前缀的词,所以此时无输出。
3. 插入acf时,因为是有序的,知道不会再有ab前缀的词了,这时就可以写tip和tim了,tim中写入后缀词块d、e和它们的倒排表位置ip_d,ip_e,tip中写入a,b和以ab为前缀的后缀词块位置(真实情况下会写入更多信息如词频等)。
4. 插入acg时,计算出和acf共享前缀ac,这时输入已经结束,所有数据写入磁盘。tim中写入后缀词块f、g和相对应的倒排表位置,tip中写入c和以ac为前缀的后缀词块位置。
以上是一个简化过程, Lucene 的 FST 实现的主要优化策略有:
2. 前缀计算基于byte,而不是char,这样可以减少后缀数,防止后缀数太多,影响性能。如对宇(e9 b8 a2)、守(e9 b8 a3)、安(e9 b8 a4)这三个汉字,FST构建出来,不是只有根节点,三个汉字为后缀,而是从unicode码出发,以e9、b8为前缀,a2、a3、a4为后缀,
倒排表就是文档号集合,但怎么存,怎么取也有很多讲究,
Lucene
现使用的倒排表结构叫
Frame of reference,
它主要有两个特点:
1.
数据压缩,可以看下图怎么将
6
个数字从原先的
24bytes
压缩到
7bytes
。
2.
跳跃表加速合并,因为布尔查询时,
and
和
or
操作都需要合并倒排表,这时就需要快速定位相同文档号,所以利用跳跃表来进行相同文档号查找。
这部分可参考
ElasticSearch
的一篇博客,里面有一些性能测试:
ElasticSearch
倒排表
正向文件指的就是原始文档,
Lucene
对原始文档也提供了存储功能,它存储特点就是分块
+
压缩,
fdt
文件就是存放原始文档的文件,它占了索引库
90%
的磁盘空间,
fdx
文件为索引文件,通过文档号(自增数字)快速得到文档位置,它们的文件结构如下:
fnm
中为元信息存放了各列类型、列名、存储方式等信息。
fdt
为文档值,里面一个
chunk
就是一个块,
Lucene
索引文档时,先缓存文档,缓存大于
16KB
时,就会把文档压缩存储。一个
chunk
包含了该
chunk
起始文档、多少个文档、压缩后的文档内容。
fdx
为文档号索引,倒排表存放的时文档号,通过
fdx
才能快速定位到文档位置即
chunk
位置,它的索引结构比较简单,就是跳跃表结构,首先它会把
1024
个
chunk
归为一个
block,
每个
block
记载了起始文档值,
block
就相当于一级跳表。
所以查找文档,就分为三步:
第一步二分查找
block
,定位属于哪个
block
。
第二步就是根据从
block
里根据每个
chunk
的起始文档号,找到属于哪个
chunk
和
chunk
位置。
第三步就是去加载
fdt
的
chunk
,找到文档。这里还有一个细节就是存放
chunk
起始文档值和
chunk
位置不是简单的数组,而是采用了平均值压缩法。所以第
N
个
chunk
的起始文档值由
DocBase + AvgChunkDocs * n + DocBaseDeltas[n]
恢复而来,而第
N
个
chunk
再
fdt
中的位置由
StartPointerBase + AvgChunkSize * n + StartPointerDeltas[n]
恢复而来。
从上面分析可以看出,
lucene
对原始文件的存放是行是存储,并且为了提高空间利用率,是多文档一起压缩,因此取文档时需要读入和解压额外文档,因此取文档过程非常依赖随机
IO
,以及
lucene
虽然提供了取特定列,但从存储结构可以看出,并不会减少取文档时间。
我们知道倒排索引能够解决从词到文档的快速映射,但当我们需要对检索结果进行分类、排序、数学计算等聚合操作时需要文档号到值的快速映射,而原先不管是倒排索引还是行式存储的文档都无法满足要求。
原先
4.0
版本之前,
Lucene
实现这种需求是通过
FieldCache
,它的原理是通过按列逆转倒排表将(
field value ->doc
)映射变成(
doc -> field value
)映射
,
但这种实现方法有着两大显著问题:
1.
构建时间长。
2.
内存占用大,易
OutOfMemory
,且影响垃圾回收。
因此
4.0
版本后
Lucene
推出了
DocValues
来解决这一问题,它和
FieldCache
一样,都为列式存储,但它有如下优点:
1.
预先构建,写入文件。
2.
基于映射文件来做,脱离
JVM
堆内存,系统调度缺页。
DocValues
这种实现方法只比内存
FieldCache
慢大概
10~25%
,但稳定性却得到了极大提升。
Lucene
目前有五种类型的
DocValues
:
NUMERIC
、
BINARY
、
SORTED
、
SORTED_SET
、
SORTED_NUMERIC
,针对每种类型
Lucene
都有特定的压缩方法。
如对
NUMERIC
类型即数字类型,数字类型压缩方法很多,如:增量、表压缩、最大公约数,根据数据特征选取不同压缩方法。
SORTED
类型即字符串类型,压缩方法就是表压缩:预先对字符串字典排序分配数字
ID
,存储时只需存储字符串映射表,和数字数组即可,而这数字数组又可以采用
NUMERIC
压缩方法再压缩,图示如下:
这样就将原先的字符串数组变成数字数组,一是减少了空间,文件映射更有效率,二是原先变成访问方式变成固长访问。
对
DocValues
的应用,
ElasticSearch
功能实现地更系统、更完整,即
ElasticSearch
的
Aggregations——
聚合功能,它的聚合功能分为三类:
1. Metric ->
统计
典型功能:
sum
、
min
、
max
、
avg
、
cardinality
、
percent
等
2. Bucket ->
分组
典型功能:日期直方图,分组,地理位置分区
3. Pipline ->
基于聚合再聚合
典型功能:基于各分组的平均值求最大值。
基于这些聚合功能,
ElasticSearch
不再局限与检索,而能够回答如下
SQL
的问题
销售部门男女人数、平均年龄是多少
我们看下
ElasticSearch
如何基于倒排索引和
DocValues
实现上述
SQL
的。
1.
从倒排索引中找出销售部门的倒排表。
2.
根据倒排表去性别的
DocValues
里取出每个人对应的性别,并分组到
Female
和
Male
里。
3.
根据分组情况和年龄
DocValues
,计算各分组人数和平均年龄
4.
因为
ElasticSearch
是分区的,所以对每个分区的返回结果进行合并就是最终的结果。
上面就是
ElasticSearch
进行聚合的整体流程,也可以看出
ElasticSearch
做聚合的一个瓶颈就是最后一步的聚合只能单机聚合,也因此一些统计会有误差,比如
count(*) group by producet limit 5,
最终总数不是精确的。因为单点内存聚合,所以每个分区不可能返回所有分组统计信息,只能返回部分,汇总时就会导致最终结果不正确,具体如下:
原始数据:
|
Shard 1 |
Shard 2 |
Shard 3 |
|
Product A (25) |
Product A (30) |
Product A (45) |
|
Product B (18) |
Product B (25) |
Product C (44) |
|
Product C (6) |
Product F (17) |
Product Z (36) |
|
Product D (3) |
Product Z (16) |
Product G (30) |
|
Product E (2) |
Product G (15) |
Product E (29) |
|
Product F (2) |
Product H (14) |
Product H (28) |
|
Product G (2) |
Product I (10) |
Product Q (2) |
|
Product H (2) |
Product Q (6) |
Product D (1) |
|
Product I (1) |
Product J (8) |
|
|
Product J (1) |
Product C (4) |
|
count(*) group by producet limit 5 ,每个节点返回的数据如下:
|
Shard 1 |
Shard 2 |
Shard 3 |
|
Product A (25) |
Product A (30) |
Product A (45) |
|
Product B (18) |
Product B (25) |
Product C (44) |
|
Product C (6) |
Product F (17) |
Product Z (36) |
|
Product D (3) |
Product Z (16) |
Product G (30) |
|
Product E (2) |
Product G (15) |
Product E (29) |
合并后:
|
Merged |
|
Product A (100) |
|
Product Z (52) |
|
Product C (50) |
|
Product G (45) |
|
Product B (43) |
商品 A 的总数是对的,因为每个节点都返回了,但商品 C 在节点 2 因为排不到前 5 所以没有返回,因此总数是错的。
以上就是 Lucene 简介和底层原理分析,侧重于 Lucene 实现策略与特点,下一篇将介绍我们如何从这些底层原理出发来优化我们的全文检索系统。
来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/69992957/viewspace-2754878/ ,如需转载,请注明出处,否则将追究法律责任。