All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.google.cloud.discoveryengine.v1.stub.HttpJsonDataStoreServiceStub Maven / Gradle / Ivy

Go to download

Discovery Engine API A Cloud API that offers search and recommendation discoverability for documents from different industry verticals (e.g. media, retail, etc.).

There is a newer version: 0.51.0
Show newest version
/*
 * Copyright 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.cloud.discoveryengine.v1.stub;

import static com.google.cloud.discoveryengine.v1.DataStoreServiceClient.ListDataStoresPagedResponse;

import com.google.api.HttpRule;
import com.google.api.core.InternalApi;
import com.google.api.gax.core.BackgroundResource;
import com.google.api.gax.core.BackgroundResourceAggregation;
import com.google.api.gax.httpjson.ApiMethodDescriptor;
import com.google.api.gax.httpjson.HttpJsonCallSettings;
import com.google.api.gax.httpjson.HttpJsonOperationSnapshot;
import com.google.api.gax.httpjson.HttpJsonStubCallableFactory;
import com.google.api.gax.httpjson.ProtoMessageRequestFormatter;
import com.google.api.gax.httpjson.ProtoMessageResponseParser;
import com.google.api.gax.httpjson.ProtoRestSerializer;
import com.google.api.gax.httpjson.longrunning.stub.HttpJsonOperationsStub;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.OperationCallable;
import com.google.api.gax.rpc.RequestParamsBuilder;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.cloud.discoveryengine.v1.CreateDataStoreMetadata;
import com.google.cloud.discoveryengine.v1.CreateDataStoreRequest;
import com.google.cloud.discoveryengine.v1.DataStore;
import com.google.cloud.discoveryengine.v1.DeleteDataStoreMetadata;
import com.google.cloud.discoveryengine.v1.DeleteDataStoreRequest;
import com.google.cloud.discoveryengine.v1.GetDataStoreRequest;
import com.google.cloud.discoveryengine.v1.ListDataStoresRequest;
import com.google.cloud.discoveryengine.v1.ListDataStoresResponse;
import com.google.cloud.discoveryengine.v1.UpdateDataStoreRequest;
import com.google.common.collect.ImmutableMap;
import com.google.longrunning.Operation;
import com.google.protobuf.Empty;
import com.google.protobuf.TypeRegistry;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import javax.annotation.Generated;

// AUTO-GENERATED DOCUMENTATION AND CLASS.
/**
 * REST stub implementation for the DataStoreService service API.
 *
 * 

This class is for advanced usage and reflects the underlying API directly. */ @Generated("by gapic-generator-java") public class HttpJsonDataStoreServiceStub extends DataStoreServiceStub { private static final TypeRegistry typeRegistry = TypeRegistry.newBuilder() .add(DeleteDataStoreMetadata.getDescriptor()) .add(Empty.getDescriptor()) .add(DataStore.getDescriptor()) .add(CreateDataStoreMetadata.getDescriptor()) .build(); private static final ApiMethodDescriptor createDataStoreMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1.DataStoreService/CreateDataStore") .setHttpMethod("POST") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1/{parent=projects/*/locations/*}/dataStores", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "parent", request.getParent()); return fields; }) .setAdditionalPaths( "/v1/{parent=projects/*/locations/*/collections/*}/dataStores") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam( fields, "createAdvancedSiteSearch", request.getCreateAdvancedSiteSearch()); serializer.putQueryParam( fields, "dataStoreId", request.getDataStoreId()); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor( request -> ProtoRestSerializer.create() .toBody("dataStore", request.getDataStore(), true)) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(Operation.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .setOperationSnapshotFactory( (CreateDataStoreRequest request, Operation response) -> HttpJsonOperationSnapshot.create(response)) .build(); private static final ApiMethodDescriptor getDataStoreMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1.DataStoreService/GetDataStore") .setHttpMethod("GET") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1/{name=projects/*/locations/*/dataStores/*}", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "name", request.getName()); return fields; }) .setAdditionalPaths( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*}") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor(request -> null) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(DataStore.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .build(); private static final ApiMethodDescriptor listDataStoresMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1.DataStoreService/ListDataStores") .setHttpMethod("GET") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1/{parent=projects/*/locations/*}/dataStores", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "parent", request.getParent()); return fields; }) .setAdditionalPaths( "/v1/{parent=projects/*/locations/*/collections/*}/dataStores") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "filter", request.getFilter()); serializer.putQueryParam(fields, "pageSize", request.getPageSize()); serializer.putQueryParam(fields, "pageToken", request.getPageToken()); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor(request -> null) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(ListDataStoresResponse.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .build(); private static final ApiMethodDescriptor deleteDataStoreMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1.DataStoreService/DeleteDataStore") .setHttpMethod("DELETE") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1/{name=projects/*/locations/*/dataStores/*}", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam(fields, "name", request.getName()); return fields; }) .setAdditionalPaths( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*}") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor(request -> null) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(Operation.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .setOperationSnapshotFactory( (DeleteDataStoreRequest request, Operation response) -> HttpJsonOperationSnapshot.create(response)) .build(); private static final ApiMethodDescriptor updateDataStoreMethodDescriptor = ApiMethodDescriptor.newBuilder() .setFullMethodName("google.cloud.discoveryengine.v1.DataStoreService/UpdateDataStore") .setHttpMethod("PATCH") .setType(ApiMethodDescriptor.MethodType.UNARY) .setRequestFormatter( ProtoMessageRequestFormatter.newBuilder() .setPath( "/v1/{dataStore.name=projects/*/locations/*/dataStores/*}", request -> { Map fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putPathParam( fields, "dataStore.name", request.getDataStore().getName()); return fields; }) .setAdditionalPaths( "/v1/{dataStore.name=projects/*/locations/*/collections/*/dataStores/*}") .setQueryParamsExtractor( request -> { Map> fields = new HashMap<>(); ProtoRestSerializer serializer = ProtoRestSerializer.create(); serializer.putQueryParam(fields, "updateMask", request.getUpdateMask()); serializer.putQueryParam(fields, "$alt", "json;enum-encoding=int"); return fields; }) .setRequestBodyExtractor( request -> ProtoRestSerializer.create() .toBody("dataStore", request.getDataStore(), true)) .build()) .setResponseParser( ProtoMessageResponseParser.newBuilder() .setDefaultInstance(DataStore.getDefaultInstance()) .setDefaultTypeRegistry(typeRegistry) .build()) .build(); private final UnaryCallable createDataStoreCallable; private final OperationCallable createDataStoreOperationCallable; private final UnaryCallable getDataStoreCallable; private final UnaryCallable listDataStoresCallable; private final UnaryCallable listDataStoresPagedCallable; private final UnaryCallable deleteDataStoreCallable; private final OperationCallable deleteDataStoreOperationCallable; private final UnaryCallable updateDataStoreCallable; private final BackgroundResource backgroundResources; private final HttpJsonOperationsStub httpJsonOperationsStub; private final HttpJsonStubCallableFactory callableFactory; public static final HttpJsonDataStoreServiceStub create(DataStoreServiceStubSettings settings) throws IOException { return new HttpJsonDataStoreServiceStub(settings, ClientContext.create(settings)); } public static final HttpJsonDataStoreServiceStub create(ClientContext clientContext) throws IOException { return new HttpJsonDataStoreServiceStub( DataStoreServiceStubSettings.newHttpJsonBuilder().build(), clientContext); } public static final HttpJsonDataStoreServiceStub create( ClientContext clientContext, HttpJsonStubCallableFactory callableFactory) throws IOException { return new HttpJsonDataStoreServiceStub( DataStoreServiceStubSettings.newHttpJsonBuilder().build(), clientContext, callableFactory); } /** * Constructs an instance of HttpJsonDataStoreServiceStub, using the given settings. This is * protected so that it is easy to make a subclass, but otherwise, the static factory methods * should be preferred. */ protected HttpJsonDataStoreServiceStub( DataStoreServiceStubSettings settings, ClientContext clientContext) throws IOException { this(settings, clientContext, new HttpJsonDataStoreServiceCallableFactory()); } /** * Constructs an instance of HttpJsonDataStoreServiceStub, using the given settings. This is * protected so that it is easy to make a subclass, but otherwise, the static factory methods * should be preferred. */ protected HttpJsonDataStoreServiceStub( DataStoreServiceStubSettings settings, ClientContext clientContext, HttpJsonStubCallableFactory callableFactory) throws IOException { this.callableFactory = callableFactory; this.httpJsonOperationsStub = HttpJsonOperationsStub.create( clientContext, callableFactory, typeRegistry, ImmutableMap.builder() .put( "google.longrunning.Operations.CancelOperation", HttpRule.newBuilder() .setPost("/v1/{name=projects/*/operations/*}:cancel") .addAdditionalBindings( HttpRule.newBuilder() .setPost( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/branches/*/operations/*}:cancel") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setPost( "/v1/{name=projects/*/locations/*/dataStores/*/branches/*/operations/*}:cancel") .build()) .build()) .put( "google.longrunning.Operations.GetOperation", HttpRule.newBuilder() .setGet("/v1/{name=projects/*/operations/*}") .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataConnector/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/branches/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/models/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/schemas/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/siteSearchEngine/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/siteSearchEngine/targetSites/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/engines/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/dataStores/*/branches/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/dataStores/*/models/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/dataStores/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1/{name=projects/*/locations/*/operations/*}") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1/{name=projects/*/operations/*}") .build()) .build()) .put( "google.longrunning.Operations.ListOperations", HttpRule.newBuilder() .setGet("/v1/{name=projects/*}/operations") .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataConnector}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/branches/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/models/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/schemas/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/siteSearchEngine/targetSites}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*/siteSearchEngine}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/dataStores/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*/engines/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/collections/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/dataStores/*/branches/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet( "/v1/{name=projects/*/locations/*/dataStores/*/models/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1/{name=projects/*/locations/*/dataStores/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1/{name=projects/*/locations/*}/operations") .build()) .addAdditionalBindings( HttpRule.newBuilder() .setGet("/v1/{name=projects/*}/operations") .build()) .build()) .build()); HttpJsonCallSettings createDataStoreTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(createDataStoreMethodDescriptor) .setTypeRegistry(typeRegistry) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); HttpJsonCallSettings getDataStoreTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(getDataStoreMethodDescriptor) .setTypeRegistry(typeRegistry) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); HttpJsonCallSettings listDataStoresTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(listDataStoresMethodDescriptor) .setTypeRegistry(typeRegistry) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); HttpJsonCallSettings deleteDataStoreTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(deleteDataStoreMethodDescriptor) .setTypeRegistry(typeRegistry) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); HttpJsonCallSettings updateDataStoreTransportSettings = HttpJsonCallSettings.newBuilder() .setMethodDescriptor(updateDataStoreMethodDescriptor) .setTypeRegistry(typeRegistry) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("data_store.name", String.valueOf(request.getDataStore().getName())); return builder.build(); }) .build(); this.createDataStoreCallable = callableFactory.createUnaryCallable( createDataStoreTransportSettings, settings.createDataStoreSettings(), clientContext); this.createDataStoreOperationCallable = callableFactory.createOperationCallable( createDataStoreTransportSettings, settings.createDataStoreOperationSettings(), clientContext, httpJsonOperationsStub); this.getDataStoreCallable = callableFactory.createUnaryCallable( getDataStoreTransportSettings, settings.getDataStoreSettings(), clientContext); this.listDataStoresCallable = callableFactory.createUnaryCallable( listDataStoresTransportSettings, settings.listDataStoresSettings(), clientContext); this.listDataStoresPagedCallable = callableFactory.createPagedCallable( listDataStoresTransportSettings, settings.listDataStoresSettings(), clientContext); this.deleteDataStoreCallable = callableFactory.createUnaryCallable( deleteDataStoreTransportSettings, settings.deleteDataStoreSettings(), clientContext); this.deleteDataStoreOperationCallable = callableFactory.createOperationCallable( deleteDataStoreTransportSettings, settings.deleteDataStoreOperationSettings(), clientContext, httpJsonOperationsStub); this.updateDataStoreCallable = callableFactory.createUnaryCallable( updateDataStoreTransportSettings, settings.updateDataStoreSettings(), clientContext); this.backgroundResources = new BackgroundResourceAggregation(clientContext.getBackgroundResources()); } @InternalApi public static List getMethodDescriptors() { List methodDescriptors = new ArrayList<>(); methodDescriptors.add(createDataStoreMethodDescriptor); methodDescriptors.add(getDataStoreMethodDescriptor); methodDescriptors.add(listDataStoresMethodDescriptor); methodDescriptors.add(deleteDataStoreMethodDescriptor); methodDescriptors.add(updateDataStoreMethodDescriptor); return methodDescriptors; } public HttpJsonOperationsStub getHttpJsonOperationsStub() { return httpJsonOperationsStub; } @Override public UnaryCallable createDataStoreCallable() { return createDataStoreCallable; } @Override public OperationCallable createDataStoreOperationCallable() { return createDataStoreOperationCallable; } @Override public UnaryCallable getDataStoreCallable() { return getDataStoreCallable; } @Override public UnaryCallable listDataStoresCallable() { return listDataStoresCallable; } @Override public UnaryCallable listDataStoresPagedCallable() { return listDataStoresPagedCallable; } @Override public UnaryCallable deleteDataStoreCallable() { return deleteDataStoreCallable; } @Override public OperationCallable deleteDataStoreOperationCallable() { return deleteDataStoreOperationCallable; } @Override public UnaryCallable updateDataStoreCallable() { return updateDataStoreCallable; } @Override public final void close() { try { backgroundResources.close(); } catch (RuntimeException e) { throw e; } catch (Exception e) { throw new IllegalStateException("Failed to close resource", e); } } @Override public void shutdown() { backgroundResources.shutdown(); } @Override public boolean isShutdown() { return backgroundResources.isShutdown(); } @Override public boolean isTerminated() { return backgroundResources.isTerminated(); } @Override public void shutdownNow() { backgroundResources.shutdownNow(); } @Override public boolean awaitTermination(long duration, TimeUnit unit) throws InterruptedException { return backgroundResources.awaitTermination(duration, unit); } }





© 2015 - 2025 Weber Informatics LLC | Privacy Policy