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

Commit

Permalink
feat: add ListSavedQueries rpc to aiplatform v1 dataset_service.proto (…
Browse files Browse the repository at this point in the history 
…#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: googleapis/googleapis-gen@daf8d1c
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: googleapis/googleapis-gen@c5c8894
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYzVjODg5NGFiYmZjZDFmNmI5ZDM4ZDllOWZkNzBhMThmMTVjZjBjYyJ9
  • Loading branch information
@gcf-owl-bot
gcf-owl-bot[bot] committed Jun 24, 2022
1 parent 3041b97 commit 85be17b
Show file tree
Hide file tree
Showing 48 changed files with 14,287 additions and 275 deletions.
238 changes: 238 additions & 0 deletions ...e-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/DatasetServiceClient.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1212,6 +1212,167 @@ public final UnaryCallable<ListDataItemsRequest, ListDataItemsResponse> listData
return stub.listDataItemsCallable();
}

// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Lists SavedQueries in a Dataset.
*
* <p>Sample code:
*
* <pre>{@code
* // This snippet has been automatically generated for illustrative purposes only.
* // It may require modifications to work in your environment.
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
* DatasetName parent = DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]");
* for (SavedQuery element : datasetServiceClient.listSavedQueries(parent).iterateAll()) {
* // doThingsWith(element);
* }
* }
* }</pre>
*
* @param parent Required. The resource name of the Dataset to list SavedQueries from. Format:
* `projects/{project}/locations/{location}/datasets/{dataset}`
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final ListSavedQueriesPagedResponse listSavedQueries(DatasetName parent) {
ListSavedQueriesRequest request =
ListSavedQueriesRequest.newBuilder()
.setParent(parent == null ? null : parent.toString())
.build();
return listSavedQueries(request);
}

// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Lists SavedQueries in a Dataset.
*
* <p>Sample code:
*
* <pre>{@code
* // This snippet has been automatically generated for illustrative purposes only.
* // It may require modifications to work in your environment.
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
* String parent = DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString();
* for (SavedQuery element : datasetServiceClient.listSavedQueries(parent).iterateAll()) {
* // doThingsWith(element);
* }
* }
* }</pre>
*
* @param parent Required. The resource name of the Dataset to list SavedQueries from. Format:
* `projects/{project}/locations/{location}/datasets/{dataset}`
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final ListSavedQueriesPagedResponse listSavedQueries(String parent) {
ListSavedQueriesRequest request =
ListSavedQueriesRequest.newBuilder().setParent(parent).build();
return listSavedQueries(request);
}

// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Lists SavedQueries in a Dataset.
*
* <p>Sample code:
*
* <pre>{@code
* // This snippet has been automatically generated for illustrative purposes only.
* // It may require modifications to work in your environment.
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
* ListSavedQueriesRequest request =
* ListSavedQueriesRequest.newBuilder()
* .setParent(DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString())
* .setFilter("filter-1274492040")
* .setPageSize(883849137)
* .setPageToken("pageToken873572522")
* .setReadMask(FieldMask.newBuilder().build())
* .setOrderBy("orderBy-1207110587")
* .build();
* for (SavedQuery element : datasetServiceClient.listSavedQueries(request).iterateAll()) {
* // doThingsWith(element);
* }
* }
* }</pre>
*
* @param request The request object containing all of the parameters for the API call.
* @throws com.google.api.gax.rpc.ApiException if the remote call fails
*/
public final ListSavedQueriesPagedResponse listSavedQueries(ListSavedQueriesRequest request) {
return listSavedQueriesPagedCallable().call(request);
}

// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Lists SavedQueries in a Dataset.
*
* <p>Sample code:
*
* <pre>{@code
* // This snippet has been automatically generated for illustrative purposes only.
* // It may require modifications to work in your environment.
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
* ListSavedQueriesRequest request =
* ListSavedQueriesRequest.newBuilder()
* .setParent(DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString())
* .setFilter("filter-1274492040")
* .setPageSize(883849137)
* .setPageToken("pageToken873572522")
* .setReadMask(FieldMask.newBuilder().build())
* .setOrderBy("orderBy-1207110587")
* .build();
* ApiFuture<SavedQuery> future =
* datasetServiceClient.listSavedQueriesPagedCallable().futureCall(request);
* // Do something.
* for (SavedQuery element : future.get().iterateAll()) {
* // doThingsWith(element);
* }
* }
* }</pre>
*/
public final UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesPagedResponse>
listSavedQueriesPagedCallable() {
return stub.listSavedQueriesPagedCallable();
}

// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Lists SavedQueries in a Dataset.
*
* <p>Sample code:
*
* <pre>{@code
* // This snippet has been automatically generated for illustrative purposes only.
* // It may require modifications to work in your environment.
* try (DatasetServiceClient datasetServiceClient = DatasetServiceClient.create()) {
* ListSavedQueriesRequest request =
* ListSavedQueriesRequest.newBuilder()
* .setParent(DatasetName.of("[PROJECT]", "[LOCATION]", "[DATASET]").toString())
* .setFilter("filter-1274492040")
* .setPageSize(883849137)
* .setPageToken("pageToken873572522")
* .setReadMask(FieldMask.newBuilder().build())
* .setOrderBy("orderBy-1207110587")
* .build();
* while (true) {
* ListSavedQueriesResponse response =
* datasetServiceClient.listSavedQueriesCallable().call(request);
* for (SavedQuery element : response.getSavedQueriesList()) {
* // doThingsWith(element);
* }
* String nextPageToken = response.getNextPageToken();
* if (!Strings.isNullOrEmpty(nextPageToken)) {
* request = request.toBuilder().setPageToken(nextPageToken).build();
* } else {
* break;
* }
* }
* }
* }</pre>
*/
public final UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesResponse>
listSavedQueriesCallable() {
return stub.listSavedQueriesCallable();
}

// AUTO-GENERATED DOCUMENTATION AND METHOD.
/**
* Gets an AnnotationSpec.
Expand Down Expand Up @@ -1992,6 +2153,83 @@ protected ListDataItemsFixedSizeCollection createCollection(
}
}

public static class ListSavedQueriesPagedResponse
extends AbstractPagedListResponse<
ListSavedQueriesRequest,
ListSavedQueriesResponse,
SavedQuery,
ListSavedQueriesPage,
ListSavedQueriesFixedSizeCollection> {

public static ApiFuture<ListSavedQueriesPagedResponse> createAsync(
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
ApiFuture<ListSavedQueriesResponse> futureResponse) {
ApiFuture<ListSavedQueriesPage> futurePage =
ListSavedQueriesPage.createEmptyPage().createPageAsync(context, futureResponse);
return ApiFutures.transform(
futurePage,
input -> new ListSavedQueriesPagedResponse(input),
MoreExecutors.directExecutor());
}

private ListSavedQueriesPagedResponse(ListSavedQueriesPage page) {
super(page, ListSavedQueriesFixedSizeCollection.createEmptyCollection());
}
}

public static class ListSavedQueriesPage
extends AbstractPage<
ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery, ListSavedQueriesPage> {

private ListSavedQueriesPage(
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
ListSavedQueriesResponse response) {
super(context, response);
}

private static ListSavedQueriesPage createEmptyPage() {
return new ListSavedQueriesPage(null, null);
}

@Override
protected ListSavedQueriesPage createPage(
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
ListSavedQueriesResponse response) {
return new ListSavedQueriesPage(context, response);
}

@Override
public ApiFuture<ListSavedQueriesPage> createPageAsync(
PageContext<ListSavedQueriesRequest, ListSavedQueriesResponse, SavedQuery> context,
ApiFuture<ListSavedQueriesResponse> futureResponse) {
return super.createPageAsync(context, futureResponse);
}
}

public static class ListSavedQueriesFixedSizeCollection
extends AbstractFixedSizeCollection<
ListSavedQueriesRequest,
ListSavedQueriesResponse,
SavedQuery,
ListSavedQueriesPage,
ListSavedQueriesFixedSizeCollection> {

private ListSavedQueriesFixedSizeCollection(
List<ListSavedQueriesPage> pages, int collectionSize) {
super(pages, collectionSize);
}

private static ListSavedQueriesFixedSizeCollection createEmptyCollection() {
return new ListSavedQueriesFixedSizeCollection(null, 0);
}

@Override
protected ListSavedQueriesFixedSizeCollection createCollection(
List<ListSavedQueriesPage> pages, int collectionSize) {
return new ListSavedQueriesFixedSizeCollection(pages, collectionSize);
}
}

public static class ListAnnotationsPagedResponse
extends AbstractPagedListResponse<
ListAnnotationsRequest,
Expand Down
15 changes: 15 additions & 0 deletions ...cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/DatasetServiceSettings.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDataItemsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListLocationsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListSavedQueriesPagedResponse;

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

/** Returns the object with the settings used for calls to listSavedQueries. */
public PagedCallSettings<
ListSavedQueriesRequest, ListSavedQueriesResponse, ListSavedQueriesPagedResponse>
listSavedQueriesSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).listSavedQueriesSettings();
}

/** Returns the object with the settings used for calls to getAnnotationSpec. */
public UnaryCallSettings<GetAnnotationSpecRequest, AnnotationSpec> getAnnotationSpecSettings() {
return ((DatasetServiceStubSettings) getStubSettings()).getAnnotationSpecSettings();
Expand Down Expand Up @@ -359,6 +367,13 @@ public UnaryCallSettings.Builder<ExportDataRequest, Operation> exportDataSetting
return getStubSettingsBuilder().listDataItemsSettings();
}

/** Returns the builder for the settings used for calls to listSavedQueries. */
public PagedCallSettings.Builder<
ListSavedQueriesRequest, ListSavedQueriesResponse, ListSavedQueriesPagedResponse>
listSavedQueriesSettings() {
return getStubSettingsBuilder().listSavedQueriesSettings();
}

/** Returns the builder for the settings used for calls to getAnnotationSpec. */
public UnaryCallSettings.Builder<GetAnnotationSpecRequest, AnnotationSpec>
getAnnotationSpecSettings() {
Expand Down
3 changes: 3 additions & 0 deletions google-cloud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/gapic_metadata.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,9 @@
"ListLocations": {
"methods": ["listLocations", "listLocationsPagedCallable", "listLocationsCallable"]
},
"ListSavedQueries": {
"methods": ["listSavedQueries", "listSavedQueries", "listSavedQueries", "listSavedQueriesPagedCallable", "listSavedQueriesCallable"]
},
"SetIamPolicy": {
"methods": ["setIamPolicy", "setIamPolicyCallable"]
},
Expand Down
13 changes: 13 additions & 0 deletions ...loud-aiplatform/src/main/java/com/google/cloud/aiplatform/v1/stub/DatasetServiceStub.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDataItemsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListDatasetsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListLocationsPagedResponse;
import static com.google.cloud.aiplatform.v1.DatasetServiceClient.ListSavedQueriesPagedResponse;

import com.google.api.gax.core.BackgroundResource;
import com.google.api.gax.rpc.OperationCallable;
Expand All @@ -44,6 +45,8 @@
import com.google.cloud.aiplatform.v1.ListDataItemsResponse;
import com.google.cloud.aiplatform.v1.ListDatasetsRequest;
import com.google.cloud.aiplatform.v1.ListDatasetsResponse;
import com.google.cloud.aiplatform.v1.ListSavedQueriesRequest;
import com.google.cloud.aiplatform.v1.ListSavedQueriesResponse;
import com.google.cloud.aiplatform.v1.UpdateDatasetRequest;
import com.google.cloud.location.GetLocationRequest;
import com.google.cloud.location.ListLocationsRequest;
Expand Down Expand Up @@ -133,6 +136,16 @@ public UnaryCallable<ListDataItemsRequest, ListDataItemsResponse> listDataItemsC
throw new UnsupportedOperationException("Not implemented: listDataItemsCallable()");
}

public UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesPagedResponse>
listSavedQueriesPagedCallable() {
throw new UnsupportedOperationException("Not implemented: listSavedQueriesPagedCallable()");
}

public UnaryCallable<ListSavedQueriesRequest, ListSavedQueriesResponse>
listSavedQueriesCallable() {
throw new UnsupportedOperationException("Not implemented: listSavedQueriesCallable()");
}

public UnaryCallable<GetAnnotationSpecRequest, AnnotationSpec> getAnnotationSpecCallable() {
throw new UnsupportedOperationException("Not implemented: getAnnotationSpecCallable()");
}
Expand Down
Loading

0 comments on commit 85be17b

Please sign in to comment.