如何优雅的全量读取Elasticsearch索引里面的数据 博客分类: ELK elasticsearch
程序员文章站
2024-03-18 19:28:52
...
### (一)scroll的介绍
有时候我们可能想要读取整个es索引的数据或者其中的大部分数据,来重建索引或者加工数据,相信大多数人都会说这很简单啊直接用from+size就能搞定,但实际情况是from+size的分页方法不适合用于这种全量数据的抽取,越到后面这种方法的性能就越低,这也是es里面为什么限制了单次查询结果的数据不能超过1万条数据的原因。
es里面提供了scroll的方式来全量读取索引数据其与数据库里面的游标(cursor)的概念非常类似,使用scroll读取数据的时候,只需要发送一次查询请求,然后es服务端会生成一个当前请求索引的快照数据集,接着我们每次通过scrollId来读取指定大小的批次数据,直到把整个索引的数据读取完毕。
这里面需要注意,当索引快照集生成的时候,其实在es内部维护了一个search context的上下文,这个上下文在指定的时间间隔内是只读的和不可变的,也就是只要它生成,那么后续你的添加,删除,更新操作的数据都不会被感知。
### (二)scroll的使用
下面看下如何使用:
(1)要使用scroll方式来读取数据,需要两步操作,第一步先做一个search context的初始化操作,如下命令:
```` curl -XGET 'localhost:9200/twitter/tweet/_search?scroll=1m' -d ' { "query": { "match" : { "title" : "elasticsearch" } } } ' ````
注意上面url里面的scroll=1m代表,这个search context只保留一分钟的有效期。
(2)在第一步操作里面我们能够获取一个scrollId,然后后面的每个读取都会得到一个scrollId,我们在读取next批次的数据要把这个scrollId回传,如下:
```` curl -XGET 'localhost:9200/_search/scroll' -d' { "scroll" : "1m", "scroll_id" : "c2Nhbjs2OzM0NDg1ODpzRlBLc0FXNlNyNm5JWUc1" } ' ````
或者通过search lite api的方式:
```` curl -XGET 'localhost:9200/_search/scroll?scroll=1m' -d 'c2Nhbjs2OzM0NDg1ODpzRlBLc0FXNlNyNm5JWUc1' ````
这样依次循环读取直到searchHits数组为空的情况下就代表数据读取完毕。
同理聚合的scroll请求,也是如此,但聚合请求的数据体只会在初始化的search里面存在,这一点需要注意,不过聚合请求的scroll一般没有这种应用场景,毕竟聚合后的结果一般都是少了好几个数量级的。
此外scroll请求还可以添加一个或多个排序字段,如果你读取的索引数据完全忽略它的顺序,那么我们还可以使用doc字段排序来提升性能。
```` curl -XGET 'localhost:9200/_search?scroll=1m' -d ' { "sort": [ "_doc" ] } ' ````
ok,再补充下再java api里面如何全量读取es索引数据的方法:
```` ` //指定一个index和type SearchRequestBuilder search = client.prepareSearch("active2018").setTypes("active"); //使用原生排序优化性能 search.addSort("_doc", SortOrder.ASC); //设置每批读取的数据量 search.setSize(100); //默认是查询所有 search.setQuery(QueryBuilders.queryStringQuery("*:*")); //设置 search context 维护1分钟的有效期 search.setScroll(TimeValue.timeValueMinutes(1)); //获得首次的查询结果 SearchResponse scrollResp=search.get(); //打印命中数量 System.out.println("命中总数量:"+scrollResp.getHits().getTotalHits()); //打印计数 int count=1; do { System.out.println("第"+count+"次打印数据:"); //读取结果集数据 for (SearchHit hit : scrollResp.getHits().getHits()) { System.out.println(hit.getSource()) ; } count++; //将scorllId循环传递 scrollResp = client.prepareSearchScroll(scrollResp.getScrollId()).setScroll(TimeValue.timeValueMinutes(1)).execute().actionGet(); //当searchHits的数组为空的时候结束循环,至此数据全部读取完毕 } while(scrollResp.getHits().getHits().length != 0); ````
### (三)删除无用的scroll
上文提到scroll请求时会维护一个search context快照集,这是如何做到的? 通过前面的几篇文章(点底部菜单栏可以看到),我们知道es在写入数据时,会在内存中不断的生成segment,然后有一个merge线程,会不断的合并小segment到更大的segment里面,然后再删除旧的segment,来减少es对系统资源的占用, 尤其是文件句柄,那么维护一个时间段内的索引快照,则意味着这段时间内的所有segment不能被合并,否则就破坏了快照的静态性,这样以来暂时不能被合并的小segment会占系统大量的文件句柄和系统资源,所以scroll的方式一定是离线使用的而不是提供给近实时使用的。
我们需要养成一个好习惯,当我们用完之后应该手动清除scroll,虽然search context超时也会自动清除。
es中提供了可以查看当前系统中有多少个open search context的api命令:
```` curl -XGET localhost:9200/_nodes/stats/indices/search?pretty ````
下面看下删除scrollId的方式
(1)删除一个scrollId
```` DELETE /_search/scroll { "scroll_id" : "UQlNsdDcwakFMNjU1QQ==" } ````
(2)删除多个scrollId
```` DELETE /_search/scroll { "scroll_id" : [ "aNmRMaUhiQlZkMWFB==", "qNmRMaUhiQlZkMWFB==" ] } ````
(3)删除所有的scrollId
```` DELETE /_search/scroll/_all ````
(4)search lite api的删除多个scrollId用法
```` DELETE /_search/scroll/aNmRMaUhiQlZkMWFB==,qNmRMaUhiQlZkMWFB== ````
上面的所有的功能在es2.3.4的版本中已经验证过,此外在es5.x之后的版本中,还增加了一个分片读取索引的功能,通过分片支持并行的读取方式,来提高导出效率:
一个例子如下:
```` GET /twitter/_search?scroll=1m { "slice": { "id": 0, "max": 2 }, "query": { "match" : { "title" : "elasticsearch" } } } GET /twitter/_search?scroll=1m { "slice": { "id": 1, "max": 2 }, "query": { "match" : { "title" : "elasticsearch" } } } ````
注意上面的slice参数,里面id字段代表当前读取的按个分片的数据,max参数代表我们将整个索引数据切分成分片的个数,默认的分片算法:
```` slice(doc) = floorMod(hashCode(doc._uid), max) ````
从上面能看到是基于uid字段的hashCode与分片的最大个数求模得出来的,注意floorMod方法与%求模在都是正整数的情况下结果是一样的。
slice字段还可以加入自定义的字段参与分片,比如基于日期字段:
````
"slice": {
"field": "date",
"id": 0,
"max": 10
}
````
参与分片的字段必须是数值字段并需要开启doc value,另外设置的max数量最好不要超过shard的个数,否则查询性能会下降,默认es对每个索引限制的最大分片量是1024,不过在setting里面通过设置index.max_slices_per_scroll参数改变。
### (四)总结
本篇文章介绍了如何优雅的全量读取es的索引数据以及它的一些原理和注意事项,了解这些有助于我们在日常工作中更好的使用es,从而提升我们对es的认知。
有什么问题可以扫码关注微信公众号:我是攻城师(woshigcs),在后台留言咨询。 技术债不能欠,健康债更不能欠, 求道之路,与君同行。