|
78 | 78 | import org.elasticsearch.client.ml.GetOverallBucketsResponse;
|
79 | 79 | import org.elasticsearch.client.ml.GetRecordsRequest;
|
80 | 80 | import org.elasticsearch.client.ml.GetRecordsResponse;
|
| 81 | +import org.elasticsearch.client.ml.MlInfoRequest; |
| 82 | +import org.elasticsearch.client.ml.MlInfoResponse; |
81 | 83 | import org.elasticsearch.client.ml.OpenJobRequest;
|
82 | 84 | import org.elasticsearch.client.ml.OpenJobResponse;
|
83 | 85 | import org.elasticsearch.client.ml.PostCalendarEventRequest;
|
@@ -3003,6 +3005,55 @@ public void onFailure(Exception e) {
|
3003 | 3005 | }
|
3004 | 3006 | }
|
3005 | 3007 |
|
| 3008 | + public void testGetMlInfo() throws Exception { |
| 3009 | + RestHighLevelClient client = highLevelClient(); |
| 3010 | + |
| 3011 | + { |
| 3012 | + // tag::get-ml-info-request |
| 3013 | + MlInfoRequest request = new MlInfoRequest(); // <1> |
| 3014 | + // end::get-ml-info-request |
| 3015 | + |
| 3016 | + // tag::get-ml-info-execute |
| 3017 | + MlInfoResponse response = client.machineLearning() |
| 3018 | + .getMlInfo(request, RequestOptions.DEFAULT); |
| 3019 | + // end::get-ml-info-execute |
| 3020 | + |
| 3021 | + // tag::get-ml-info-response |
| 3022 | + final Map<String, Object> info = response.getInfo();// <1> |
| 3023 | + // end::get-ml-info-response |
| 3024 | + assertTrue(info.containsKey("defaults")); |
| 3025 | + assertTrue(info.containsKey("limits")); |
| 3026 | + } |
| 3027 | + { |
| 3028 | + MlInfoRequest request = new MlInfoRequest(); |
| 3029 | + |
| 3030 | + // tag::get-ml-info-execute-listener |
| 3031 | + ActionListener<MlInfoResponse> listener = new ActionListener<MlInfoResponse>() { |
| 3032 | + @Override |
| 3033 | + public void onResponse(MlInfoResponse response) { |
| 3034 | + // <1> |
| 3035 | + } |
| 3036 | + |
| 3037 | + @Override |
| 3038 | + public void onFailure(Exception e) { |
| 3039 | + // <2> |
| 3040 | + } |
| 3041 | + }; |
| 3042 | + // end::get-ml-info-execute-listener |
| 3043 | + |
| 3044 | + // Replace the empty listener by a blocking listener in test |
| 3045 | + final CountDownLatch latch = new CountDownLatch(1); |
| 3046 | + listener = new LatchedActionListener<>(listener, latch); |
| 3047 | + |
| 3048 | + // tag::get-ml-info-execute-async |
| 3049 | + client.machineLearning() |
| 3050 | + .getMlInfoAsync(request, RequestOptions.DEFAULT, listener); // <1> |
| 3051 | + // end::get-ml-info-execute-async |
| 3052 | + |
| 3053 | + assertTrue(latch.await(30L, TimeUnit.SECONDS)); |
| 3054 | + } |
| 3055 | + } |
| 3056 | + |
3006 | 3057 | private String createFilter(RestHighLevelClient client) throws IOException {
|
3007 | 3058 | MlFilter.Builder filterBuilder = MlFilter.builder("my_safe_domains")
|
3008 | 3059 | .setDescription("A list of safe domains")
|
|
0 commit comments