使用Java Low Level REST Client操作elasticsearch

Java REST客户端有两种风格:

Java低级别REST客户端(Java Low Level REST Client,以后都简称低级客户端算了,难得码字):Elasticsearch的官方low-level客户端。 它允许通过http与Elasticsearch集群进行通信。 不会对请求进行编码和响应解码。 它与所有Elasticsearch版本兼容。
Java高级REST客户端(Java High Level REST Client,以后都简称高级客户端):Elasticsearch的官方high-level客户端。 基于low-level客户端,它公开了API特定的方法,并负责处理。

低级客户端 的功能包括:

  • 依赖最小
  • 所有可用节点,会负载平衡
  • 在节点故障和响应特定状态码的情况下会进行故障转移
  • 连接失败会进行处罚(失败的节点是否重试,取决于连续失败的次数,失败次数越多,客户端等待的时间越长)
  • 持久连接
  • 跟踪记录请求和响应的日志
  • 可选的自动发现群集节点

低级客户端 快速入门

Java API文档在这里可以找到。低级客户端托管在Maven Central上。所需的最低Java版本是1.7。低级客户端与Elasticsearch的发布周期相同。发布的第一个版本为5.0.0-alpha4。客户端版本和与之通信的Elasticsearch版本没有任何关系,可以替换客户端版本为你想要的任何版本。低级客户端与所有Elasticsearch版本兼容。

Maven Repository  

Maven 配置

下面是使用maven作为依赖管理器配置依赖项。 将以下内容添加到您的pom.xml文件中:

<dependency>
    <groupId>org.elasticsearch.client</groupId>
    <artifactId>elasticsearch-rest-client</artifactId>
    <version>6.2.3</version>
</dependency>

Gradle 配置

 下面是使用gradle作为依赖项管理器来配置依赖项。在您的build.gradle中添加以下内容:

dependencies {
    compile 'org.elasticsearch.client:elasticsearch-rest-client:6.2.3'
}

Dependencies

低级客户端在内部使用Apache Http Async Client发送http请求。 它依赖于以下部件,即the async http client及其自身的传递依赖:

  • org.apache.httpcomponents:httpasyncclient
  • org.apache.httpcomponents:httpcore-nio
  • org.apache.httpcomponents:httpclient
  • org.apache.httpcomponents:httpcore
  • commons-codec:commons-codec
  • commons-logging:commons-logging

Shading

为了避免版本冲突,依赖需要shaded(翻译为被隐藏不知合不合理)和打包到一个单独的jar文件中。(该操作也被称作"uber JAR"或"fat JAR",是一种可执行的Jar包。FatJar和普通的jar不同在于它包含了依赖的jar包。)
对依赖进行隐藏需要取其内容(资源文件和java类文件),然后在放到jar文件之前会对一些包进行重命名。该操作可以使用第三方的插件,比如Gradle 和 Maven来完成。感兴趣的参照这儿。

请注意,隐藏一个JAR也是有缺点的。 例如,隐藏 Commons Logging 层,意味着也需要对依赖的第三方日志进行隐藏。

Maven 配置

下面是使用Maven Shade插件的配置。将以下内容添加到您的pom中。

<build>
    <plugins>
        <plugin>
            <groupId>org.apache.maven.plugins</groupId>
            <artifactId>maven-shade-plugin</artifactId>
            <version>3.1.0</version>
            <executions>
                <execution>
                    <phase>package</phase>
                    <goals><goal>shade</goal></goals>
                    <configuration>
                        <relocations>
                            <relocation>
                                <pattern>org.apache.http</pattern>
                                <shadedPattern>hidden.org.apache.http</shadedPattern>
                            </relocation>
                            <relocation>
                                <pattern>org.apache.logging</pattern>
                                <shadedPattern>hidden.org.apache.logging</shadedPattern>
                            </relocation>
                            <relocation>
                                <pattern>org.apache.commons.codec</pattern>
                                <shadedPattern>hidden.org.apache.commons.codec</shadedPattern>
                            </relocation>
                            <relocation>
                                <pattern>org.apache.commons.logging</pattern>
                                <shadedPattern>hidden.org.apache.commons.logging</shadedPattern>
                            </relocation>
                        </relocations>
                    </configuration>
                </execution>
            </executions>
        </plugin>
    </plugins>
</build>

Gradle 配置

下面是使用Gradle ShadowJar插件的配置。在您的 build.gradle 中添加以下内容。

shadowJar {
    relocate 'org.apache.http', 'hidden.org.apache.http'
    relocate 'org.apache.logging', 'hidden.org.apache.logging'
    relocate 'org.apache.commons.codec', 'hidden.org.apache.commons.codec'
    relocate 'org.apache.commons.logging', 'hidden.org.apache.commons.logging'
}

初始化

RestClient实例可以通过RestClientBuilder类创建,通过RestClient 的 builder(HttpHost ...)静态方法创建。 唯一需要的参数是客户端将与之通信的一个或多个主机,如下所示:

RestClient restClient = RestClient.builder(
        new HttpHost("localhost", 9200, "http"),
        new HttpHost("localhost", 9201, "http")).build();

RestClient类是线程安全的,理想情况下与使用它的应用程序具有相同的生命周期。当不再需要时关闭它是非常重要的,这样它所使用的所有资源以及底层http客户端实例及其线程都可以得到释放。

restClient.close();

RestClientBuilder还允许在构建RestClient实例时可选地设置以下配置参数:

       //配置可选参数
        RestClientBuilder builder = RestClient.builder(
                new HttpHost("localhost", 9200, "http"));
        Header[] defaultHeaders = new Header[]{new BasicHeader("header", "value")};
        //设置每个请求需要发送的默认headers,这样就不用在每个请求中指定它们。
        builder.setDefaultHeaders(defaultHeaders);
        // 设置应该授予的超时时间,以防对相同的请求进行多次尝试。默认值是30秒,与默认socket超时时间相同。
        // 如果自定义socket超时时间,则应相应地调整最大重试超时时间。
        builder.setMaxRetryTimeoutMillis(10000);
        builder.setFailureListener(new RestClient.FailureListener() {
            @Override
            public void onFailure(HttpHost host) {
                //设置一个监听程序,每次节点发生故障时都会收到通知,这样就可以采取相应的措施。
                //Used internally when sniffing on failure is enabled.(这句话没搞懂啥意思)
            }
        });
        builder.setRequestConfigCallback(new RestClientBuilder.RequestConfigCallback() {
            @Override
            public RequestConfig.Builder customizeRequestConfig(RequestConfig.Builder requestConfigBuilder) {
                //设置允许修改默认请求配置的回调
                // (例如,请求超时,身份验证或org.apache.http.client.config.RequestConfig.Builder允许设置的任何内容)
                return requestConfigBuilder.setSocketTimeout(10000);
            }
        });
        builder.setHttpClientConfigCallback(new RestClientBuilder.HttpClientConfigCallback() {
            @Override
            public HttpAsyncClientBuilder customizeHttpClient(HttpAsyncClientBuilder httpClientBuilder) {
                //设置允许修改http客户端配置的回调
                // (例如,通过SSL的加密通信,或者org.apache.http.impl.nio.client.HttpAsyncClientBuilder允许设置的任何内容)
                return httpClientBuilder.setProxy(new HttpHost("proxy", 9000, "http"));
            }
        });

执行请求

 一旦创建了RestClient,就可以调用performRequest或performRequestAsync方法来发送请求。

performRequest方法是同步的,直接返回响应,这意味着客户端将被阻塞并等待响应返回。

performRequestAsync方法返回void,并接受一个ResponseListener作为参数,这意味着它们是异步执行的。当请求完成或失败时,监听器将被通知。

发送同步请求

    //方式1:只提供谓词和终节点,这两个参数是必需要的参数
        Response response = restClient.performRequest("GET", "/");

        //方式2:提供谓词和终节点以及一些查询字符串参数来发送请求
        Map<String, String> params = Collections.singletonMap("pretty", "true");
        response = restClient.performRequest("GET", "/", params);

        //方式3:提供谓词和终节点以及可选查询字符串参数和org.apache.http.HttpEntity对象中包含的请求主体来发送请求
        params = Collections.emptyMap();
        String jsonString = "{" +
                "\"user\":\"kimchy\"," +
                "\"postDate\":\"2013-01-30\"," +
                "\"message\":\"trying out Elasticsearch\"" +
                "}";
        //为HttpEntity指定ContentType非常重要,因为它将用于设置Content-Type请求头,以便Elasticsearch可以正确解析内容。
        HttpEntity entity = new NStringEntity(jsonString, ContentType.APPLICATION_JSON);
        response = restClient.performRequest("PUT", "/posts/doc/1", params, entity);

        //方式4:提供谓词,终节点,可选查询字符串参数,可选请求主体
        // 以及用于为每个请求尝试创建org.apache.http.nio.protocol.HttpAsyncResponseConsumer回调实例的可选工厂来发送请求。
        // 控制响应正文如何从客户端的非阻塞HTTP连接进行流式传输。
        // 如果未提供,则使用默认实现,将整个响应主体缓存在堆内存中,最大为100 MB。
        params = Collections.emptyMap();
        HttpAsyncResponseConsumerFactory.HeapBufferedResponseConsumerFactory consumerFactory =
                new HttpAsyncResponseConsumerFactory.HeapBufferedResponseConsumerFactory(30 * 1024 * 1024);
        response = restClient.performRequest("GET", "/posts/_search", params, null, consumerFactory);

 发送异步请求

    //方式1: 提供谓词,终节点和响应监听器来发送异步请求,一旦请求完成,就会通知响应监听器,这三个参数是必需要的参数
        ResponseListener responseListener = new ResponseListener() {
            @Override
            public void onSuccess(Response response) {
                // 定义请求成功执行时需要做的事情
            }
            @Override
            public void onFailure(Exception exception) {
                // 定义请求失败时需要做的事情,即每当发生连接错误或返回错误状态码时做的操作。
            }
        };
        restClient.performRequestAsync("GET", "/", responseListener);

        //方式2: 提供谓词,终节点,一些查询字符串参数和响应监听器来发送异步请求
        Map<String, String>  params = Collections.singletonMap("pretty", "true");
        restClient.performRequestAsync("GET", "/", params, responseListener);

        //方式3:提供谓词,终节点,可选查询字符串参数,
        // org.apache.http.HttpEntity对象中包含的请求主体以及在请求完成后通知响应侦听器 来发送异步请求
        String jsonString = "{" +
                "\"user\":\"kimchy\"," +
                "\"postDate\":\"2013-01-30\"," +
                "\"message\":\"trying out Elasticsearch\"" +
                "}";
        NStringEntity entity = new NStringEntity(jsonString, ContentType.APPLICATION_JSON);
        restClient.performRequestAsync("PUT", "/posts/doc/1", params, entity, responseListener);

        //方式4:提供谓词,终节点,可选查询字符串参数,可选请求主体
        // 以及用于为每个请求尝试创建org.apache.http.nio.protocol.HttpAsyncResponseConsumer回调实例的可选工厂 来发送异步请求。
        // 控制响应正文如何从客户端的非阻塞HTTP连接进行流式传输。
        // 如果未提供,则使用默认实现,将整个响应主体缓存在堆内存中,最大为100 MB。
        HttpAsyncResponseConsumerFactory.HeapBufferedResponseConsumerFactory  consumerFactory =
                new HttpAsyncResponseConsumerFactory.HeapBufferedResponseConsumerFactory(30 * 1024 * 1024);
        restClient.performRequestAsync("GET", "/posts/_search", params, null, consumerFactory, responseListener);

接下来是一个发送异步请求的基本示例:

final CountDownLatch latch = new CountDownLatch(documents.length);
        for (int i = 0; i < documents.length; i++) {
            restClient.performRequestAsync(
                    "PUT",
                    "/posts/doc/" + i,
                    Collections.<String, String>emptyMap(),
                    //此处假设文档已存在 HttpEntity数组里
                    documents[i],
                    new ResponseListener() {
                        @Override
                        public void onSuccess(Response response) {
                            //处理返回的响应内容
                            latch.countDown();
                        }

                        @Override
                        public void onFailure(Exception exception) {
                            // 由于通信错误或带有指示错误的状态码的响应,用于处理返回的异常
                            latch.countDown();
                        }
                    }
            );
        }
        latch.await();

上面列出的每个方法都支持通过Header 可变参数发送请求头,源码如下:

一个header时, 如下例所示:

Response response = restClient.performRequest("GET", "/", new BasicHeader("header", "value"));

 多个header时,如下所示:

Header[] headers = {
        new BasicHeader("header1", "value1"),
        new BasicHeader("header2", "value2")
};
restClient.performRequestAsync("GET", "/", responseListener, headers);

获取响应

 Response对象(由同步performRequest方法返回或由ResponseListener的onSuccess(Response)中的参数接收),包装从http客户端返回的响应对象并公开一些其他的信息。

 RestClient restClient = RestClient.builder(
                new HttpHost("localhost", 9200, "http"),
                new HttpHost("localhost", 9201, "http")).build();
        Response response = restClient.performRequest("GET", "/");
        RequestLine requestLine = response.getRequestLine();//关于已执行请求的信息
        HttpHost host = response.getHost();//返回响应的主机
        int statusCode = response.getStatusLine().getStatusCode();//响应状态行,可以从中获取状态码
        Header[] headers = response.getHeaders();// 获取响应头
        String header=response.getHeader("content-type");// 获取指定名称的响应头
        String responseBody = EntityUtils.toString(response.getEntity());//响应体包含在org.apache.http.HttpEntity对象中

执行请求时,会在以下情况中引发异常(异步时在ResponseListener#onFailure(Exception)中作为参数接收到该异常):

  1. IOException,通信问题(例如SocketTimeoutException)
  2. ResponseException,返回了一个响应,但是它的状态码表明是错误的(不是2xx)。 ResponseException是一个有效的http响应,因此它暴露了其相应的Response对象,可以访问返回的响应。

对于返回404状态代码的HEAD请求,不会引发ResponseException,因为它是预期的HEAD响应,它只是表示找不到资源。 

Response response = restClient.performRequest("HEAD", "/s");//不会抛异常

除非ignore参数包含404,否则所有其他HTTP方法(例如GET)都会为404响应抛出ResponseException。

ignore是一个特殊的客户端参数,它不会发送到Elasticsearch,且包含以逗号分隔的错误状态码列表。 它允许控制是否应将某些错误状态码视为预期响应而不是异常。

这对于get api来说很有用,因为它可以在缺少文档时返回404,在这种情况下,响应主体将不会包含错误,而是通常的get api响应,只是没有文档,因为它没有找到。

注意,低级客户端不会序列化或反序列化json。用户可以自由使用他们喜欢的库。

底层的Apache Async Http Client附带不同的org.apache.http.HttpEntity实现,允许以不同格式(流,字节数组,字符串等)提供请求体。

至于读取响应体,HttpEntity的getContent方法很方便,它会返回来自先前缓冲的响应体的InputStream。作为一种替代方法,可以提供一个自定义org.apache.http.nio.protocol.HttpAsyncResponseConsumer来控制如何读取和缓冲字节。

 日志

The Java REST client使用了和Apache Async Http Client相同的日志库:Apache Commons Logging,
它支持许多流行的日志实现。 用于启用日志记录功能的java包分别是客户端本身的org.elasticsearch.client,以及嗅探器的org.elasticsearch.client.sniffer。

也可以启用以curl格式来记录每个请求和相应的响应。 这使得调试时非常方便。

例如在需要手动执行请求以检查它是否仍然产生相同的响应时。 为跟踪器包启用日志记录以打印出此类日志。 请注意,该类型的日志记录非常昂贵,不应始终在生产环境中启用,而只是在需要时暂时使用。

官方文档:https://www.elastic.co/guide/en/elasticsearch/client/java-rest/current/java-rest-overview.html

posted @ 2018-03-31 18:21  ~冰  阅读(21718)  评论(4编辑  收藏  举报