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

com.google.cloud.aiplatform.v1.stub.GrpcPipelineServiceStub Maven / Gradle / Ivy

There is a newer version: 3.55.0
Show newest version
/*
 * Copyright 2023 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.v1.stub;

import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListLocationsPagedResponse;
import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListPipelineJobsPagedResponse;
import static com.google.cloud.aiplatform.v1.PipelineServiceClient.ListTrainingPipelinesPagedResponse;

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.v1.CancelPipelineJobRequest;
import com.google.cloud.aiplatform.v1.CancelTrainingPipelineRequest;
import com.google.cloud.aiplatform.v1.CreatePipelineJobRequest;
import com.google.cloud.aiplatform.v1.CreateTrainingPipelineRequest;
import com.google.cloud.aiplatform.v1.DeleteOperationMetadata;
import com.google.cloud.aiplatform.v1.DeletePipelineJobRequest;
import com.google.cloud.aiplatform.v1.DeleteTrainingPipelineRequest;
import com.google.cloud.aiplatform.v1.GetPipelineJobRequest;
import com.google.cloud.aiplatform.v1.GetTrainingPipelineRequest;
import com.google.cloud.aiplatform.v1.ListPipelineJobsRequest;
import com.google.cloud.aiplatform.v1.ListPipelineJobsResponse;
import com.google.cloud.aiplatform.v1.ListTrainingPipelinesRequest;
import com.google.cloud.aiplatform.v1.ListTrainingPipelinesResponse;
import com.google.cloud.aiplatform.v1.PipelineJob;
import com.google.cloud.aiplatform.v1.TrainingPipeline;
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 PipelineService service API.
 *
 * 

This class is for advanced usage and reflects the underlying API directly. */ @Generated("by gapic-generator-java") public class GrpcPipelineServiceStub extends PipelineServiceStub { private static final MethodDescriptor createTrainingPipelineMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1.PipelineService/CreateTrainingPipeline") .setRequestMarshaller( ProtoUtils.marshaller(CreateTrainingPipelineRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(TrainingPipeline.getDefaultInstance())) .build(); private static final MethodDescriptor getTrainingPipelineMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/GetTrainingPipeline") .setRequestMarshaller( ProtoUtils.marshaller(GetTrainingPipelineRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(TrainingPipeline.getDefaultInstance())) .build(); private static final MethodDescriptor listTrainingPipelinesMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/ListTrainingPipelines") .setRequestMarshaller( ProtoUtils.marshaller(ListTrainingPipelinesRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(ListTrainingPipelinesResponse.getDefaultInstance())) .build(); private static final MethodDescriptor deleteTrainingPipelineMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1.PipelineService/DeleteTrainingPipeline") .setRequestMarshaller( ProtoUtils.marshaller(DeleteTrainingPipelineRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor cancelTrainingPipelineMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName( "google.cloud.aiplatform.v1.PipelineService/CancelTrainingPipeline") .setRequestMarshaller( ProtoUtils.marshaller(CancelTrainingPipelineRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Empty.getDefaultInstance())) .build(); private static final MethodDescriptor createPipelineJobMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/CreatePipelineJob") .setRequestMarshaller( ProtoUtils.marshaller(CreatePipelineJobRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(PipelineJob.getDefaultInstance())) .build(); private static final MethodDescriptor getPipelineJobMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/GetPipelineJob") .setRequestMarshaller( ProtoUtils.marshaller(GetPipelineJobRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(PipelineJob.getDefaultInstance())) .build(); private static final MethodDescriptor listPipelineJobsMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/ListPipelineJobs") .setRequestMarshaller( ProtoUtils.marshaller(ListPipelineJobsRequest.getDefaultInstance())) .setResponseMarshaller( ProtoUtils.marshaller(ListPipelineJobsResponse.getDefaultInstance())) .build(); private static final MethodDescriptor deletePipelineJobMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/DeletePipelineJob") .setRequestMarshaller( ProtoUtils.marshaller(DeletePipelineJobRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance())) .build(); private static final MethodDescriptor cancelPipelineJobMethodDescriptor = MethodDescriptor.newBuilder() .setType(MethodDescriptor.MethodType.UNARY) .setFullMethodName("google.cloud.aiplatform.v1.PipelineService/CancelPipelineJob") .setRequestMarshaller( ProtoUtils.marshaller(CancelPipelineJobRequest.getDefaultInstance())) .setResponseMarshaller(ProtoUtils.marshaller(Empty.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 createTrainingPipelineCallable; private final UnaryCallable getTrainingPipelineCallable; private final UnaryCallable listTrainingPipelinesCallable; private final UnaryCallable listTrainingPipelinesPagedCallable; private final UnaryCallable deleteTrainingPipelineCallable; private final OperationCallable deleteTrainingPipelineOperationCallable; private final UnaryCallable cancelTrainingPipelineCallable; private final UnaryCallable createPipelineJobCallable; private final UnaryCallable getPipelineJobCallable; private final UnaryCallable listPipelineJobsCallable; private final UnaryCallable listPipelineJobsPagedCallable; private final UnaryCallable deletePipelineJobCallable; private final OperationCallable deletePipelineJobOperationCallable; private final UnaryCallable cancelPipelineJobCallable; 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 GrpcPipelineServiceStub create(PipelineServiceStubSettings settings) throws IOException { return new GrpcPipelineServiceStub(settings, ClientContext.create(settings)); } public static final GrpcPipelineServiceStub create(ClientContext clientContext) throws IOException { return new GrpcPipelineServiceStub( PipelineServiceStubSettings.newBuilder().build(), clientContext); } public static final GrpcPipelineServiceStub create( ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException { return new GrpcPipelineServiceStub( PipelineServiceStubSettings.newBuilder().build(), clientContext, callableFactory); } /** * Constructs an instance of GrpcPipelineServiceStub, 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 GrpcPipelineServiceStub( PipelineServiceStubSettings settings, ClientContext clientContext) throws IOException { this(settings, clientContext, new GrpcPipelineServiceCallableFactory()); } /** * Constructs an instance of GrpcPipelineServiceStub, 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 GrpcPipelineServiceStub( PipelineServiceStubSettings settings, ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException { this.callableFactory = callableFactory; this.operationsStub = GrpcOperationsStub.create(clientContext, callableFactory); GrpcCallSettings createTrainingPipelineTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(createTrainingPipelineMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings getTrainingPipelineTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(getTrainingPipelineMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings listTrainingPipelinesTransportSettings = GrpcCallSettings .newBuilder() .setMethodDescriptor(listTrainingPipelinesMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings deleteTrainingPipelineTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(deleteTrainingPipelineMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings cancelTrainingPipelineTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(cancelTrainingPipelineMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings createPipelineJobTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(createPipelineJobMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings getPipelineJobTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(getPipelineJobMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings listPipelineJobsTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(listPipelineJobsMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("parent", String.valueOf(request.getParent())); return builder.build(); }) .build(); GrpcCallSettings deletePipelineJobTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(deletePipelineJobMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); return builder.build(); }) .build(); GrpcCallSettings cancelPipelineJobTransportSettings = GrpcCallSettings.newBuilder() .setMethodDescriptor(cancelPipelineJobMethodDescriptor) .setParamsExtractor( request -> { RequestParamsBuilder builder = RequestParamsBuilder.create(); builder.add("name", String.valueOf(request.getName())); 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.createTrainingPipelineCallable = callableFactory.createUnaryCallable( createTrainingPipelineTransportSettings, settings.createTrainingPipelineSettings(), clientContext); this.getTrainingPipelineCallable = callableFactory.createUnaryCallable( getTrainingPipelineTransportSettings, settings.getTrainingPipelineSettings(), clientContext); this.listTrainingPipelinesCallable = callableFactory.createUnaryCallable( listTrainingPipelinesTransportSettings, settings.listTrainingPipelinesSettings(), clientContext); this.listTrainingPipelinesPagedCallable = callableFactory.createPagedCallable( listTrainingPipelinesTransportSettings, settings.listTrainingPipelinesSettings(), clientContext); this.deleteTrainingPipelineCallable = callableFactory.createUnaryCallable( deleteTrainingPipelineTransportSettings, settings.deleteTrainingPipelineSettings(), clientContext); this.deleteTrainingPipelineOperationCallable = callableFactory.createOperationCallable( deleteTrainingPipelineTransportSettings, settings.deleteTrainingPipelineOperationSettings(), clientContext, operationsStub); this.cancelTrainingPipelineCallable = callableFactory.createUnaryCallable( cancelTrainingPipelineTransportSettings, settings.cancelTrainingPipelineSettings(), clientContext); this.createPipelineJobCallable = callableFactory.createUnaryCallable( createPipelineJobTransportSettings, settings.createPipelineJobSettings(), clientContext); this.getPipelineJobCallable = callableFactory.createUnaryCallable( getPipelineJobTransportSettings, settings.getPipelineJobSettings(), clientContext); this.listPipelineJobsCallable = callableFactory.createUnaryCallable( listPipelineJobsTransportSettings, settings.listPipelineJobsSettings(), clientContext); this.listPipelineJobsPagedCallable = callableFactory.createPagedCallable( listPipelineJobsTransportSettings, settings.listPipelineJobsSettings(), clientContext); this.deletePipelineJobCallable = callableFactory.createUnaryCallable( deletePipelineJobTransportSettings, settings.deletePipelineJobSettings(), clientContext); this.deletePipelineJobOperationCallable = callableFactory.createOperationCallable( deletePipelineJobTransportSettings, settings.deletePipelineJobOperationSettings(), clientContext, operationsStub); this.cancelPipelineJobCallable = callableFactory.createUnaryCallable( cancelPipelineJobTransportSettings, settings.cancelPipelineJobSettings(), clientContext); 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 createTrainingPipelineCallable() { return createTrainingPipelineCallable; } @Override public UnaryCallable getTrainingPipelineCallable() { return getTrainingPipelineCallable; } @Override public UnaryCallable listTrainingPipelinesCallable() { return listTrainingPipelinesCallable; } @Override public UnaryCallable listTrainingPipelinesPagedCallable() { return listTrainingPipelinesPagedCallable; } @Override public UnaryCallable deleteTrainingPipelineCallable() { return deleteTrainingPipelineCallable; } @Override public OperationCallable deleteTrainingPipelineOperationCallable() { return deleteTrainingPipelineOperationCallable; } @Override public UnaryCallable cancelTrainingPipelineCallable() { return cancelTrainingPipelineCallable; } @Override public UnaryCallable createPipelineJobCallable() { return createPipelineJobCallable; } @Override public UnaryCallable getPipelineJobCallable() { return getPipelineJobCallable; } @Override public UnaryCallable listPipelineJobsCallable() { return listPipelineJobsCallable; } @Override public UnaryCallable listPipelineJobsPagedCallable() { return listPipelineJobsPagedCallable; } @Override public UnaryCallable deletePipelineJobCallable() { return deletePipelineJobCallable; } @Override public OperationCallable deletePipelineJobOperationCallable() { return deletePipelineJobOperationCallable; } @Override public UnaryCallable cancelPipelineJobCallable() { return cancelPipelineJobCallable; } @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