Java代码：Elasticsearch存储及查询地理信息数据（经纬度）

ゝ一世哀愁。 2023-10-15 09:39 79阅读 0赞

## 存储 ##

以下是一个使用Java代码将地理信息数据存储到Elasticsearch的示例：

import org.elasticsearch.action.bulk.BulkRequest;
    import org.elasticsearch.action.bulk.BulkResponse;
    import org.elasticsearch.action.index.IndexRequest;
    import org.elasticsearch.client.RequestOptions;
    import org.elasticsearch.client.RestClient;
    import org.elasticsearch.client.RestHighLevelClient;
    import org.elasticsearch.common.geo.GeoPoint;
    import org.elasticsearch.common.xcontent.XContentBuilder;
    import org.elasticsearch.common.xcontent.XContentFactory;
    import org.elasticsearch.common.xcontent.XContentType;
    import org.elasticsearch.index.Index;
    
    import java.io.IOException;
    
    public class ElasticsearchGeoDataExample {
        
        public static void main(String[] args) {
        
            // 创建Elasticsearch客户端
            RestHighLevelClient client = new RestHighLevelClient(
                    RestClient.builder("localhost:9200"));
    
            // 准备地理信息数据
            GeoPoint location1 = new GeoPoint(40.7128, -74.0060);
            GeoPoint location2 = new GeoPoint(34.0522, -118.2437);
    
            // 创建索引请求
            BulkRequest bulkRequest = new BulkRequest();
    
            // 添加文档到索引请求
            try {
        
                bulkRequest.add(createIndexRequest("places", "1", "New York", location1));
                bulkRequest.add(createIndexRequest("places", "2", "Los Angeles", location2));
            } catch (IOException e) {
        
                e.printStackTrace();
            }
    
            // 执行索引请求
            try {
        
                BulkResponse bulkResponse = client.bulk(bulkRequest, RequestOptions.DEFAULT);
                if (bulkResponse.hasFailures()) {
        
                    System.out.println("存储地理信息数据失败：" + bulkResponse.buildFailureMessage());
                } else {
        
                    System.out.println("地理信息数据已成功存储到Elasticsearch。");
                }
            } catch (IOException e) {
        
                e.printStackTrace();
            }
    
            // 关闭Elasticsearch客户端
            try {
        
                client.close();
            } catch (IOException e) {
        
                e.printStackTrace();
            }
        }
    
        private static IndexRequest createIndexRequest(String index, String id, String name, GeoPoint location) throws IOException {
        
            XContentBuilder builder = XContentFactory.jsonBuilder();
            builder.startObject();
            {
        
                builder.field("name", name);
                builder.field("location", location);
            }
            builder.endObject();
    
            return new IndexRequest(index)
                    .id(id)
                    .source(builder, XContentType.JSON);
        }
    }

上述示例代码使用了Elasticsearch的Java高级客户端（RestHighLevelClient）来与Elasticsearch进行交互。首先，创建了一个RestHighLevelClient实例，连接到本地的Elasticsearch节点（localhost:9200）。

然后，定义了两个地理信息数据点（location1和location2），分别表示纽约和洛杉矶的经纬度。

接下来，创建了一个批量请求（BulkRequest），并使用createIndexRequest方法创建了两个索引请求（IndexRequest），分别将地理信息数据存储到名为"places"的索引中。

最后，通过调用client.bulk方法执行批量请求，将地理信息数据发送到Elasticsearch进行存储。

请注意，上述代码中的索引名称、文档ID和字段名称可以根据实际需求进行修改。还可以根据需要添加其他字段和设置。

确保在运行示例代码之前，已正确安装并启动了Elasticsearch，并确保已添加Elasticsearch Java客户端的依赖项。

## 查询 ##

要查询刚刚存储到Elasticsearch中的地理信息数据，你可以使用Elasticsearch的Java高级客户端执行相应的搜索操作。以下是一个示例代码，演示如何查询存储的地理信息数据：

import org.elasticsearch.action.search.SearchRequest;
    import org.elasticsearch.action.search.SearchResponse;
    import org.elasticsearch.client.RequestOptions;
    import org.elasticsearch.client.RestClient;
    import org.elasticsearch.client.RestHighLevelClient;
    import org.elasticsearch.common.unit.DistanceUnit;
    import org.elasticsearch.index.query.QueryBuilders;
    import org.elasticsearch.search.SearchHit;
    import org.elasticsearch.search.builder.SearchSourceBuilder;
    import org.elasticsearch.search.sort.GeoDistanceSortBuilder;
    import org.elasticsearch.search.sort.SortOrder;
    
    import java.io.IOException;
    
    public class ElasticsearchGeoDataQueryExample {
        
        public static void main(String[] args) {
        
            // 创建Elasticsearch客户端
            RestHighLevelClient client = new RestHighLevelClient(
                    RestClient.builder("localhost:9200"));
    
            // 创建搜索请求
            SearchRequest searchRequest = new SearchRequest("places");
    
            // 创建搜索条件
            SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
            sourceBuilder.query(QueryBuilders.matchAllQuery());
    
            // 添加排序规则
            GeoDistanceSortBuilder sortBuilder = new GeoDistanceSortBuilder("location", 40.7128, -74.0060);
            sortBuilder.unit(DistanceUnit.KILOMETERS);
            sortBuilder.order(SortOrder.ASC);
            sourceBuilder.sort(sortBuilder);
    
            // 设置搜索源
            searchRequest.source(sourceBuilder);
    
            // 执行搜索请求
            try {
        
                SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
    
                // 处理搜索结果
                for (SearchHit hit : searchResponse.getHits().getHits()) {
        
                    String id = hit.getId();
                    String name = hit.getSourceAsMap().get("name").toString();
                    System.out.println("ID: " + id + ", Name: " + name);
                }
            } catch (IOException e) {
        
                e.printStackTrace();
            }
    
            // 关闭Elasticsearch客户端
            try {
        
                client.close();
            } catch (IOException e) {
        
                e.printStackTrace();
            }
        }
    }

上述示例代码创建了一个SearchRequest对象，并设置索引名称为"places"，表示要从该索引中进行查询。

然后，创建SearchSourceBuilder对象，设置查询条件为match\_all\_query()，表示查询所有文档。

接下来，添加了一个排序规则，使用GeoDistanceSortBuilder按照给定的经纬度（纽约的经纬度）对结果进行距离排序。

将SearchSourceBuilder设置为SearchRequest的搜索源。

最后，通过调用client.search方法执行搜索请求，并遍历SearchResponse中的搜索结果（SearchHit），提取所需的字段值（在这里是文档的ID和名称）。

请注意，上述代码中的索引名称、字段名称和排序规则可以根据实际情况进行修改。还可以根据需要添加其他过滤条件和查询操作。

确保在运行示例代码之前，已正确安装并启动了Elasticsearch，并确保已添加Elasticsearch Java客户端的依赖项。