feat: elasticsearch 6.x 示例

Author: dunwu

codes/javadb/elasticsearch/elasticsearch6/src/main/java/io/github/dunwu/javadb/elasticsearch/ElasticsearchTemplate.java

@@ -15,6 +15,7 @@
 import org.elasticsearch.action.DocWriteResponse;
 import org.elasticsearch.action.admin.indices.alias.Alias;
 import org.elasticsearch.action.admin.indices.alias.IndicesAliasesRequest;
+import org.elasticsearch.action.admin.indices.alias.get.GetAliasesRequest;
 import org.elasticsearch.action.admin.indices.create.CreateIndexRequest;
 import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
 import org.elasticsearch.action.admin.indices.get.GetIndexRequest;
@@ -40,8 +41,10 @@
 import org.elasticsearch.action.support.master.AcknowledgedResponse;
 import org.elasticsearch.action.update.UpdateRequest;
 import org.elasticsearch.action.update.UpdateResponse;
+import org.elasticsearch.client.GetAliasesResponse;
 import org.elasticsearch.client.RequestOptions;
 import org.elasticsearch.client.RestHighLevelClient;
+import org.elasticsearch.cluster.metadata.AliasMetaData;
 import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.common.unit.ByteSizeUnit;
 import org.elasticsearch.common.unit.ByteSizeValue;
@@ -65,6 +68,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
+import java.util.Set;
 import java.util.concurrent.TimeUnit;
 import java.util.function.BiConsumer;
 import java.util.stream.Collectors;
@@ -194,6 +198,20 @@ public boolean isIndexExists(String index) throws IOException {
         return client.indices().exists(request.indices(index), RequestOptions.DEFAULT);
     }
 
+    public Set<String> getIndexSet(String alias) throws IOException {
+        GetAliasesRequest request = new GetAliasesRequest(alias);
+        GetAliasesResponse response = client.indices().getAlias(request, RequestOptions.DEFAULT);
+        if (StrUtil.isNotBlank(response.getError())) {
+            String msg = StrUtil.format("【ES】获取索引失败！alias: {}, error: {}", alias, response.getError());
+            throw new ElasticsearchException(msg);
+        }
+        if (response.getException() != null) {
+            throw response.getException();
+        }
+        Map<String, Set<AliasMetaData>> aliasMap = response.getAliases();
+        return aliasMap.keySet();
+    }
+
     public void setMapping(String index, String type, Map<String, String> propertiesMap) throws IOException {
 
         if (MapUtil.isEmpty(propertiesMap)) {
@@ -459,7 +477,7 @@ public <T> List<T> pojoListByIds(String index, String type, Collection<String> i
         throws IOException {
 
         if (CollectionUtil.isEmpty(ids)) {
-            return null;
+            return new ArrayList<>(0);
         }
 
         MultiGetRequest request = new MultiGetRequest();
@@ -471,7 +489,7 @@ public <T> List<T> pojoListByIds(String index, String type, Collection<String> i
         if (null == multiGetResponse
             || multiGetResponse.getResponses() == null
             || multiGetResponse.getResponses().length <= 0) {
-            return new ArrayList<>();
+            return new ArrayList<>(0);
         }
 
         List<T> list = new ArrayList<>();
@@ -491,7 +509,7 @@ public <T> List<T> pojoListByIds(String index, String type, Collection<String> i
     public long count(String index, String type, SearchSourceBuilder builder) throws IOException {
         SearchResponse response = query(index, type, builder);
         if (response == null || response.status() != RestStatus.OK) {
-            return -1L;
+            return 0L;
         }
         SearchHits searchHits = response.getHits();
         return searchHits.getTotalHits();
@@ -550,15 +568,15 @@ public <T> PageData<T> pojoPage(String index, String type, int from, int size, Q
     /**
      * search after 分页
      */
-    public <T extends BaseEsEntity> ScrollData<T> pojoPageByLastId(String index, String type, String lastId, int size,
+    public <T extends BaseEsEntity> ScrollData<T> pojoPageByScrollId(String index, String type, String scrollId, int size,
         QueryBuilder queryBuilder, Class<T> clazz) throws IOException {
 
         SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
         searchSourceBuilder.size(size);
         searchSourceBuilder.sort(BaseEsEntity.DOC_ID, SortOrder.ASC);
-        if (StrUtil.isNotBlank(lastId)) {
+        if (StrUtil.isNotBlank(scrollId)) {
             BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
-            boolQueryBuilder.must(queryBuilder).must(QueryBuilders.rangeQuery(BaseEsEntity.DOC_ID).gt(lastId));
+            boolQueryBuilder.must(queryBuilder).must(QueryBuilders.rangeQuery(BaseEsEntity.DOC_ID).gt(scrollId));
             searchSourceBuilder.query(boolQueryBuilder);
         } else {
             searchSourceBuilder.query(queryBuilder);
@@ -639,25 +657,20 @@ public boolean pojoScrollEnd(String scrollId) throws IOException {
     }
 
     public <T> T toPojo(GetResponse response, Class<T> clazz) {
-        if (null == response) {
-            return null;
-        } else if (StrUtil.isBlank(response.getSourceAsString())) {
+        if (null == response || StrUtil.isBlank(response.getSourceAsString())) {
             return null;
         } else {
             return JsonUtil.toBean(response.getSourceAsString(), clazz);
         }
     }
 
     public <T> List<T> toPojoList(SearchResponse response, Class<T> clazz) {
-
         if (response == null || response.status() != RestStatus.OK) {
-            return new ArrayList<>();
+            return new ArrayList<>(0);
         }
-
         if (ArrayUtil.isEmpty(response.getHits().getHits())) {
-            return new ArrayList<>();
+            return new ArrayList<>(0);
         }
-
         return Stream.of(response.getHits().getHits())
                      .map(hit -> JsonUtil.toBean(hit.getSourceAsString(), clazz))
                      .collect(Collectors.toList());