Skip to content

Commit 1e1c2db

Browse files
feat: Implement configurable recommenders and update .bazel files (googleapis#778)
- [ ] Regenerate this pull request now. PiperOrigin-RevId: 457101611 Source-Link: googleapis/googleapis@c13c88b Source-Link: https://github.com/googleapis/googleapis-gen/commit/4eb3716b18c24627d23a5998c1d3f95edcc897c3 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNGViMzcxNmIxOGMyNDYyN2QyM2E1OTk4YzFkM2Y5NWVkY2M4OTdjMyJ9
1 parent 0b6c2b0 commit 1e1c2db

55 files changed

Lines changed: 15367 additions & 304 deletions

File tree

Some content is hidden

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

java-recommender/google-cloud-recommender/src/main/java/com/google/cloud/recommender/v1/RecommenderClient.java

Lines changed: 449 additions & 0 deletions
Large diffs are not rendered by default.

java-recommender/google-cloud-recommender/src/main/java/com/google/cloud/recommender/v1/RecommenderSettings.java

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,30 @@ public UnaryCallSettings<GetRecommendationRequest, Recommendation> getRecommenda
118118
return ((RecommenderStubSettings) getStubSettings()).markRecommendationFailedSettings();
119119
}
120120

121+
/** Returns the object with the settings used for calls to getRecommenderConfig. */
122+
public UnaryCallSettings<GetRecommenderConfigRequest, RecommenderConfig>
123+
getRecommenderConfigSettings() {
124+
return ((RecommenderStubSettings) getStubSettings()).getRecommenderConfigSettings();
125+
}
126+
127+
/** Returns the object with the settings used for calls to updateRecommenderConfig. */
128+
public UnaryCallSettings<UpdateRecommenderConfigRequest, RecommenderConfig>
129+
updateRecommenderConfigSettings() {
130+
return ((RecommenderStubSettings) getStubSettings()).updateRecommenderConfigSettings();
131+
}
132+
133+
/** Returns the object with the settings used for calls to getInsightTypeConfig. */
134+
public UnaryCallSettings<GetInsightTypeConfigRequest, InsightTypeConfig>
135+
getInsightTypeConfigSettings() {
136+
return ((RecommenderStubSettings) getStubSettings()).getInsightTypeConfigSettings();
137+
}
138+
139+
/** Returns the object with the settings used for calls to updateInsightTypeConfig. */
140+
public UnaryCallSettings<UpdateInsightTypeConfigRequest, InsightTypeConfig>
141+
updateInsightTypeConfigSettings() {
142+
return ((RecommenderStubSettings) getStubSettings()).updateInsightTypeConfigSettings();
143+
}
144+
121145
public static final RecommenderSettings create(RecommenderStubSettings stub) throws IOException {
122146
return new RecommenderSettings.Builder(stub.toBuilder()).build();
123147
}
@@ -283,6 +307,30 @@ public UnaryCallSettings.Builder<GetInsightRequest, Insight> getInsightSettings(
283307
return getStubSettingsBuilder().markRecommendationFailedSettings();
284308
}
285309

310+
/** Returns the builder for the settings used for calls to getRecommenderConfig. */
311+
public UnaryCallSettings.Builder<GetRecommenderConfigRequest, RecommenderConfig>
312+
getRecommenderConfigSettings() {
313+
return getStubSettingsBuilder().getRecommenderConfigSettings();
314+
}
315+
316+
/** Returns the builder for the settings used for calls to updateRecommenderConfig. */
317+
public UnaryCallSettings.Builder<UpdateRecommenderConfigRequest, RecommenderConfig>
318+
updateRecommenderConfigSettings() {
319+
return getStubSettingsBuilder().updateRecommenderConfigSettings();
320+
}
321+
322+
/** Returns the builder for the settings used for calls to getInsightTypeConfig. */
323+
public UnaryCallSettings.Builder<GetInsightTypeConfigRequest, InsightTypeConfig>
324+
getInsightTypeConfigSettings() {
325+
return getStubSettingsBuilder().getInsightTypeConfigSettings();
326+
}
327+
328+
/** Returns the builder for the settings used for calls to updateInsightTypeConfig. */
329+
public UnaryCallSettings.Builder<UpdateInsightTypeConfigRequest, InsightTypeConfig>
330+
updateInsightTypeConfigSettings() {
331+
return getStubSettingsBuilder().updateInsightTypeConfigSettings();
332+
}
333+
286334
@Override
287335
public RecommenderSettings build() throws IOException {
288336
return new RecommenderSettings(this);

java-recommender/google-cloud-recommender/src/main/java/com/google/cloud/recommender/v1/gapic_metadata.json

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,15 @@
1313
"GetInsight": {
1414
"methods": ["getInsight", "getInsight", "getInsight", "getInsightCallable"]
1515
},
16+
"GetInsightTypeConfig": {
17+
"methods": ["getInsightTypeConfig", "getInsightTypeConfig", "getInsightTypeConfig", "getInsightTypeConfigCallable"]
18+
},
1619
"GetRecommendation": {
1720
"methods": ["getRecommendation", "getRecommendation", "getRecommendation", "getRecommendationCallable"]
1821
},
22+
"GetRecommenderConfig": {
23+
"methods": ["getRecommenderConfig", "getRecommenderConfig", "getRecommenderConfig", "getRecommenderConfigCallable"]
24+
},
1925
"ListInsights": {
2026
"methods": ["listInsights", "listInsights", "listInsights", "listInsightsPagedCallable", "listInsightsCallable"]
2127
},
@@ -33,6 +39,12 @@
3339
},
3440
"MarkRecommendationSucceeded": {
3541
"methods": ["markRecommendationSucceeded", "markRecommendationSucceeded", "markRecommendationSucceeded", "markRecommendationSucceededCallable"]
42+
},
43+
"UpdateInsightTypeConfig": {
44+
"methods": ["updateInsightTypeConfig", "updateInsightTypeConfig", "updateInsightTypeConfigCallable"]
45+
},
46+
"UpdateRecommenderConfig": {
47+
"methods": ["updateRecommenderConfig", "updateRecommenderConfig", "updateRecommenderConfigCallable"]
3648
}
3749
}
3850
}

java-recommender/google-cloud-recommender/src/main/java/com/google/cloud/recommender/v1/stub/GrpcRecommenderStub.java

Lines changed: 146 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,11 @@
2626
import com.google.api.gax.rpc.ClientContext;
2727
import com.google.api.gax.rpc.UnaryCallable;
2828
import com.google.cloud.recommender.v1.GetInsightRequest;
29+
import com.google.cloud.recommender.v1.GetInsightTypeConfigRequest;
2930
import com.google.cloud.recommender.v1.GetRecommendationRequest;
31+
import com.google.cloud.recommender.v1.GetRecommenderConfigRequest;
3032
import com.google.cloud.recommender.v1.Insight;
33+
import com.google.cloud.recommender.v1.InsightTypeConfig;
3134
import com.google.cloud.recommender.v1.ListInsightsRequest;
3235
import com.google.cloud.recommender.v1.ListInsightsResponse;
3336
import com.google.cloud.recommender.v1.ListRecommendationsRequest;
@@ -37,6 +40,9 @@
3740
import com.google.cloud.recommender.v1.MarkRecommendationFailedRequest;
3841
import com.google.cloud.recommender.v1.MarkRecommendationSucceededRequest;
3942
import com.google.cloud.recommender.v1.Recommendation;
43+
import com.google.cloud.recommender.v1.RecommenderConfig;
44+
import com.google.cloud.recommender.v1.UpdateInsightTypeConfigRequest;
45+
import com.google.cloud.recommender.v1.UpdateRecommenderConfigRequest;
4046
import com.google.common.collect.ImmutableMap;
4147
import com.google.longrunning.stub.GrpcOperationsStub;
4248
import io.grpc.MethodDescriptor;
@@ -134,6 +140,46 @@ public class GrpcRecommenderStub extends RecommenderStub {
134140
.setResponseMarshaller(ProtoUtils.marshaller(Recommendation.getDefaultInstance()))
135141
.build();
136142

143+
private static final MethodDescriptor<GetRecommenderConfigRequest, RecommenderConfig>
144+
getRecommenderConfigMethodDescriptor =
145+
MethodDescriptor.<GetRecommenderConfigRequest, RecommenderConfig>newBuilder()
146+
.setType(MethodDescriptor.MethodType.UNARY)
147+
.setFullMethodName("google.cloud.recommender.v1.Recommender/GetRecommenderConfig")
148+
.setRequestMarshaller(
149+
ProtoUtils.marshaller(GetRecommenderConfigRequest.getDefaultInstance()))
150+
.setResponseMarshaller(ProtoUtils.marshaller(RecommenderConfig.getDefaultInstance()))
151+
.build();
152+
153+
private static final MethodDescriptor<UpdateRecommenderConfigRequest, RecommenderConfig>
154+
updateRecommenderConfigMethodDescriptor =
155+
MethodDescriptor.<UpdateRecommenderConfigRequest, RecommenderConfig>newBuilder()
156+
.setType(MethodDescriptor.MethodType.UNARY)
157+
.setFullMethodName("google.cloud.recommender.v1.Recommender/UpdateRecommenderConfig")
158+
.setRequestMarshaller(
159+
ProtoUtils.marshaller(UpdateRecommenderConfigRequest.getDefaultInstance()))
160+
.setResponseMarshaller(ProtoUtils.marshaller(RecommenderConfig.getDefaultInstance()))
161+
.build();
162+
163+
private static final MethodDescriptor<GetInsightTypeConfigRequest, InsightTypeConfig>
164+
getInsightTypeConfigMethodDescriptor =
165+
MethodDescriptor.<GetInsightTypeConfigRequest, InsightTypeConfig>newBuilder()
166+
.setType(MethodDescriptor.MethodType.UNARY)
167+
.setFullMethodName("google.cloud.recommender.v1.Recommender/GetInsightTypeConfig")
168+
.setRequestMarshaller(
169+
ProtoUtils.marshaller(GetInsightTypeConfigRequest.getDefaultInstance()))
170+
.setResponseMarshaller(ProtoUtils.marshaller(InsightTypeConfig.getDefaultInstance()))
171+
.build();
172+
173+
private static final MethodDescriptor<UpdateInsightTypeConfigRequest, InsightTypeConfig>
174+
updateInsightTypeConfigMethodDescriptor =
175+
MethodDescriptor.<UpdateInsightTypeConfigRequest, InsightTypeConfig>newBuilder()
176+
.setType(MethodDescriptor.MethodType.UNARY)
177+
.setFullMethodName("google.cloud.recommender.v1.Recommender/UpdateInsightTypeConfig")
178+
.setRequestMarshaller(
179+
ProtoUtils.marshaller(UpdateInsightTypeConfigRequest.getDefaultInstance()))
180+
.setResponseMarshaller(ProtoUtils.marshaller(InsightTypeConfig.getDefaultInstance()))
181+
.build();
182+
137183
private final UnaryCallable<ListInsightsRequest, ListInsightsResponse> listInsightsCallable;
138184
private final UnaryCallable<ListInsightsRequest, ListInsightsPagedResponse>
139185
listInsightsPagedCallable;
@@ -150,6 +196,14 @@ public class GrpcRecommenderStub extends RecommenderStub {
150196
markRecommendationSucceededCallable;
151197
private final UnaryCallable<MarkRecommendationFailedRequest, Recommendation>
152198
markRecommendationFailedCallable;
199+
private final UnaryCallable<GetRecommenderConfigRequest, RecommenderConfig>
200+
getRecommenderConfigCallable;
201+
private final UnaryCallable<UpdateRecommenderConfigRequest, RecommenderConfig>
202+
updateRecommenderConfigCallable;
203+
private final UnaryCallable<GetInsightTypeConfigRequest, InsightTypeConfig>
204+
getInsightTypeConfigCallable;
205+
private final UnaryCallable<UpdateInsightTypeConfigRequest, InsightTypeConfig>
206+
updateInsightTypeConfigCallable;
153207

154208
private final BackgroundResource backgroundResources;
155209
private final GrpcOperationsStub operationsStub;
@@ -277,6 +331,54 @@ protected GrpcRecommenderStub(
277331
return params.build();
278332
})
279333
.build();
334+
GrpcCallSettings<GetRecommenderConfigRequest, RecommenderConfig>
335+
getRecommenderConfigTransportSettings =
336+
GrpcCallSettings.<GetRecommenderConfigRequest, RecommenderConfig>newBuilder()
337+
.setMethodDescriptor(getRecommenderConfigMethodDescriptor)
338+
.setParamsExtractor(
339+
request -> {
340+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
341+
params.put("name", String.valueOf(request.getName()));
342+
return params.build();
343+
})
344+
.build();
345+
GrpcCallSettings<UpdateRecommenderConfigRequest, RecommenderConfig>
346+
updateRecommenderConfigTransportSettings =
347+
GrpcCallSettings.<UpdateRecommenderConfigRequest, RecommenderConfig>newBuilder()
348+
.setMethodDescriptor(updateRecommenderConfigMethodDescriptor)
349+
.setParamsExtractor(
350+
request -> {
351+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
352+
params.put(
353+
"recommender_config.name",
354+
String.valueOf(request.getRecommenderConfig().getName()));
355+
return params.build();
356+
})
357+
.build();
358+
GrpcCallSettings<GetInsightTypeConfigRequest, InsightTypeConfig>
359+
getInsightTypeConfigTransportSettings =
360+
GrpcCallSettings.<GetInsightTypeConfigRequest, InsightTypeConfig>newBuilder()
361+
.setMethodDescriptor(getInsightTypeConfigMethodDescriptor)
362+
.setParamsExtractor(
363+
request -> {
364+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
365+
params.put("name", String.valueOf(request.getName()));
366+
return params.build();
367+
})
368+
.build();
369+
GrpcCallSettings<UpdateInsightTypeConfigRequest, InsightTypeConfig>
370+
updateInsightTypeConfigTransportSettings =
371+
GrpcCallSettings.<UpdateInsightTypeConfigRequest, InsightTypeConfig>newBuilder()
372+
.setMethodDescriptor(updateInsightTypeConfigMethodDescriptor)
373+
.setParamsExtractor(
374+
request -> {
375+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
376+
params.put(
377+
"insight_type_config.name",
378+
String.valueOf(request.getInsightTypeConfig().getName()));
379+
return params.build();
380+
})
381+
.build();
280382

281383
this.listInsightsCallable =
282384
callableFactory.createUnaryCallable(
@@ -322,6 +424,26 @@ protected GrpcRecommenderStub(
322424
markRecommendationFailedTransportSettings,
323425
settings.markRecommendationFailedSettings(),
324426
clientContext);
427+
this.getRecommenderConfigCallable =
428+
callableFactory.createUnaryCallable(
429+
getRecommenderConfigTransportSettings,
430+
settings.getRecommenderConfigSettings(),
431+
clientContext);
432+
this.updateRecommenderConfigCallable =
433+
callableFactory.createUnaryCallable(
434+
updateRecommenderConfigTransportSettings,
435+
settings.updateRecommenderConfigSettings(),
436+
clientContext);
437+
this.getInsightTypeConfigCallable =
438+
callableFactory.createUnaryCallable(
439+
getInsightTypeConfigTransportSettings,
440+
settings.getInsightTypeConfigSettings(),
441+
clientContext);
442+
this.updateInsightTypeConfigCallable =
443+
callableFactory.createUnaryCallable(
444+
updateInsightTypeConfigTransportSettings,
445+
settings.updateInsightTypeConfigSettings(),
446+
clientContext);
325447

326448
this.backgroundResources =
327449
new BackgroundResourceAggregation(clientContext.getBackgroundResources());
@@ -386,6 +508,30 @@ public UnaryCallable<GetRecommendationRequest, Recommendation> getRecommendation
386508
return markRecommendationFailedCallable;
387509
}
388510

511+
@Override
512+
public UnaryCallable<GetRecommenderConfigRequest, RecommenderConfig>
513+
getRecommenderConfigCallable() {
514+
return getRecommenderConfigCallable;
515+
}
516+
517+
@Override
518+
public UnaryCallable<UpdateRecommenderConfigRequest, RecommenderConfig>
519+
updateRecommenderConfigCallable() {
520+
return updateRecommenderConfigCallable;
521+
}
522+
523+
@Override
524+
public UnaryCallable<GetInsightTypeConfigRequest, InsightTypeConfig>
525+
getInsightTypeConfigCallable() {
526+
return getInsightTypeConfigCallable;
527+
}
528+
529+
@Override
530+
public UnaryCallable<UpdateInsightTypeConfigRequest, InsightTypeConfig>
531+
updateInsightTypeConfigCallable() {
532+
return updateInsightTypeConfigCallable;
533+
}
534+
389535
@Override
390536
public final void close() {
391537
try {

0 commit comments

Comments
 (0)