Elasticsearch-经纬度查询(8.x)

目录

一、开发环境

二、pom文件

三、ES配置文件

四、ES相关字段

五、ES半径查询

---

ES的字段类型:geo_point，可以实现以一个点为中心的半径查询(geo_distance query)

ES地理位置查询:

• 半径查询(geo_distance query)
• 查询指定矩形内的数据(geo_bounding_box query)
• 查询指定多边形内的数据(geo_polygon query)

官方文档：Geo queries | Elasticsearch Guide [8.14] | Elastic

本案例实现以某个点的经纬，查询半径查询，并计算出其他点和中心点的距离

一、开发环境

工具	版本
JDK	21
SpringBoot	3.2.4
ElasticSearch	8.13.2

二、pom文件

        <!-- Elasticsearch Java API Client -->
        <dependency>
            <groupId>co.elastic.clients</groupId>
            <artifactId>elasticsearch-java</artifactId>
            <version>8.13.2</version>
        </dependency>

三、ES配置文件

properties:

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;

import java.util.List;

/**
 * ES 配置属性
 *
 * @author meng
 * @date 2024-05-07
 */
@Data
@ConfigurationProperties(prefix = "es")
public class ElasticsearchProperties {
    /**
     * es 请求方式
     */
    private String scheme;
    /**
     * es 集群host ip 地址
     */
    private List<String> hosts;
    /**
     * es 账号
     */
    private String username;
    /**
     * es 密码
     */
    private String password;
    /**
     * es 连接超时时间
     */
    private Integer connectTimeOut;
    /**
     * es socket 连接超时时间
     */
    private Integer socketTimeOut;
    /**
     * es 请求超时时间
     */
    private Integer connectionRequestTimeOut;
    /**
     * es 最大连接数
     */
    private Integer maxConnectNum;
    /**
     * es 每个路由的最大连接数
     */
    private Integer maxConnectNumPerRoute;
}

config:

import co.elastic.clients.elasticsearch.ElasticsearchClient;
import co.elastic.clients.json.jackson.JacksonJsonpMapper;
import co.elastic.clients.transport.ElasticsearchTransport;
import co.elastic.clients.transport.rest_client.RestClientTransport;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpHost;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.conn.ssl.NoopHostnameVerifier;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.ssl.SSLContextBuilder;
import org.apache.http.ssl.TrustStrategy;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestClientBuilder;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.net.ssl.SSLContext;
import java.security.KeyManagementException;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.time.Duration;

/**
 * ES 配置类，当 es.enable=true 时自动配置
 *
 * @author meng
 * @date 2024-05-07
 */
@Slf4j
@Configuration
@EnableConfigurationProperties(value = ElasticsearchProperties.class)
@ConditionalOnProperty(prefix = "es", name = "enable")
public class ElasticsearchConfig {

    @Bean(name = "ElasticsearchClient")
    public ElasticsearchClient initClient(ElasticsearchProperties elasticsearchProperties) {
        HttpHost[] esHosts = elasticsearchProperties.getHosts().stream().filter(StringUtils::isNotBlank).map(it -> {
            String[] split = it.split(":");
            return new HttpHost(split[0], Integer.parseInt(split[1]), elasticsearchProperties.getScheme());
        }).toArray(HttpHost[]::new);
        final CredentialsProvider credentialsProvider = new BasicCredentialsProvider();
        credentialsProvider.setCredentials(AuthScope.ANY,
                new UsernamePasswordCredentials(elasticsearchProperties.getUsername(),
                        elasticsearchProperties.getPassword()));

        RestClientBuilder restClientBuilder = RestClient.builder(esHosts);
        // 连接延时配置
        restClientBuilder.setRequestConfigCallback(requestConfigBuilder -> {
            requestConfigBuilder.setConnectTimeout(elasticsearchProperties.getConnectTimeOut());
            requestConfigBuilder.setSocketTimeout(elasticsearchProperties.getSocketTimeOut());
            requestConfigBuilder.setConnectionRequestTimeout(elasticsearchProperties.getConnectionRequestTimeOut());
            return requestConfigBuilder;
        });
        // HttpClient 连接数配置
        restClientBuilder.setHttpClientConfigCallback(httpClientBuilder -> {
            httpClientBuilder.setMaxConnTotal(elasticsearchProperties.getMaxConnectNum());
            httpClientBuilder.setMaxConnPerRoute(elasticsearchProperties.getMaxConnectNumPerRoute());
            return httpClientBuilder;
        });
        // 设置keepalive时间
        restClientBuilder.setHttpClientConfigCallback(requestConfig ->
                requestConfig.setKeepAliveStrategy((response, context) -> Duration.ofMinutes(3).toMillis())
        );
        // 忽略HTTPS方式链接，SSL证书、主机名验证器
        restClientBuilder.setHttpClientConfigCallback(httpAsyncClientBuilder -> {
            httpAsyncClientBuilder.disableAuthCaching();
            httpAsyncClientBuilder.setDefaultCredentialsProvider(credentialsProvider);
            try {
                // 创建一个信任所有证书的 TrustStrategy 策略
                TrustStrategy acceptTrustStrategy = (chain, authType) -> true;
                // 使用 SSLContextBuilder 创建 SSLContext
                SSLContext sslContext = SSLContextBuilder.create().loadTrustMaterial(null, acceptTrustStrategy).build();
                httpAsyncClientBuilder.setSSLContext(sslContext);
            } catch (NoSuchAlgorithmException | KeyStoreException | KeyManagementException e) {
                e.printStackTrace();
            }
            // 忽略主机名验证
            httpAsyncClientBuilder.setSSLHostnameVerifier(NoopHostnameVerifier.INSTANCE);
            return httpAsyncClientBuilder;
        });
        RestClient restClient = restClientBuilder.build();
        // 使用Jackson映射器创建传输层
        ElasticsearchTransport transport = new RestClientTransport(
                restClient, new JacksonJsonpMapper()
        );
        ElasticsearchClient client = new ElasticsearchClient(transport);
        return client;
    }

}

四、ES相关字段

五、ES半径查询

具体查询对象，可自行定义，本方法只提供思路，莫直接粘贴使用

	@Resource
    private ElasticsearchClient esClient;

	/**
	 * 功能描述: 根据经纬度判断十米内是否有poi数据
	 *
	 * @Param: [request]
	 * @Author: meng
	 * @Date: 2024/6/6 19:57
	 */
	private PoiDTO getPoiNear(PoiRequest request) {
		if (Objects.isNull(request.getLongitude()) && Objects.isNull(request.getLatitude())) {
			return null;
		}
		// 中心点查询
		GeoLocation geoLocation = new GeoLocation.Builder().latlon(
				l -> l.lon(request.getLongitude().doubleValue()).lat(request.getLatitude().doubleValue())).build();
		GeoDistanceQuery geoDistanceQuery = GeoDistanceQuery.of(
				geoDis -> geoDis.field(PoiIndexConstant.LOCATION).location(geoLocation)
						.distance(PoiIndexConstant.DISTANCE).distanceType(GeoDistanceType.Arc));
		// 排序
		List<SortOptions> sorts = new ArrayList<>();
		// 距离排序 米
		GeoDistanceSort geoDistanceSort = GeoDistanceSort.of(
				geoDis -> geoDis.field(PoiIndexConstant.LOCATION).location(geoLocation).order(SortOrder.Asc)
						.unit(DistanceUnit.Meters));
		SortOptions sortOptions = SortOptions.of(sort -> sort.geoDistance(geoDistanceSort));
		sorts.add(sortOptions);
		SearchRequest.Builder searchRequestBuilder = new SearchRequest.Builder();
		searchRequestBuilder.index(PoiIndexConstant.INDEX_READ)
				.query(query -> query.bool(builder -> builder.filter(f -> f.geoDistance(geoDistanceQuery)))).sort(sorts)
				.size(CommonConstant.DEFAULT_NUM);
		// ES查询
		SearchRequest searchRequest = searchRequestBuilder.build();
		log.info("getPoiNear query:{}", searchRequest.toString());
		try {
            SearchResponse<PoiIndex> searchResponse = esClient.search(searchRequest, PoiIndex.class);
            // 未查到数据
			if (Objects.nonNull(searchResponse.hits().total()) && searchResponse.hits().total().value() == 0) {
				return null;
			}
			for (Hit<PoiIndex> hit : searchResponse.hits().hits()) {
				if (ObjectUtil.isNotNull(hit.source())) {
					PoiIndex poiIndex = BeanUtil.copyProperties(hit.source(), PoiIndex.class);
					return BeanUtil.copyProperties(poiIndex, PoiDTO.class);
				}
			}
        } catch (Exception e) {
            log.error("es searchRequest Exception:", e);
            throw new ServiceException("searchRequest Exception");
        }
		return null;
	}