com.google.cloud.dialogflow.v2beta1.stub.GrpcAgentsStub Maven / Gradle / Ivy
Show all versions of google-cloud-dialogflow Show documentation
/*
* Copyright 2021 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.dialogflow.v2beta1.stub;
import static com.google.cloud.dialogflow.v2beta1.AgentsClient.SearchAgentsPagedResponse;
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.UnaryCallable;
import com.google.cloud.dialogflow.v2beta1.Agent;
import com.google.cloud.dialogflow.v2beta1.DeleteAgentRequest;
import com.google.cloud.dialogflow.v2beta1.ExportAgentRequest;
import com.google.cloud.dialogflow.v2beta1.ExportAgentResponse;
import com.google.cloud.dialogflow.v2beta1.GetAgentRequest;
import com.google.cloud.dialogflow.v2beta1.GetValidationResultRequest;
import com.google.cloud.dialogflow.v2beta1.ImportAgentRequest;
import com.google.cloud.dialogflow.v2beta1.RestoreAgentRequest;
import com.google.cloud.dialogflow.v2beta1.SearchAgentsRequest;
import com.google.cloud.dialogflow.v2beta1.SearchAgentsResponse;
import com.google.cloud.dialogflow.v2beta1.SetAgentRequest;
import com.google.cloud.dialogflow.v2beta1.TrainAgentRequest;
import com.google.cloud.dialogflow.v2beta1.ValidationResult;
import com.google.common.collect.ImmutableMap;
import com.google.longrunning.Operation;
import com.google.longrunning.stub.GrpcOperationsStub;
import com.google.protobuf.Empty;
import com.google.protobuf.Struct;
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 Agents service API.
*
* This class is for advanced usage and reflects the underlying API directly.
*/
@BetaApi
@Generated("by gapic-generator-java")
public class GrpcAgentsStub extends AgentsStub {
private static final MethodDescriptor getAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/GetAgent")
.setRequestMarshaller(ProtoUtils.marshaller(GetAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Agent.getDefaultInstance()))
.build();
private static final MethodDescriptor setAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/SetAgent")
.setRequestMarshaller(ProtoUtils.marshaller(SetAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Agent.getDefaultInstance()))
.build();
private static final MethodDescriptor deleteAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/DeleteAgent")
.setRequestMarshaller(ProtoUtils.marshaller(DeleteAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Empty.getDefaultInstance()))
.build();
private static final MethodDescriptor
searchAgentsMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/SearchAgents")
.setRequestMarshaller(ProtoUtils.marshaller(SearchAgentsRequest.getDefaultInstance()))
.setResponseMarshaller(
ProtoUtils.marshaller(SearchAgentsResponse.getDefaultInstance()))
.build();
private static final MethodDescriptor trainAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/TrainAgent")
.setRequestMarshaller(ProtoUtils.marshaller(TrainAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
.build();
private static final MethodDescriptor exportAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/ExportAgent")
.setRequestMarshaller(ProtoUtils.marshaller(ExportAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
.build();
private static final MethodDescriptor importAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/ImportAgent")
.setRequestMarshaller(ProtoUtils.marshaller(ImportAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
.build();
private static final MethodDescriptor
restoreAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/RestoreAgent")
.setRequestMarshaller(ProtoUtils.marshaller(RestoreAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
.build();
private static final MethodDescriptor
getValidationResultMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2beta1.Agents/GetValidationResult")
.setRequestMarshaller(
ProtoUtils.marshaller(GetValidationResultRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(ValidationResult.getDefaultInstance()))
.build();
private final UnaryCallable getAgentCallable;
private final UnaryCallable setAgentCallable;
private final UnaryCallable deleteAgentCallable;
private final UnaryCallable searchAgentsCallable;
private final UnaryCallable
searchAgentsPagedCallable;
private final UnaryCallable trainAgentCallable;
private final OperationCallable trainAgentOperationCallable;
private final UnaryCallable exportAgentCallable;
private final OperationCallable
exportAgentOperationCallable;
private final UnaryCallable importAgentCallable;
private final OperationCallable importAgentOperationCallable;
private final UnaryCallable restoreAgentCallable;
private final OperationCallable restoreAgentOperationCallable;
private final UnaryCallable
getValidationResultCallable;
private final BackgroundResource backgroundResources;
private final GrpcOperationsStub operationsStub;
private final GrpcStubCallableFactory callableFactory;
public static final GrpcAgentsStub create(AgentsStubSettings settings) throws IOException {
return new GrpcAgentsStub(settings, ClientContext.create(settings));
}
public static final GrpcAgentsStub create(ClientContext clientContext) throws IOException {
return new GrpcAgentsStub(AgentsStubSettings.newBuilder().build(), clientContext);
}
public static final GrpcAgentsStub create(
ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException {
return new GrpcAgentsStub(
AgentsStubSettings.newBuilder().build(), clientContext, callableFactory);
}
/**
* Constructs an instance of GrpcAgentsStub, 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 GrpcAgentsStub(AgentsStubSettings settings, ClientContext clientContext)
throws IOException {
this(settings, clientContext, new GrpcAgentsCallableFactory());
}
/**
* Constructs an instance of GrpcAgentsStub, 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 GrpcAgentsStub(
AgentsStubSettings settings,
ClientContext clientContext,
GrpcStubCallableFactory callableFactory)
throws IOException {
this.callableFactory = callableFactory;
this.operationsStub = GrpcOperationsStub.create(clientContext, callableFactory);
GrpcCallSettings getAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(getAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings setAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(setAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("agent.parent", String.valueOf(request.getAgent().getParent()));
return params.build();
})
.build();
GrpcCallSettings deleteAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(deleteAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings searchAgentsTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(searchAgentsMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings trainAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(trainAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings exportAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(exportAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings importAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(importAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings restoreAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(restoreAgentMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
GrpcCallSettings
getValidationResultTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(getValidationResultMethodDescriptor)
.setParamsExtractor(
request -> {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
})
.build();
this.getAgentCallable =
callableFactory.createUnaryCallable(
getAgentTransportSettings, settings.getAgentSettings(), clientContext);
this.setAgentCallable =
callableFactory.createUnaryCallable(
setAgentTransportSettings, settings.setAgentSettings(), clientContext);
this.deleteAgentCallable =
callableFactory.createUnaryCallable(
deleteAgentTransportSettings, settings.deleteAgentSettings(), clientContext);
this.searchAgentsCallable =
callableFactory.createUnaryCallable(
searchAgentsTransportSettings, settings.searchAgentsSettings(), clientContext);
this.searchAgentsPagedCallable =
callableFactory.createPagedCallable(
searchAgentsTransportSettings, settings.searchAgentsSettings(), clientContext);
this.trainAgentCallable =
callableFactory.createUnaryCallable(
trainAgentTransportSettings, settings.trainAgentSettings(), clientContext);
this.trainAgentOperationCallable =
callableFactory.createOperationCallable(
trainAgentTransportSettings,
settings.trainAgentOperationSettings(),
clientContext,
operationsStub);
this.exportAgentCallable =
callableFactory.createUnaryCallable(
exportAgentTransportSettings, settings.exportAgentSettings(), clientContext);
this.exportAgentOperationCallable =
callableFactory.createOperationCallable(
exportAgentTransportSettings,
settings.exportAgentOperationSettings(),
clientContext,
operationsStub);
this.importAgentCallable =
callableFactory.createUnaryCallable(
importAgentTransportSettings, settings.importAgentSettings(), clientContext);
this.importAgentOperationCallable =
callableFactory.createOperationCallable(
importAgentTransportSettings,
settings.importAgentOperationSettings(),
clientContext,
operationsStub);
this.restoreAgentCallable =
callableFactory.createUnaryCallable(
restoreAgentTransportSettings, settings.restoreAgentSettings(), clientContext);
this.restoreAgentOperationCallable =
callableFactory.createOperationCallable(
restoreAgentTransportSettings,
settings.restoreAgentOperationSettings(),
clientContext,
operationsStub);
this.getValidationResultCallable =
callableFactory.createUnaryCallable(
getValidationResultTransportSettings,
settings.getValidationResultSettings(),
clientContext);
this.backgroundResources =
new BackgroundResourceAggregation(clientContext.getBackgroundResources());
}
public GrpcOperationsStub getOperationsStub() {
return operationsStub;
}
@Override
public UnaryCallable getAgentCallable() {
return getAgentCallable;
}
@Override
public UnaryCallable setAgentCallable() {
return setAgentCallable;
}
@Override
public UnaryCallable deleteAgentCallable() {
return deleteAgentCallable;
}
@Override
public UnaryCallable searchAgentsCallable() {
return searchAgentsCallable;
}
@Override
public UnaryCallable searchAgentsPagedCallable() {
return searchAgentsPagedCallable;
}
@Override
public UnaryCallable trainAgentCallable() {
return trainAgentCallable;
}
@Override
public OperationCallable trainAgentOperationCallable() {
return trainAgentOperationCallable;
}
@Override
public UnaryCallable exportAgentCallable() {
return exportAgentCallable;
}
@Override
public OperationCallable
exportAgentOperationCallable() {
return exportAgentOperationCallable;
}
@Override
public UnaryCallable importAgentCallable() {
return importAgentCallable;
}
@Override
public OperationCallable importAgentOperationCallable() {
return importAgentOperationCallable;
}
@Override
public UnaryCallable restoreAgentCallable() {
return restoreAgentCallable;
}
@Override
public OperationCallable restoreAgentOperationCallable() {
return restoreAgentOperationCallable;
}
@Override
public UnaryCallable getValidationResultCallable() {
return getValidationResultCallable;
}
@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);
}
}