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

com.google.cloud.discoveryengine.v1alpha.stub.GrpcDataStoreServiceStub 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.v1alpha.stub;

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

import com.google.api.core.BetaApi;
import com.google.api.gax.core.BackgroundResource;
import com.google.api.gax.core.BackgroundResourceAggregation;
import com.google.api.gax.grpc.GrpcCallSettings;
import com.google.api.gax.grpc.GrpcStubCallableFactory;
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.v1alpha.CreateDataStoreMetadata;
import com.google.cloud.discoveryengine.v1alpha.CreateDataStoreRequest;
import com.google.cloud.discoveryengine.v1alpha.DataStore;
import com.google.cloud.discoveryengine.v1alpha.DeleteDataStoreMetadata;
import com.google.cloud.discoveryengine.v1alpha.DeleteDataStoreRequest;
import com.google.cloud.discoveryengine.v1alpha.DocumentProcessingConfig;
import com.google.cloud.discoveryengine.v1alpha.GetDataStoreRequest;
import com.google.cloud.discoveryengine.v1alpha.GetDocumentProcessingConfigRequest;
import com.google.cloud.discoveryengine.v1alpha.ListDataStoresRequest;
import com.google.cloud.discoveryengine.v1alpha.ListDataStoresResponse;
import com.google.cloud.discoveryengine.v1alpha.UpdateDataStoreRequest;
import com.google.cloud.discoveryengine.v1alpha.UpdateDocumentProcessingConfigRequest;
import com.google.longrunning.Operation;
import com.google.longrunning.stub.GrpcOperationsStub;
import com.google.protobuf.Empty;
import io.grpc.MethodDescriptor;
import io.grpc.protobuf.ProtoUtils;
import java.io.IOException;
import java.util.concurrent.TimeUnit;
import javax.annotation.Generated;

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

This class is for advanced usage and reflects the underlying API directly. */ @BetaApi @Generated("by gapic-generator-java") public class GrpcDataStoreServiceStub extends DataStoreServiceStub { private static final MethodDescriptor createDataStoreMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/CreateDataStore") .setRequestMarshaller( ProtoUtils.marshaller(CreateDataStoreRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor getDataStoreMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/GetDataStore") .setRequestMarshaller(ProtoUtils.marshaller(GetDataStoreRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(DataStore.getDefaultInstance())) .build(); private static final MethodDescriptor listDataStoresMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/ListDataStores") .setRequestMarshaller( ProtoUtils.marshaller(ListDataStoresRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(ListDataStoresResponse.getDefaultInstance())) .build(); private static final MethodDescriptor deleteDataStoreMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/DeleteDataStore") .setRequestMarshaller( ProtoUtils.marshaller(DeleteDataStoreRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor updateDataStoreMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/UpdateDataStore") .setRequestMarshaller( ProtoUtils.marshaller(UpdateDataStoreRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(DataStore.getDefaultInstance())) .build(); private static final MethodDescriptor< GetDocumentProcessingConfigRequest, DocumentProcessingConfig> getDocumentProcessingConfigMethodDescriptor = MethodDescriptor .newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/GetDocumentProcessingConfig") .setRequestMarshaller( ProtoUtils.marshaller(GetDocumentProcessingConfigRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(DocumentProcessingConfig.getDefaultInstance())) .build(); private static final MethodDescriptor< UpdateDocumentProcessingConfigRequest, DocumentProcessingConfig> updateDocumentProcessingConfigMethodDescriptor = MethodDescriptor .newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.discoveryengine.v1alpha.DataStoreService/UpdateDocumentProcessingConfig") .setRequestMarshaller( ProtoUtils.marshaller(UpdateDocumentProcessingConfigRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(DocumentProcessingConfig.getDefaultInstance())) .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 UnaryCallable getDocumentProcessingConfigCallable; private final UnaryCallable updateDocumentProcessingConfigCallable; private final BackgroundResource backgroundResources; private final GrpcOperationsStub operationsStub; private final GrpcStubCallableFactory callableFactory; public static final GrpcDataStoreServiceStub create(DataStoreServiceStubSettings settings) throws IOException { return new GrpcDataStoreServiceStub(settings, ClientContext.create(settings)); } public static final GrpcDataStoreServiceStub create(ClientContext clientContext) throws IOException { return new GrpcDataStoreServiceStub( DataStoreServiceStubSettings.newBuilder().build(), clientContext); } public static final GrpcDataStoreServiceStub create( ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException { return new GrpcDataStoreServiceStub( DataStoreServiceStubSettings.newBuilder().build(), clientContext, callableFactory); } /** * Constructs an instance of GrpcDataStoreServiceStub, 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 GrpcDataStoreServiceStub( DataStoreServiceStubSettings settings, ClientContext clientContext) throws IOException { this(settings, clientContext, new GrpcDataStoreServiceCallableFactory()); } /** * Constructs an instance of GrpcDataStoreServiceStub, 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 GrpcDataStoreServiceStub( DataStoreServiceStubSettings settings, ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException { this.callableFactory = callableFactory; this.operationsStub = GrpcOperationsStub.create(clientContext, callableFactory); GrpcCallSettings createDataStoreTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(createDataStoreMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings getDataStoreTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(getDataStoreMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings listDataStoresTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(listDataStoresMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings deleteDataStoreTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(deleteDataStoreMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings updateDataStoreTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(updateDataStoreMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("data_store.name", String.valueOf(request.getDataStore().getName())); return builder.build(); }) .build(); GrpcCallSettings getDocumentProcessingConfigTransportSettings = GrpcCallSettings .newBuilder() .setMethodDescriptor(getDocumentProcessingConfigMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings updateDocumentProcessingConfigTransportSettings = GrpcCallSettings .newBuilder() .setMethodDescriptor(updateDocumentProcessingConfigMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add( "document_processing_config.name", String.valueOf(request.getDocumentProcessingConfig().getName())); return builder.build(); }) .build(); this.createDataStoreCallable = callableFactory.createUnaryCallable( createDataStoreTransportSettings, settings.createDataStoreSettings(), clientContext); this.createDataStoreOperationCallable = callableFactory.createOperationCallable( createDataStoreTransportSettings, settings.createDataStoreOperationSettings(), clientContext, operationsStub); 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, operationsStub); this.updateDataStoreCallable = callableFactory.createUnaryCallable( updateDataStoreTransportSettings, settings.updateDataStoreSettings(), clientContext); this.getDocumentProcessingConfigCallable = callableFactory.createUnaryCallable( getDocumentProcessingConfigTransportSettings, settings.getDocumentProcessingConfigSettings(), clientContext); this.updateDocumentProcessingConfigCallable = callableFactory.createUnaryCallable( updateDocumentProcessingConfigTransportSettings, settings.updateDocumentProcessingConfigSettings(), clientContext); this.backgroundResources = new BackgroundResourceAggregation(clientContext.getBackgroundResources()); } public GrpcOperationsStub getOperationsStub() { return operationsStub; } @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 UnaryCallable getDocumentProcessingConfigCallable() { return getDocumentProcessingConfigCallable; } @Override public UnaryCallable updateDocumentProcessingConfigCallable() { return updateDocumentProcessingConfigCallable; } @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