|
32 | 32 | import org.elasticsearch.client.ml.DeleteCalendarEventRequest;
|
33 | 33 | import org.elasticsearch.client.ml.DeleteCalendarJobRequest;
|
34 | 34 | import org.elasticsearch.client.ml.DeleteCalendarRequest;
|
| 35 | +import org.elasticsearch.client.ml.DeleteDataFrameAnalyticsRequest; |
35 | 36 | import org.elasticsearch.client.ml.DeleteDatafeedRequest;
|
36 | 37 | import org.elasticsearch.client.ml.DeleteExpiredDataRequest;
|
37 | 38 | import org.elasticsearch.client.ml.DeleteFilterRequest;
|
38 | 39 | import org.elasticsearch.client.ml.DeleteForecastRequest;
|
39 | 40 | import org.elasticsearch.client.ml.DeleteJobRequest;
|
40 | 41 | import org.elasticsearch.client.ml.DeleteModelSnapshotRequest;
|
| 42 | +import org.elasticsearch.client.ml.EvaluateDataFrameRequest; |
41 | 43 | import org.elasticsearch.client.ml.FindFileStructureRequest;
|
42 | 44 | import org.elasticsearch.client.ml.FlushJobRequest;
|
43 | 45 | import org.elasticsearch.client.ml.ForecastJobRequest;
|
44 | 46 | import org.elasticsearch.client.ml.GetBucketsRequest;
|
45 | 47 | import org.elasticsearch.client.ml.GetCalendarEventsRequest;
|
46 | 48 | import org.elasticsearch.client.ml.GetCalendarsRequest;
|
47 | 49 | import org.elasticsearch.client.ml.GetCategoriesRequest;
|
| 50 | +import org.elasticsearch.client.ml.GetDataFrameAnalyticsRequest; |
| 51 | +import org.elasticsearch.client.ml.GetDataFrameAnalyticsStatsRequest; |
48 | 52 | import org.elasticsearch.client.ml.GetDatafeedRequest;
|
49 | 53 | import org.elasticsearch.client.ml.GetDatafeedStatsRequest;
|
50 | 54 | import org.elasticsearch.client.ml.GetFiltersRequest;
|
|
61 | 65 | import org.elasticsearch.client.ml.PreviewDatafeedRequest;
|
62 | 66 | import org.elasticsearch.client.ml.PutCalendarJobRequest;
|
63 | 67 | import org.elasticsearch.client.ml.PutCalendarRequest;
|
| 68 | +import org.elasticsearch.client.ml.PutDataFrameAnalyticsRequest; |
64 | 69 | import org.elasticsearch.client.ml.PutDatafeedRequest;
|
65 | 70 | import org.elasticsearch.client.ml.PutFilterRequest;
|
66 | 71 | import org.elasticsearch.client.ml.PutJobRequest;
|
67 | 72 | import org.elasticsearch.client.ml.RevertModelSnapshotRequest;
|
68 | 73 | import org.elasticsearch.client.ml.SetUpgradeModeRequest;
|
| 74 | +import org.elasticsearch.client.ml.StartDataFrameAnalyticsRequest; |
69 | 75 | import org.elasticsearch.client.ml.StartDatafeedRequest;
|
| 76 | +import org.elasticsearch.client.ml.StopDataFrameAnalyticsRequest; |
70 | 77 | import org.elasticsearch.client.ml.StopDatafeedRequest;
|
71 | 78 | import org.elasticsearch.client.ml.UpdateDatafeedRequest;
|
72 | 79 | import org.elasticsearch.client.ml.UpdateFilterRequest;
|
@@ -581,6 +588,115 @@ static Request deleteCalendarEvent(DeleteCalendarEventRequest deleteCalendarEven
|
581 | 588 | return new Request(HttpDelete.METHOD_NAME, endpoint);
|
582 | 589 | }
|
583 | 590 |
|
| 591 | + static Request putDataFrameAnalytics(PutDataFrameAnalyticsRequest putRequest) throws IOException { |
| 592 | + String endpoint = new EndpointBuilder() |
| 593 | + .addPathPartAsIs("_ml", "data_frame", "analytics") |
| 594 | + .addPathPart(putRequest.getConfig().getId()) |
| 595 | + .build(); |
| 596 | + Request request = new Request(HttpPut.METHOD_NAME, endpoint); |
| 597 | + request.setEntity(createEntity(putRequest, REQUEST_BODY_CONTENT_TYPE)); |
| 598 | + return request; |
| 599 | + } |
| 600 | + |
| 601 | + static Request getDataFrameAnalytics(GetDataFrameAnalyticsRequest getRequest) { |
| 602 | + String endpoint = new EndpointBuilder() |
| 603 | + .addPathPartAsIs("_ml", "data_frame", "analytics") |
| 604 | + .addPathPart(Strings.collectionToCommaDelimitedString(getRequest.getIds())) |
| 605 | + .build(); |
| 606 | + Request request = new Request(HttpGet.METHOD_NAME, endpoint); |
| 607 | + RequestConverters.Params params = new RequestConverters.Params(); |
| 608 | + if (getRequest.getPageParams() != null) { |
| 609 | + PageParams pageParams = getRequest.getPageParams(); |
| 610 | + if (pageParams.getFrom() != null) { |
| 611 | + params.putParam(PageParams.FROM.getPreferredName(), pageParams.getFrom().toString()); |
| 612 | + } |
| 613 | + if (pageParams.getSize() != null) { |
| 614 | + params.putParam(PageParams.SIZE.getPreferredName(), pageParams.getSize().toString()); |
| 615 | + } |
| 616 | + } |
| 617 | + if (getRequest.getAllowNoMatch() != null) { |
| 618 | + params.putParam(GetDataFrameAnalyticsRequest.ALLOW_NO_MATCH.getPreferredName(), Boolean.toString(getRequest.getAllowNoMatch())); |
| 619 | + } |
| 620 | + request.addParameters(params.asMap()); |
| 621 | + return request; |
| 622 | + } |
| 623 | + |
| 624 | + static Request getDataFrameAnalyticsStats(GetDataFrameAnalyticsStatsRequest getStatsRequest) { |
| 625 | + String endpoint = new EndpointBuilder() |
| 626 | + .addPathPartAsIs("_ml", "data_frame", "analytics") |
| 627 | + .addPathPart(Strings.collectionToCommaDelimitedString(getStatsRequest.getIds())) |
| 628 | + .addPathPartAsIs("_stats") |
| 629 | + .build(); |
| 630 | + Request request = new Request(HttpGet.METHOD_NAME, endpoint); |
| 631 | + RequestConverters.Params params = new RequestConverters.Params(); |
| 632 | + if (getStatsRequest.getPageParams() != null) { |
| 633 | + PageParams pageParams = getStatsRequest.getPageParams(); |
| 634 | + if (pageParams.getFrom() != null) { |
| 635 | + params.putParam(PageParams.FROM.getPreferredName(), pageParams.getFrom().toString()); |
| 636 | + } |
| 637 | + if (pageParams.getSize() != null) { |
| 638 | + params.putParam(PageParams.SIZE.getPreferredName(), pageParams.getSize().toString()); |
| 639 | + } |
| 640 | + } |
| 641 | + if (getStatsRequest.getAllowNoMatch() != null) { |
| 642 | + params.putParam(GetDataFrameAnalyticsStatsRequest.ALLOW_NO_MATCH.getPreferredName(), |
| 643 | + Boolean.toString(getStatsRequest.getAllowNoMatch())); |
| 644 | + } |
| 645 | + request.addParameters(params.asMap()); |
| 646 | + return request; |
| 647 | + } |
| 648 | + |
| 649 | + static Request startDataFrameAnalytics(StartDataFrameAnalyticsRequest startRequest) { |
| 650 | + String endpoint = new EndpointBuilder() |
| 651 | + .addPathPartAsIs("_ml", "data_frame", "analytics") |
| 652 | + .addPathPart(startRequest.getId()) |
| 653 | + .addPathPartAsIs("_start") |
| 654 | + .build(); |
| 655 | + Request request = new Request(HttpPost.METHOD_NAME, endpoint); |
| 656 | + RequestConverters.Params params = new RequestConverters.Params(); |
| 657 | + if (startRequest.getTimeout() != null) { |
| 658 | + params.withTimeout(startRequest.getTimeout()); |
| 659 | + } |
| 660 | + request.addParameters(params.asMap()); |
| 661 | + return request; |
| 662 | + } |
| 663 | + |
| 664 | + static Request stopDataFrameAnalytics(StopDataFrameAnalyticsRequest stopRequest) { |
| 665 | + String endpoint = new EndpointBuilder() |
| 666 | + .addPathPartAsIs("_ml", "data_frame", "analytics") |
| 667 | + .addPathPart(stopRequest.getId()) |
| 668 | + .addPathPartAsIs("_stop") |
| 669 | + .build(); |
| 670 | + Request request = new Request(HttpPost.METHOD_NAME, endpoint); |
| 671 | + RequestConverters.Params params = new RequestConverters.Params(); |
| 672 | + if (stopRequest.getTimeout() != null) { |
| 673 | + params.withTimeout(stopRequest.getTimeout()); |
| 674 | + } |
| 675 | + if (stopRequest.getAllowNoMatch() != null) { |
| 676 | + params.putParam( |
| 677 | + StopDataFrameAnalyticsRequest.ALLOW_NO_MATCH.getPreferredName(), Boolean.toString(stopRequest.getAllowNoMatch())); |
| 678 | + } |
| 679 | + request.addParameters(params.asMap()); |
| 680 | + return request; |
| 681 | + } |
| 682 | + |
| 683 | + static Request deleteDataFrameAnalytics(DeleteDataFrameAnalyticsRequest deleteRequest) { |
| 684 | + String endpoint = new EndpointBuilder() |
| 685 | + .addPathPartAsIs("_ml", "data_frame", "analytics") |
| 686 | + .addPathPart(deleteRequest.getId()) |
| 687 | + .build(); |
| 688 | + return new Request(HttpDelete.METHOD_NAME, endpoint); |
| 689 | + } |
| 690 | + |
| 691 | + static Request evaluateDataFrame(EvaluateDataFrameRequest evaluateRequest) throws IOException { |
| 692 | + String endpoint = new EndpointBuilder() |
| 693 | + .addPathPartAsIs("_ml", "data_frame", "_evaluate") |
| 694 | + .build(); |
| 695 | + Request request = new Request(HttpPost.METHOD_NAME, endpoint); |
| 696 | + request.setEntity(createEntity(evaluateRequest, REQUEST_BODY_CONTENT_TYPE)); |
| 697 | + return request; |
| 698 | + } |
| 699 | + |
584 | 700 | static Request putFilter(PutFilterRequest putFilterRequest) throws IOException {
|
585 | 701 | String endpoint = new EndpointBuilder()
|
586 | 702 | .addPathPartAsIs("_ml")
|
|
0 commit comments