Skip to content

Commit b84ec6c

Browse files
authored
Generate ml operations (#1389)
Signed-off-by: Thomas Farr <[email protected]>
1 parent 335a75c commit b84ec6c

File tree

80 files changed

+22426
-157
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

80 files changed

+22426
-157
lines changed

java-client/src/generated/java/org/opensearch/client/opensearch/cat/IndicesRequest.java

+6-3
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,8 @@ public final Time clusterManagerTimeout() {
137137
}
138138

139139
/**
140-
* The type of index that wildcard patterns can match.
140+
* The type of index that wildcard patterns can match. Supported values are <code>all</code>, <code>open</code>, <code>closed</code>,
141+
* <code>hidden</code>, and <code>none</code>.
141142
* <p>
142143
* API name: {@code expand_wildcards}
143144
* </p>
@@ -327,7 +328,8 @@ public final Builder clusterManagerTimeout(Function<Time.Builder, ObjectBuilder<
327328
}
328329

329330
/**
330-
* The type of index that wildcard patterns can match.
331+
* The type of index that wildcard patterns can match. Supported values are <code>all</code>, <code>open</code>,
332+
* <code>closed</code>, <code>hidden</code>, and <code>none</code>.
331333
* <p>
332334
* API name: {@code expand_wildcards}
333335
* </p>
@@ -343,7 +345,8 @@ public final Builder expandWildcards(List<ExpandWildcard> list) {
343345
}
344346

345347
/**
346-
* The type of index that wildcard patterns can match.
348+
* The type of index that wildcard patterns can match. Supported values are <code>all</code>, <code>open</code>,
349+
* <code>closed</code>, <code>hidden</code>, and <code>none</code>.
347350
* <p>
348351
* API name: {@code expand_wildcards}
349352
* </p>

java-client/src/generated/java/org/opensearch/client/opensearch/cat/SegmentReplicationRequest.java

+6-3
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,8 @@ public final Boolean detailed() {
183183
}
184184

185185
/**
186-
* Whether to expand the wildcard expression to include concrete indexes that are open, closed, or both.
186+
* The type of index that wildcard patterns can match. Supported values are <code>all</code>, <code>open</code>, <code>closed</code>,
187+
* <code>hidden</code>, and <code>none</code>.
187188
* <p>
188189
* API name: {@code expand_wildcards}
189190
* </p>
@@ -413,7 +414,8 @@ public final Builder detailed(@Nullable Boolean value) {
413414
}
414415

415416
/**
416-
* Whether to expand the wildcard expression to include concrete indexes that are open, closed, or both.
417+
* The type of index that wildcard patterns can match. Supported values are <code>all</code>, <code>open</code>,
418+
* <code>closed</code>, <code>hidden</code>, and <code>none</code>.
417419
* <p>
418420
* API name: {@code expand_wildcards}
419421
* </p>
@@ -429,7 +431,8 @@ public final Builder expandWildcards(List<ExpandWildcard> list) {
429431
}
430432

431433
/**
432-
* Whether to expand the wildcard expression to include concrete indexes that are open, closed, or both.
434+
* The type of index that wildcard patterns can match. Supported values are <code>all</code>, <code>open</code>,
435+
* <code>closed</code>, <code>hidden</code>, and <code>none</code>.
433436
* <p>
434437
* API name: {@code expand_wildcards}
435438
* </p>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,268 @@
1+
/*
2+
* SPDX-License-Identifier: Apache-2.0
3+
*
4+
* The OpenSearch Contributors require contributions made to
5+
* this file be licensed under the Apache-2.0 license or a
6+
* compatible open source license.
7+
*/
8+
9+
//----------------------------------------------------
10+
// THIS CODE IS GENERATED. MANUAL EDITS WILL BE LOST.
11+
//----------------------------------------------------
12+
13+
package org.opensearch.client.opensearch.ml;
14+
15+
import jakarta.json.stream.JsonGenerator;
16+
import java.util.HashMap;
17+
import java.util.Map;
18+
import java.util.Objects;
19+
import java.util.function.Function;
20+
import javax.annotation.Generated;
21+
import javax.annotation.Nonnull;
22+
import javax.annotation.Nullable;
23+
import org.opensearch.client.json.JsonpDeserializable;
24+
import org.opensearch.client.json.JsonpDeserializer;
25+
import org.opensearch.client.json.JsonpMapper;
26+
import org.opensearch.client.json.ObjectBuilderDeserializer;
27+
import org.opensearch.client.json.ObjectDeserializer;
28+
import org.opensearch.client.json.PlainJsonSerializable;
29+
import org.opensearch.client.opensearch._types.ErrorResponse;
30+
import org.opensearch.client.opensearch._types.RequestBase;
31+
import org.opensearch.client.transport.Endpoint;
32+
import org.opensearch.client.transport.endpoints.SimpleEndpoint;
33+
import org.opensearch.client.util.ApiTypeHelper;
34+
import org.opensearch.client.util.CopyableBuilder;
35+
import org.opensearch.client.util.ObjectBuilder;
36+
import org.opensearch.client.util.ToCopyableBuilder;
37+
38+
// typedef: ml.create_controller.Request
39+
40+
/**
41+
* Creates a controller.
42+
*/
43+
@JsonpDeserializable
44+
@Generated("org.opensearch.client.codegen.CodeGenerator")
45+
public final class CreateControllerRequest extends RequestBase
46+
implements
47+
PlainJsonSerializable,
48+
ToCopyableBuilder<CreateControllerRequest.Builder, CreateControllerRequest> {
49+
50+
@Nonnull
51+
private final String modelId;
52+
53+
@Nonnull
54+
private final Map<String, RateLimiter> userRateLimiter;
55+
56+
// ---------------------------------------------------------------------------------------------
57+
58+
private CreateControllerRequest(Builder builder) {
59+
super(builder);
60+
this.modelId = ApiTypeHelper.requireNonNull(builder.modelId, this, "modelId");
61+
this.userRateLimiter = ApiTypeHelper.unmodifiable(builder.userRateLimiter);
62+
}
63+
64+
public static CreateControllerRequest of(Function<CreateControllerRequest.Builder, ObjectBuilder<CreateControllerRequest>> fn) {
65+
return fn.apply(new Builder()).build();
66+
}
67+
68+
/**
69+
* Required - API name: {@code model_id}
70+
*/
71+
@Nonnull
72+
public final String modelId() {
73+
return this.modelId;
74+
}
75+
76+
/**
77+
* API name: {@code user_rate_limiter}
78+
*/
79+
@Nonnull
80+
public final Map<String, RateLimiter> userRateLimiter() {
81+
return this.userRateLimiter;
82+
}
83+
84+
/**
85+
* Serialize this object to JSON.
86+
*/
87+
@Override
88+
public void serialize(JsonGenerator generator, JsonpMapper mapper) {
89+
generator.writeStartObject();
90+
serializeInternal(generator, mapper);
91+
generator.writeEnd();
92+
}
93+
94+
protected void serializeInternal(JsonGenerator generator, JsonpMapper mapper) {
95+
if (ApiTypeHelper.isDefined(this.userRateLimiter)) {
96+
generator.writeKey("user_rate_limiter");
97+
generator.writeStartObject();
98+
for (Map.Entry<String, RateLimiter> item0 : this.userRateLimiter.entrySet()) {
99+
generator.writeKey(item0.getKey());
100+
item0.getValue().serialize(generator, mapper);
101+
}
102+
generator.writeEnd();
103+
}
104+
}
105+
// ---------------------------------------------------------------------------------------------
106+
107+
@Override
108+
@Nonnull
109+
public Builder toBuilder() {
110+
return new Builder(this);
111+
}
112+
113+
@Nonnull
114+
public static Builder builder() {
115+
return new Builder();
116+
}
117+
118+
/**
119+
* Builder for {@link CreateControllerRequest}.
120+
*/
121+
public static class Builder extends RequestBase.AbstractBuilder<Builder> implements CopyableBuilder<Builder, CreateControllerRequest> {
122+
private String modelId;
123+
@Nullable
124+
private Map<String, RateLimiter> userRateLimiter;
125+
126+
public Builder() {}
127+
128+
private Builder(CreateControllerRequest o) {
129+
super(o);
130+
this.modelId = o.modelId;
131+
this.userRateLimiter = _mapCopy(o.userRateLimiter);
132+
}
133+
134+
private Builder(Builder o) {
135+
super(o);
136+
this.modelId = o.modelId;
137+
this.userRateLimiter = _mapCopy(o.userRateLimiter);
138+
}
139+
140+
@Override
141+
@Nonnull
142+
public Builder copy() {
143+
return new Builder(this);
144+
}
145+
146+
@Override
147+
@Nonnull
148+
protected Builder self() {
149+
return this;
150+
}
151+
152+
/**
153+
* Required - API name: {@code model_id}
154+
*/
155+
@Nonnull
156+
public final Builder modelId(String value) {
157+
this.modelId = value;
158+
return this;
159+
}
160+
161+
/**
162+
* API name: {@code user_rate_limiter}
163+
*
164+
* <p>
165+
* Adds all elements of <code>map</code> to <code>userRateLimiter</code>.
166+
* </p>
167+
*/
168+
@Nonnull
169+
public final Builder userRateLimiter(Map<String, RateLimiter> map) {
170+
this.userRateLimiter = _mapPutAll(this.userRateLimiter, map);
171+
return this;
172+
}
173+
174+
/**
175+
* API name: {@code user_rate_limiter}
176+
*
177+
* <p>
178+
* Adds an entry to <code>userRateLimiter</code>.
179+
* </p>
180+
*/
181+
@Nonnull
182+
public final Builder userRateLimiter(String key, RateLimiter value) {
183+
this.userRateLimiter = _mapPut(this.userRateLimiter, key, value);
184+
return this;
185+
}
186+
187+
/**
188+
* API name: {@code user_rate_limiter}
189+
*
190+
* <p>
191+
* Adds a value to <code>userRateLimiter</code> using a builder lambda.
192+
* </p>
193+
*/
194+
@Nonnull
195+
public final Builder userRateLimiter(String key, Function<RateLimiter.Builder, ObjectBuilder<RateLimiter>> fn) {
196+
return userRateLimiter(key, fn.apply(new RateLimiter.Builder()).build());
197+
}
198+
199+
/**
200+
* Builds a {@link CreateControllerRequest}.
201+
*
202+
* @throws NullPointerException if some of the required fields are null.
203+
*/
204+
@Override
205+
@Nonnull
206+
public CreateControllerRequest build() {
207+
_checkSingleUse();
208+
209+
return new CreateControllerRequest(this);
210+
}
211+
}
212+
213+
// ---------------------------------------------------------------------------------------------
214+
215+
/**
216+
* Json deserializer for {@link CreateControllerRequest}
217+
*/
218+
public static final JsonpDeserializer<CreateControllerRequest> _DESERIALIZER = ObjectBuilderDeserializer.lazy(
219+
Builder::new,
220+
CreateControllerRequest::setupCreateControllerRequestDeserializer
221+
);
222+
223+
protected static void setupCreateControllerRequestDeserializer(ObjectDeserializer<CreateControllerRequest.Builder> op) {
224+
op.add(Builder::userRateLimiter, JsonpDeserializer.stringMapDeserializer(RateLimiter._DESERIALIZER), "user_rate_limiter");
225+
}
226+
227+
// ---------------------------------------------------------------------------------------------
228+
229+
/**
230+
* Endpoint "{@code ml.create_controller}".
231+
*/
232+
public static final Endpoint<CreateControllerRequest, CreateControllerResponse, ErrorResponse> _ENDPOINT = new SimpleEndpoint<>(
233+
// Request method
234+
request -> "POST",
235+
// Request path
236+
request -> {
237+
StringBuilder buf = new StringBuilder();
238+
buf.append("/_plugins/_ml/controllers/");
239+
SimpleEndpoint.pathEncode(request.modelId, buf);
240+
return buf.toString();
241+
},
242+
// Request parameters
243+
request -> {
244+
Map<String, String> params = new HashMap<>();
245+
request.applyQueryParameters(params);
246+
return params;
247+
},
248+
SimpleEndpoint.emptyMap(),
249+
true,
250+
CreateControllerResponse._DESERIALIZER
251+
);
252+
253+
@Override
254+
public int hashCode() {
255+
int result = 17;
256+
result = 31 * result + this.modelId.hashCode();
257+
result = 31 * result + Objects.hashCode(this.userRateLimiter);
258+
return result;
259+
}
260+
261+
@Override
262+
public boolean equals(Object o) {
263+
if (this == o) return true;
264+
if (o == null || this.getClass() != o.getClass()) return false;
265+
CreateControllerRequest other = (CreateControllerRequest) o;
266+
return this.modelId.equals(other.modelId) && Objects.equals(this.userRateLimiter, other.userRateLimiter);
267+
}
268+
}

0 commit comments

Comments
 (0)