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

com.google.cloud.aiplatform.v1beta1.stub.GrpcIndexEndpointServiceStub Maven / Gradle / Ivy

There is a newer version: 3.55.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.aiplatform.v1beta1.stub;

import static com.google.cloud.aiplatform.v1beta1.IndexEndpointServiceClient.ListIndexEndpointsPagedResponse;
import static com.google.cloud.aiplatform.v1beta1.IndexEndpointServiceClient.ListLocationsPagedResponse;

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.aiplatform.v1beta1.CreateIndexEndpointOperationMetadata;
import com.google.cloud.aiplatform.v1beta1.CreateIndexEndpointRequest;
import com.google.cloud.aiplatform.v1beta1.DeleteIndexEndpointRequest;
import com.google.cloud.aiplatform.v1beta1.DeleteOperationMetadata;
import com.google.cloud.aiplatform.v1beta1.DeployIndexOperationMetadata;
import com.google.cloud.aiplatform.v1beta1.DeployIndexRequest;
import com.google.cloud.aiplatform.v1beta1.DeployIndexResponse;
import com.google.cloud.aiplatform.v1beta1.GetIndexEndpointRequest;
import com.google.cloud.aiplatform.v1beta1.IndexEndpoint;
import com.google.cloud.aiplatform.v1beta1.ListIndexEndpointsRequest;
import com.google.cloud.aiplatform.v1beta1.ListIndexEndpointsResponse;
import com.google.cloud.aiplatform.v1beta1.MutateDeployedIndexOperationMetadata;
import com.google.cloud.aiplatform.v1beta1.MutateDeployedIndexRequest;
import com.google.cloud.aiplatform.v1beta1.MutateDeployedIndexResponse;
import com.google.cloud.aiplatform.v1beta1.UndeployIndexOperationMetadata;
import com.google.cloud.aiplatform.v1beta1.UndeployIndexRequest;
import com.google.cloud.aiplatform.v1beta1.UndeployIndexResponse;
import com.google.cloud.aiplatform.v1beta1.UpdateIndexEndpointRequest;
import com.google.cloud.location.GetLocationRequest;
import com.google.cloud.location.ListLocationsRequest;
import com.google.cloud.location.ListLocationsResponse;
import com.google.cloud.location.Location;
import com.google.iam.v1.GetIamPolicyRequest;
import com.google.iam.v1.Policy;
import com.google.iam.v1.SetIamPolicyRequest;
import com.google.iam.v1.TestIamPermissionsRequest;
import com.google.iam.v1.TestIamPermissionsResponse;
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 IndexEndpointService service API.
 *
 * 

This class is for advanced usage and reflects the underlying API directly. */ @BetaApi @Generated("by gapic-generator-java") public class GrpcIndexEndpointServiceStub extends IndexEndpointServiceStub { private static final MethodDescriptor createIndexEndpointMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/CreateIndexEndpoint") .setRequestMarshaller( ProtoUtils.marshaller(CreateIndexEndpointRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor getIndexEndpointMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/GetIndexEndpoint") .setRequestMarshaller( ProtoUtils.marshaller(GetIndexEndpointRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(IndexEndpoint.getDefaultInstance())) .build(); private static final MethodDescriptor listIndexEndpointsMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/ListIndexEndpoints") .setRequestMarshaller( ProtoUtils.marshaller(ListIndexEndpointsRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(ListIndexEndpointsResponse.getDefaultInstance())) .build(); private static final MethodDescriptor updateIndexEndpointMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/UpdateIndexEndpoint") .setRequestMarshaller( ProtoUtils.marshaller(UpdateIndexEndpointRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(IndexEndpoint.getDefaultInstance())) .build(); private static final MethodDescriptor deleteIndexEndpointMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/DeleteIndexEndpoint") .setRequestMarshaller( ProtoUtils.marshaller(DeleteIndexEndpointRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor deployIndexMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1beta1.IndexEndpointService/DeployIndex") .setRequestMarshaller(ProtoUtils.marshaller(DeployIndexRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor undeployIndexMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/UndeployIndex") .setRequestMarshaller( ProtoUtils.marshaller(UndeployIndexRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor mutateDeployedIndexMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1beta1.IndexEndpointService/MutateDeployedIndex") .setRequestMarshaller( ProtoUtils.marshaller(MutateDeployedIndexRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor listLocationsMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.location.Locations/ListLocations") .setRequestMarshaller( ProtoUtils.marshaller(ListLocationsRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(ListLocationsResponse.getDefaultInstance())) .build(); private static final MethodDescriptor getLocationMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.location.Locations/GetLocation") .setRequestMarshaller(ProtoUtils.marshaller(GetLocationRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Location.getDefaultInstance())) .build(); private static final MethodDescriptor setIamPolicyMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.iam.v1.IAMPolicy/SetIamPolicy") .setRequestMarshaller(ProtoUtils.marshaller(SetIamPolicyRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Policy.getDefaultInstance())) .build(); private static final MethodDescriptor getIamPolicyMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.iam.v1.IAMPolicy/GetIamPolicy") .setRequestMarshaller(ProtoUtils.marshaller(GetIamPolicyRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Policy.getDefaultInstance())) .build(); private static final MethodDescriptor testIamPermissionsMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.iam.v1.IAMPolicy/TestIamPermissions") .setRequestMarshaller( ProtoUtils.marshaller(TestIamPermissionsRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(TestIamPermissionsResponse.getDefaultInstance())) .build(); private final UnaryCallable createIndexEndpointCallable; private final OperationCallable< CreateIndexEndpointRequest, IndexEndpoint, CreateIndexEndpointOperationMetadata> createIndexEndpointOperationCallable; private final UnaryCallable getIndexEndpointCallable; private final UnaryCallable listIndexEndpointsCallable; private final UnaryCallable listIndexEndpointsPagedCallable; private final UnaryCallable updateIndexEndpointCallable; private final UnaryCallable deleteIndexEndpointCallable; private final OperationCallable deleteIndexEndpointOperationCallable; private final UnaryCallable deployIndexCallable; private final OperationCallable< DeployIndexRequest, DeployIndexResponse, DeployIndexOperationMetadata> deployIndexOperationCallable; private final UnaryCallable undeployIndexCallable; private final OperationCallable< UndeployIndexRequest, UndeployIndexResponse, UndeployIndexOperationMetadata> undeployIndexOperationCallable; private final UnaryCallable mutateDeployedIndexCallable; private final OperationCallable< MutateDeployedIndexRequest, MutateDeployedIndexResponse, MutateDeployedIndexOperationMetadata> mutateDeployedIndexOperationCallable; private final UnaryCallable listLocationsCallable; private final UnaryCallable listLocationsPagedCallable; private final UnaryCallable getLocationCallable; private final UnaryCallable setIamPolicyCallable; private final UnaryCallable getIamPolicyCallable; private final UnaryCallable testIamPermissionsCallable; private final BackgroundResource backgroundResources; private final GrpcOperationsStub operationsStub; private final GrpcStubCallableFactory callableFactory; public static final GrpcIndexEndpointServiceStub create(IndexEndpointServiceStubSettings settings) throws IOException { return new GrpcIndexEndpointServiceStub(settings, ClientContext.create(settings)); } public static final GrpcIndexEndpointServiceStub create(ClientContext clientContext) throws IOException { return new GrpcIndexEndpointServiceStub( IndexEndpointServiceStubSettings.newBuilder().build(), clientContext); } public static final GrpcIndexEndpointServiceStub create( ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException { return new GrpcIndexEndpointServiceStub( IndexEndpointServiceStubSettings.newBuilder().build(), clientContext, callableFactory); } /** * Constructs an instance of GrpcIndexEndpointServiceStub, 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 GrpcIndexEndpointServiceStub( IndexEndpointServiceStubSettings settings, ClientContext clientContext) throws IOException { this(settings, clientContext, new GrpcIndexEndpointServiceCallableFactory()); } /** * Constructs an instance of GrpcIndexEndpointServiceStub, 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 GrpcIndexEndpointServiceStub( IndexEndpointServiceStubSettings settings, ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException { this.callableFactory = callableFactory; this.operationsStub = GrpcOperationsStub.create(clientContext, callableFactory); GrpcCallSettings createIndexEndpointTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(createIndexEndpointMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings getIndexEndpointTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(getIndexEndpointMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings listIndexEndpointsTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(listIndexEndpointsMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings updateIndexEndpointTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(updateIndexEndpointMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add( "index_endpoint.name", String.valueOf(request.getIndexEndpoint().getName())); return builder.build(); }) .build(); GrpcCallSettings deleteIndexEndpointTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(deleteIndexEndpointMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings deployIndexTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(deployIndexMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("index_endpoint", String.valueOf(request.getIndexEndpoint())); return builder.build(); }) .build(); GrpcCallSettings undeployIndexTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(undeployIndexMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("index_endpoint", String.valueOf(request.getIndexEndpoint())); return builder.build(); }) .build(); GrpcCallSettings mutateDeployedIndexTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(mutateDeployedIndexMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("index_endpoint", String.valueOf(request.getIndexEndpoint())); return builder.build(); }) .build(); GrpcCallSettings listLocationsTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(listLocationsMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings getLocationTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(getLocationMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings setIamPolicyTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(setIamPolicyMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("resource", String.valueOf(request.getResource())); return builder.build(); }) .build(); GrpcCallSettings getIamPolicyTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(getIamPolicyMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("resource", String.valueOf(request.getResource())); return builder.build(); }) .build(); GrpcCallSettings testIamPermissionsTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(testIamPermissionsMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("resource", String.valueOf(request.getResource())); return builder.build(); }) .build(); this.createIndexEndpointCallable = callableFactory.createUnaryCallable( createIndexEndpointTransportSettings, settings.createIndexEndpointSettings(), clientContext); this.createIndexEndpointOperationCallable = callableFactory.createOperationCallable( createIndexEndpointTransportSettings, settings.createIndexEndpointOperationSettings(), clientContext, operationsStub); this.getIndexEndpointCallable = callableFactory.createUnaryCallable( getIndexEndpointTransportSettings, settings.getIndexEndpointSettings(), clientContext); this.listIndexEndpointsCallable = callableFactory.createUnaryCallable( listIndexEndpointsTransportSettings, settings.listIndexEndpointsSettings(), clientContext); this.listIndexEndpointsPagedCallable = callableFactory.createPagedCallable( listIndexEndpointsTransportSettings, settings.listIndexEndpointsSettings(), clientContext); this.updateIndexEndpointCallable = callableFactory.createUnaryCallable( updateIndexEndpointTransportSettings, settings.updateIndexEndpointSettings(), clientContext); this.deleteIndexEndpointCallable = callableFactory.createUnaryCallable( deleteIndexEndpointTransportSettings, settings.deleteIndexEndpointSettings(), clientContext); this.deleteIndexEndpointOperationCallable = callableFactory.createOperationCallable( deleteIndexEndpointTransportSettings, settings.deleteIndexEndpointOperationSettings(), clientContext, operationsStub); this.deployIndexCallable = callableFactory.createUnaryCallable( deployIndexTransportSettings, settings.deployIndexSettings(), clientContext); this.deployIndexOperationCallable = callableFactory.createOperationCallable( deployIndexTransportSettings, settings.deployIndexOperationSettings(), clientContext, operationsStub); this.undeployIndexCallable = callableFactory.createUnaryCallable( undeployIndexTransportSettings, settings.undeployIndexSettings(), clientContext); this.undeployIndexOperationCallable = callableFactory.createOperationCallable( undeployIndexTransportSettings, settings.undeployIndexOperationSettings(), clientContext, operationsStub); this.mutateDeployedIndexCallable = callableFactory.createUnaryCallable( mutateDeployedIndexTransportSettings, settings.mutateDeployedIndexSettings(), clientContext); this.mutateDeployedIndexOperationCallable = callableFactory.createOperationCallable( mutateDeployedIndexTransportSettings, settings.mutateDeployedIndexOperationSettings(), clientContext, operationsStub); this.listLocationsCallable = callableFactory.createUnaryCallable( listLocationsTransportSettings, settings.listLocationsSettings(), clientContext); this.listLocationsPagedCallable = callableFactory.createPagedCallable( listLocationsTransportSettings, settings.listLocationsSettings(), clientContext); this.getLocationCallable = callableFactory.createUnaryCallable( getLocationTransportSettings, settings.getLocationSettings(), clientContext); this.setIamPolicyCallable = callableFactory.createUnaryCallable( setIamPolicyTransportSettings, settings.setIamPolicySettings(), clientContext); this.getIamPolicyCallable = callableFactory.createUnaryCallable( getIamPolicyTransportSettings, settings.getIamPolicySettings(), clientContext); this.testIamPermissionsCallable = callableFactory.createUnaryCallable( testIamPermissionsTransportSettings, settings.testIamPermissionsSettings(), clientContext); this.backgroundResources = new BackgroundResourceAggregation(clientContext.getBackgroundResources()); } public GrpcOperationsStub getOperationsStub() { return operationsStub; } @Override public UnaryCallable createIndexEndpointCallable() { return createIndexEndpointCallable; } @Override public OperationCallable< CreateIndexEndpointRequest, IndexEndpoint, CreateIndexEndpointOperationMetadata> createIndexEndpointOperationCallable() { return createIndexEndpointOperationCallable; } @Override public UnaryCallable getIndexEndpointCallable() { return getIndexEndpointCallable; } @Override public UnaryCallable listIndexEndpointsCallable() { return listIndexEndpointsCallable; } @Override public UnaryCallable listIndexEndpointsPagedCallable() { return listIndexEndpointsPagedCallable; } @Override public UnaryCallable updateIndexEndpointCallable() { return updateIndexEndpointCallable; } @Override public UnaryCallable deleteIndexEndpointCallable() { return deleteIndexEndpointCallable; } @Override public OperationCallable deleteIndexEndpointOperationCallable() { return deleteIndexEndpointOperationCallable; } @Override public UnaryCallable deployIndexCallable() { return deployIndexCallable; } @Override public OperationCallable deployIndexOperationCallable() { return deployIndexOperationCallable; } @Override public UnaryCallable undeployIndexCallable() { return undeployIndexCallable; } @Override public OperationCallable< UndeployIndexRequest, UndeployIndexResponse, UndeployIndexOperationMetadata> undeployIndexOperationCallable() { return undeployIndexOperationCallable; } @Override public UnaryCallable mutateDeployedIndexCallable() { return mutateDeployedIndexCallable; } @Override public OperationCallable< MutateDeployedIndexRequest, MutateDeployedIndexResponse, MutateDeployedIndexOperationMetadata> mutateDeployedIndexOperationCallable() { return mutateDeployedIndexOperationCallable; } @Override public UnaryCallable listLocationsCallable() { return listLocationsCallable; } @Override public UnaryCallable listLocationsPagedCallable() { return listLocationsPagedCallable; } @Override public UnaryCallable getLocationCallable() { return getLocationCallable; } @Override public UnaryCallable setIamPolicyCallable() { return setIamPolicyCallable; } @Override public UnaryCallable getIamPolicyCallable() { return getIamPolicyCallable; } @Override public UnaryCallable testIamPermissionsCallable() { return testIamPermissionsCallable; } @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 - 2024 Weber Informatics LLC | Privacy Policy