com.google.cloud.bigquery.migration.v2alpha.stub.GrpcMigrationServiceStub Maven / Gradle / Ivy
Show all versions of google-cloud-bigquerymigration Show documentation
/*
* 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.bigquery.migration.v2alpha.stub;
import static com.google.cloud.bigquery.migration.v2alpha.MigrationServiceClient.ListMigrationSubtasksPagedResponse;
import static com.google.cloud.bigquery.migration.v2alpha.MigrationServiceClient.ListMigrationWorkflowsPagedResponse;
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.RequestParamsBuilder;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.cloud.bigquery.migration.v2alpha.CreateMigrationWorkflowRequest;
import com.google.cloud.bigquery.migration.v2alpha.DeleteMigrationWorkflowRequest;
import com.google.cloud.bigquery.migration.v2alpha.GetMigrationSubtaskRequest;
import com.google.cloud.bigquery.migration.v2alpha.GetMigrationWorkflowRequest;
import com.google.cloud.bigquery.migration.v2alpha.ListMigrationSubtasksRequest;
import com.google.cloud.bigquery.migration.v2alpha.ListMigrationSubtasksResponse;
import com.google.cloud.bigquery.migration.v2alpha.ListMigrationWorkflowsRequest;
import com.google.cloud.bigquery.migration.v2alpha.ListMigrationWorkflowsResponse;
import com.google.cloud.bigquery.migration.v2alpha.MigrationSubtask;
import com.google.cloud.bigquery.migration.v2alpha.MigrationWorkflow;
import com.google.cloud.bigquery.migration.v2alpha.StartMigrationWorkflowRequest;
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 MigrationService service API.
*
* This class is for advanced usage and reflects the underlying API directly.
*/
@BetaApi
@Generated("by gapic-generator-java")
public class GrpcMigrationServiceStub extends MigrationServiceStub {
private static final MethodDescriptor
createMigrationWorkflowMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/CreateMigrationWorkflow")
.setRequestMarshaller(
ProtoUtils.marshaller(CreateMigrationWorkflowRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(MigrationWorkflow.getDefaultInstance()))
.build();
private static final MethodDescriptor
getMigrationWorkflowMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/GetMigrationWorkflow")
.setRequestMarshaller(
ProtoUtils.marshaller(GetMigrationWorkflowRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(MigrationWorkflow.getDefaultInstance()))
.build();
private static final MethodDescriptor<
ListMigrationWorkflowsRequest, ListMigrationWorkflowsResponse>
listMigrationWorkflowsMethodDescriptor =
MethodDescriptor
.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/ListMigrationWorkflows")
.setRequestMarshaller(
ProtoUtils.marshaller(ListMigrationWorkflowsRequest.getDefaultInstance()))
.setResponseMarshaller(
ProtoUtils.marshaller(ListMigrationWorkflowsResponse.getDefaultInstance()))
.build();
private static final MethodDescriptor
deleteMigrationWorkflowMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/DeleteMigrationWorkflow")
.setRequestMarshaller(
ProtoUtils.marshaller(DeleteMigrationWorkflowRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Empty.getDefaultInstance()))
.build();
private static final MethodDescriptor
startMigrationWorkflowMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/StartMigrationWorkflow")
.setRequestMarshaller(
ProtoUtils.marshaller(StartMigrationWorkflowRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Empty.getDefaultInstance()))
.build();
private static final MethodDescriptor
getMigrationSubtaskMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/GetMigrationSubtask")
.setRequestMarshaller(
ProtoUtils.marshaller(GetMigrationSubtaskRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(MigrationSubtask.getDefaultInstance()))
.build();
private static final MethodDescriptor
listMigrationSubtasksMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName(
"google.cloud.bigquery.migration.v2alpha.MigrationService/ListMigrationSubtasks")
.setRequestMarshaller(
ProtoUtils.marshaller(ListMigrationSubtasksRequest.getDefaultInstance()))
.setResponseMarshaller(
ProtoUtils.marshaller(ListMigrationSubtasksResponse.getDefaultInstance()))
.build();
private final UnaryCallable
createMigrationWorkflowCallable;
private final UnaryCallable
getMigrationWorkflowCallable;
private final UnaryCallable
listMigrationWorkflowsCallable;
private final UnaryCallable
listMigrationWorkflowsPagedCallable;
private final UnaryCallable
deleteMigrationWorkflowCallable;
private final UnaryCallable startMigrationWorkflowCallable;
private final UnaryCallable
getMigrationSubtaskCallable;
private final UnaryCallable
listMigrationSubtasksCallable;
private final UnaryCallable
listMigrationSubtasksPagedCallable;
private final BackgroundResource backgroundResources;
private final GrpcOperationsStub operationsStub;
private final GrpcStubCallableFactory callableFactory;
public static final GrpcMigrationServiceStub create(MigrationServiceStubSettings settings)
throws IOException {
return new GrpcMigrationServiceStub(settings, ClientContext.create(settings));
}
public static final GrpcMigrationServiceStub create(ClientContext clientContext)
throws IOException {
return new GrpcMigrationServiceStub(
MigrationServiceStubSettings.newBuilder().build(), clientContext);
}
public static final GrpcMigrationServiceStub create(
ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException {
return new GrpcMigrationServiceStub(
MigrationServiceStubSettings.newBuilder().build(), clientContext, callableFactory);
}
/**
* Constructs an instance of GrpcMigrationServiceStub, 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 GrpcMigrationServiceStub(
MigrationServiceStubSettings settings, ClientContext clientContext) throws IOException {
this(settings, clientContext, new GrpcMigrationServiceCallableFactory());
}
/**
* Constructs an instance of GrpcMigrationServiceStub, 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 GrpcMigrationServiceStub(
MigrationServiceStubSettings settings,
ClientContext clientContext,
GrpcStubCallableFactory callableFactory)
throws IOException {
this.callableFactory = callableFactory;
this.operationsStub = GrpcOperationsStub.create(clientContext, callableFactory);
GrpcCallSettings
createMigrationWorkflowTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(createMigrationWorkflowMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("parent", String.valueOf(request.getParent()));
return builder.build();
})
.build();
GrpcCallSettings
getMigrationWorkflowTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(getMigrationWorkflowMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("name", String.valueOf(request.getName()));
return builder.build();
})
.build();
GrpcCallSettings
listMigrationWorkflowsTransportSettings =
GrpcCallSettings
.newBuilder()
.setMethodDescriptor(listMigrationWorkflowsMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("parent", String.valueOf(request.getParent()));
return builder.build();
})
.build();
GrpcCallSettings
deleteMigrationWorkflowTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(deleteMigrationWorkflowMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("name", String.valueOf(request.getName()));
return builder.build();
})
.build();
GrpcCallSettings startMigrationWorkflowTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(startMigrationWorkflowMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("name", String.valueOf(request.getName()));
return builder.build();
})
.build();
GrpcCallSettings
getMigrationSubtaskTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(getMigrationSubtaskMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("name", String.valueOf(request.getName()));
return builder.build();
})
.build();
GrpcCallSettings
listMigrationSubtasksTransportSettings =
GrpcCallSettings
.newBuilder()
.setMethodDescriptor(listMigrationSubtasksMethodDescriptor)
.setParamsExtractor(
request -> {
RequestParamsBuilder builder = RequestParamsBuilder.create();
builder.add("parent", String.valueOf(request.getParent()));
return builder.build();
})
.build();
this.createMigrationWorkflowCallable =
callableFactory.createUnaryCallable(
createMigrationWorkflowTransportSettings,
settings.createMigrationWorkflowSettings(),
clientContext);
this.getMigrationWorkflowCallable =
callableFactory.createUnaryCallable(
getMigrationWorkflowTransportSettings,
settings.getMigrationWorkflowSettings(),
clientContext);
this.listMigrationWorkflowsCallable =
callableFactory.createUnaryCallable(
listMigrationWorkflowsTransportSettings,
settings.listMigrationWorkflowsSettings(),
clientContext);
this.listMigrationWorkflowsPagedCallable =
callableFactory.createPagedCallable(
listMigrationWorkflowsTransportSettings,
settings.listMigrationWorkflowsSettings(),
clientContext);
this.deleteMigrationWorkflowCallable =
callableFactory.createUnaryCallable(
deleteMigrationWorkflowTransportSettings,
settings.deleteMigrationWorkflowSettings(),
clientContext);
this.startMigrationWorkflowCallable =
callableFactory.createUnaryCallable(
startMigrationWorkflowTransportSettings,
settings.startMigrationWorkflowSettings(),
clientContext);
this.getMigrationSubtaskCallable =
callableFactory.createUnaryCallable(
getMigrationSubtaskTransportSettings,
settings.getMigrationSubtaskSettings(),
clientContext);
this.listMigrationSubtasksCallable =
callableFactory.createUnaryCallable(
listMigrationSubtasksTransportSettings,
settings.listMigrationSubtasksSettings(),
clientContext);
this.listMigrationSubtasksPagedCallable =
callableFactory.createPagedCallable(
listMigrationSubtasksTransportSettings,
settings.listMigrationSubtasksSettings(),
clientContext);
this.backgroundResources =
new BackgroundResourceAggregation(clientContext.getBackgroundResources());
}
public GrpcOperationsStub getOperationsStub() {
return operationsStub;
}
@Override
public UnaryCallable
createMigrationWorkflowCallable() {
return createMigrationWorkflowCallable;
}
@Override
public UnaryCallable
getMigrationWorkflowCallable() {
return getMigrationWorkflowCallable;
}
@Override
public UnaryCallable
listMigrationWorkflowsCallable() {
return listMigrationWorkflowsCallable;
}
@Override
public UnaryCallable
listMigrationWorkflowsPagedCallable() {
return listMigrationWorkflowsPagedCallable;
}
@Override
public UnaryCallable deleteMigrationWorkflowCallable() {
return deleteMigrationWorkflowCallable;
}
@Override
public UnaryCallable startMigrationWorkflowCallable() {
return startMigrationWorkflowCallable;
}
@Override
public UnaryCallable getMigrationSubtaskCallable() {
return getMigrationSubtaskCallable;
}
@Override
public UnaryCallable
listMigrationSubtasksCallable() {
return listMigrationSubtasksCallable;
}
@Override
public UnaryCallable
listMigrationSubtasksPagedCallable() {
return listMigrationSubtasksPagedCallable;
}
@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);
}
}