Skip to content
This repository was archived by the owner on Sep 9, 2023. It is now read-only.

Commit 85be17b

Browse files
gcf-owl-bot[bot]gcf-owl-bot[bot]
authored
feat: add ListSavedQueries rpc to aiplatform v1 dataset_service.proto (#975)
- [ ] Regenerate this pull request now. feat: add saved_query.proto to aiplatform v1 feat: add saved_query_id to InputDataConfig in aiplatform v1 training_pipeline.proto PiperOrigin-RevId: 456872211 Source-Link: googleapis/googleapis@88f3973 Source-Link: https://github.com/googleapis/googleapis-gen/commit/daf8d1cb20fadabdd8914581da1ceeef048a34f7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZGFmOGQxY2IyMGZhZGFiZGQ4OTE0NTgxZGExY2VlZWYwNDhhMzRmNyJ9 feat: add ListSavedQueries rpc to aiplatform v1beta1 dataset_service.proto feat: add saved_query.proto to aiplatform v1beta1 feat: add saved_query_id to InputDataConfig in aiplatform v1beta1 training_pipeline.proto PiperOrigin-RevId: 456872207 Source-Link: googleapis/googleapis@aeb384a Source-Link: https://github.com/googleapis/googleapis-gen/commit/c5c8894abbfcd1f6b9d38d9e9fd70a18f15cf0cc Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYzVjODg5NGFiYmZjZDFmNmI5ZDM4ZDllOWZkNzBhMThmMTVjZjBjYyJ9
1 parent 3041b97 commit 85be17b

File tree

48 files changed

+14287
-275
lines changed

Some content is hidden

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

48 files changed

+14287
-275
lines changed

‎google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/DatasetServiceClient.java‎

Lines changed: 238 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1212,6 +1212,167 @@ public final UnaryCallable<ListDataItemsRequest, ListDataItemsResponse> listData
12121212
return stub.listDataItemsCallable();
12131213
}
12141214

1215+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1216+
/**
1217+
* Lists SavedQueries in a Dataset.
1218+
*
1219+
* <p>Sample code:
1220+
*
1221+
* <pre>{@code
1222+
* // This snippet has been automatically generated for illustrative purposes only.
1223+
* // It may require modifications to work in your environment.
1224+
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
1225+
* DatasetName parent = DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]");
1226+
* for (SavedQuery element : datasetServiceClient.listSavedQueries(parent).iterateAll()) {
1227+
* // doThingsWith(element);
1228+
* }
1229+
* }
1230+
* }</pre>
1231+
*
1232+
* @param parent Required. The resource name of the Dataset to list SavedQueries from. Format:
1233+
* `projects/{project}/locations/{location}/datasets/{dataset}`
1234+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
1235+
*/
1236+
public final ListSavedQueriesPagedResponse listSavedQueries(DatasetName parent) {
1237+
ListSavedQueriesRequest request =
1238+
ListSavedQueriesRequest.newBuilder()
1239+
.setParent(parent == null ? null : parent.toString())
1240+
.build();
1241+
return listSavedQueries(request);
1242+
}
1243+
1244+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1245+
/**
1246+
* Lists SavedQueries in a Dataset.
1247+
*
1248+
* <p>Sample code:
1249+
*
1250+
* <pre>{@code
1251+
* // This snippet has been automatically generated for illustrative purposes only.
1252+
* // It may require modifications to work in your environment.
1253+
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
1254+
* String parent = DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString();
1255+
* for (SavedQuery element : datasetServiceClient.listSavedQueries(parent).iterateAll()) {
1256+
* // doThingsWith(element);
1257+
* }
1258+
* }
1259+
* }</pre>
1260+
*
1261+
* @param parent Required. The resource name of the Dataset to list SavedQueries from. Format:
1262+
* `projects/{project}/locations/{location}/datasets/{dataset}`
1263+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
1264+
*/
1265+
public final ListSavedQueriesPagedResponse listSavedQueries(String parent) {
1266+
ListSavedQueriesRequest request =
1267+
ListSavedQueriesRequest.newBuilder().setParent(parent).build();
1268+
return listSavedQueries(request);
1269+
}
1270+
1271+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1272+
/**
1273+
* Lists SavedQueries in a Dataset.
1274+
*
1275+
* <p>Sample code:
1276+
*
1277+
* <pre>{@code
1278+
* // This snippet has been automatically generated for illustrative purposes only.
1279+
* // It may require modifications to work in your environment.
1280+
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
1281+
* ListSavedQueriesRequest request =
1282+
* ListSavedQueriesRequest.newBuilder()
1283+
* .setParent(DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString())
1284+
* .setFilter("filter-1274492040")
1285+
* .setPageSize(883849137)
1286+
* .setPageToken("pageToken873572522")
1287+
* .setReadMask(FieldMask.newBuilder().build())
1288+
* .setOrderBy("orderBy-1207110587")
1289+
* .build();
1290+
* for (SavedQuery element : datasetServiceClient.listSavedQueries(request).iterateAll()) {
1291+
* // doThingsWith(element);
1292+
* }
1293+
* }
1294+
* }</pre>
1295+
*
1296+
* @param request The request object containing all of the parameters for the API call.
1297+
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
1298+
*/
1299+
public final ListSavedQueriesPagedResponse listSavedQueries(ListSavedQueriesRequest request) {
1300+
return listSavedQueriesPagedCallable().call(request);
1301+
}
1302+
1303+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1304+
/**
1305+
* Lists SavedQueries in a Dataset.
1306+
*
1307+
* <p>Sample code:
1308+
*
1309+
* <pre>{@code
1310+
* // This snippet has been automatically generated for illustrative purposes only.
1311+
* // It may require modifications to work in your environment.
1312+
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
1313+
* ListSavedQueriesRequest request =
1314+
* ListSavedQueriesRequest.newBuilder()
1315+
* .setParent(DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString())
1316+
* .setFilter("filter-1274492040")
1317+
* .setPageSize(883849137)
1318+
* .setPageToken("pageToken873572522")
1319+
* .setReadMask(FieldMask.newBuilder().build())
1320+
* .setOrderBy("orderBy-1207110587")
1321+
* .build();
1322+
* ApiFuture<SavedQuery> future =
1323+
* datasetServiceClient.listSavedQueriesPagedCallable().futureCall(request);
1324+
* // Do something.
1325+
* for (SavedQuery element : future.get().iterateAll()) {
1326+
* // doThingsWith(element);
1327+
* }
1328+
* }
1329+
* }</pre>
1330+
*/
1331+
public final UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesPagedResponse>
1332+
listSavedQueriesPagedCallable() {
1333+
return stub.listSavedQueriesPagedCallable();
1334+
}
1335+
1336+
// AUTO-GENERATED DOCUMENTATION AND METHOD.
1337+
/**
1338+
* Lists SavedQueries in a Dataset.
1339+
*
1340+
* <p>Sample code:
1341+
*
1342+
* <pre>{@code
1343+
* // This snippet has been automatically generated for illustrative purposes only.
1344+
* // It may require modifications to work in your environment.
1345+
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
1346+
* ListSavedQueriesRequest request =
1347+
* ListSavedQueriesRequest.newBuilder()
1348+
* .setParent(DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString())
1349+
* .setFilter("filter-1274492040")
1350+
* .setPageSize(883849137)
1351+
* .setPageToken("pageToken873572522")
1352+
* .setReadMask(FieldMask.newBuilder().build())
1353+
* .setOrderBy("orderBy-1207110587")
1354+
* .build();
1355+
* while (true) {
1356+
* ListSavedQueriesResponse response =
1357+
* datasetServiceClient.listSavedQueriesCallable().call(request);
1358+
* for (SavedQuery element : response.getSavedQueriesList()) {
1359+
* // doThingsWith(element);
1360+
* }
1361+
* String nextPageToken = response.getNextPageToken();
1362+
* if (!Strings.isNullOrEmpty(nextPageToken)) {
1363+
* request = request.toBuilder().setPageToken(nextPageToken).build();
1364+
* } else {
1365+
* break;
1366+
* }
1367+
* }
1368+
* }
1369+
* }</pre>
1370+
*/
1371+
public final UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesResponse>
1372+
listSavedQueriesCallable() {
1373+
return stub.listSavedQueriesCallable();
1374+
}
1375+
12151376
// AUTO-GENERATED DOCUMENTATION AND METHOD.
12161377
/**
12171378
* Gets an AnnotationSpec.
@@ -1992,6 +2153,83 @@ protected ListDataItemsFixedSizeCollection createCollection(
19922153
}
19932154
}
19942155

2156+
public static class ListSavedQueriesPagedResponse
2157+
extends AbstractPagedListResponse<
2158+
ListSavedQueriesRequest,
2159+
ListSavedQueriesResponse,
2160+
SavedQuery,
2161+
ListSavedQueriesPage,
2162+
ListSavedQueriesFixedSizeCollection> {
2163+
2164+
public static ApiFuture<ListSavedQueriesPagedResponse> createAsync(
2165+
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
2166+
ApiFuture<ListSavedQueriesResponse> futureResponse) {
2167+
ApiFuture<ListSavedQueriesPage> futurePage =
2168+
ListSavedQueriesPage.createEmptyPage().createPageAsync(context, futureResponse);
2169+
return ApiFutures.transform(
2170+
futurePage,
2171+
input -> new ListSavedQueriesPagedResponse(input),
2172+
MoreExecutors.directExecutor());
2173+
}
2174+
2175+
private ListSavedQueriesPagedResponse(ListSavedQueriesPage page) {
2176+
super(page, ListSavedQueriesFixedSizeCollection.createEmptyCollection());
2177+
}
2178+
}
2179+
2180+
public static class ListSavedQueriesPage
2181+
extends AbstractPage<
2182+
ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery, ListSavedQueriesPage> {
2183+
2184+
private ListSavedQueriesPage(
2185+
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
2186+
ListSavedQueriesResponse response) {
2187+
super(context, response);
2188+
}
2189+
2190+
private static ListSavedQueriesPage createEmptyPage() {
2191+
return new ListSavedQueriesPage(null, null);
2192+
}
2193+
2194+
@Override
2195+
protected ListSavedQueriesPage createPage(
2196+
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
2197+
ListSavedQueriesResponse response) {
2198+
return new ListSavedQueriesPage(context, response);
2199+
}
2200+
2201+
@Override
2202+
public ApiFuture<ListSavedQueriesPage> createPageAsync(
2203+
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
2204+
ApiFuture<ListSavedQueriesResponse> futureResponse) {
2205+
return super.createPageAsync(context, futureResponse);
2206+
}
2207+
}
2208+
2209+
public static class ListSavedQueriesFixedSizeCollection
2210+
extends AbstractFixedSizeCollection<
2211+
ListSavedQueriesRequest,
2212+
ListSavedQueriesResponse,
2213+
SavedQuery,
2214+
ListSavedQueriesPage,
2215+
ListSavedQueriesFixedSizeCollection> {
2216+
2217+
private ListSavedQueriesFixedSizeCollection(
2218+
List<ListSavedQueriesPage> pages, int collectionSize) {
2219+
super(pages, collectionSize);
2220+
}
2221+
2222+
private static ListSavedQueriesFixedSizeCollection createEmptyCollection() {
2223+
return new ListSavedQueriesFixedSizeCollection(null, 0);
2224+
}
2225+
2226+
@Override
2227+
protected ListSavedQueriesFixedSizeCollection createCollection(
2228+
List<ListSavedQueriesPage> pages, int collectionSize) {
2229+
return new ListSavedQueriesFixedSizeCollection(pages, collectionSize);
2230+
}
2231+
}
2232+
19952233
public static class ListAnnotationsPagedResponse
19962234
extends AbstractPagedListResponse<
19972235
ListAnnotationsRequest,

‎google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/DatasetServiceSettings.java‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDataItemsPagedResponse;
2121
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetsPagedResponse;
2222
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListLocationsPagedResponse;
23+
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListSavedQueriesPagedResponse;
2324

2425
import com.google.api.core.ApiFunction;
2526
import com.google.api.core.BetaApi;
@@ -152,6 +153,13 @@ public UnaryCallSettings<ExportDataRequest, Operation> exportDataSettings() {
152153
return ((DatasetServiceStubSettings) getStubSettings()).listDataItemsSettings();
153154
}
154155

156+
/** Returns the object with the settings used for calls to listSavedQueries. */
157+
public PagedCallSettings<
158+
ListSavedQueriesRequest, ListSavedQueriesResponse, ListSavedQueriesPagedResponse>
159+
listSavedQueriesSettings() {
160+
return ((DatasetServiceStubSettings) getStubSettings()).listSavedQueriesSettings();
161+
}
162+
155163
/** Returns the object with the settings used for calls to getAnnotationSpec. */
156164
public UnaryCallSettings<GetAnnotationSpecRequest, AnnotationSpec> getAnnotationSpecSettings() {
157165
return ((DatasetServiceStubSettings) getStubSettings()).getAnnotationSpecSettings();
@@ -359,6 +367,13 @@ public UnaryCallSettings.Builder<ExportDataRequest, Operation> exportDataSetting
359367
return getStubSettingsBuilder().listDataItemsSettings();
360368
}
361369

370+
/** Returns the builder for the settings used for calls to listSavedQueries. */
371+
public PagedCallSettings.Builder<
372+
ListSavedQueriesRequest, ListSavedQueriesResponse, ListSavedQueriesPagedResponse>
373+
listSavedQueriesSettings() {
374+
return getStubSettingsBuilder().listSavedQueriesSettings();
375+
}
376+
362377
/** Returns the builder for the settings used for calls to getAnnotationSpec. */
363378
public UnaryCallSettings.Builder<GetAnnotationSpecRequest, AnnotationSpec>
364379
getAnnotationSpecSettings() {

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,9 @@
4646
"ListLocations": {
4747
"methods": ["listLocations", "listLocationsPagedCallable", "listLocationsCallable"]
4848
},
49+
"ListSavedQueries": {
50+
"methods": ["listSavedQueries", "listSavedQueries", "listSavedQueries", "listSavedQueriesPagedCallable", "listSavedQueriesCallable"]
51+
},
4952
"SetIamPolicy": {
5053
"methods": ["setIamPolicy", "setIamPolicyCallable"]
5154
},

‎google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/stub/DatasetServiceStub.java‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDataItemsPagedResponse;
2121
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetsPagedResponse;
2222
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListLocationsPagedResponse;
23+
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListSavedQueriesPagedResponse;
2324

2425
import com.google.api.gax.core.BackgroundResource;
2526
import com.google.api.gax.rpc.OperationCallable;
@@ -44,6 +45,8 @@
4445
import com.google.cloud.aiplatform.v1.ListDataItemsResponse;
4546
import com.google.cloud.aiplatform.v1.ListDatasetsRequest;
4647
import com.google.cloud.aiplatform.v1.ListDatasetsResponse;
48+
import com.google.cloud.aiplatform.v1.ListSavedQueriesRequest;
49+
import com.google.cloud.aiplatform.v1.ListSavedQueriesResponse;
4750
import com.google.cloud.aiplatform.v1.UpdateDatasetRequest;
4851
import com.google.cloud.location.GetLocationRequest;
4952
import com.google.cloud.location.ListLocationsRequest;
@@ -133,6 +136,16 @@ public UnaryCallable<ListDataItemsRequest, ListDataItemsResponse> listDataItemsC
133136
throw new UnsupportedOperationException("Not implemented: listDataItemsCallable()");
134137
}
135138

139+
public UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesPagedResponse>
140+
listSavedQueriesPagedCallable() {
141+
throw new UnsupportedOperationException("Not implemented: listSavedQueriesPagedCallable()");
142+
}
143+
144+
public UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesResponse>
145+
listSavedQueriesCallable() {
146+
throw new UnsupportedOperationException("Not implemented: listSavedQueriesCallable()");
147+
}
148+
136149
public UnaryCallable<GetAnnotationSpecRequest, AnnotationSpec> getAnnotationSpecCallable() {
137150
throw new UnsupportedOperationException("Not implemented: getAnnotationSpecCallable()");
138151
}

0 commit comments

Comments
 (0)