com.google.cloud.dialogflow.v2.stub.GrpcAgentsStub Maven / Gradle / Ivy
Show all versions of google-cloud-dialogflow Show documentation
/*
* Copyright 2019 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.v2.stub;
import static com.google.cloud.dialogflow.v2.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.RequestParamsExtractor;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.cloud.dialogflow.v2.Agent;
import com.google.cloud.dialogflow.v2.ExportAgentRequest;
import com.google.cloud.dialogflow.v2.ExportAgentResponse;
import com.google.cloud.dialogflow.v2.GetAgentRequest;
import com.google.cloud.dialogflow.v2.ImportAgentRequest;
import com.google.cloud.dialogflow.v2.RestoreAgentRequest;
import com.google.cloud.dialogflow.v2.SearchAgentsRequest;
import com.google.cloud.dialogflow.v2.SearchAgentsResponse;
import com.google.cloud.dialogflow.v2.TrainAgentRequest;
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.Map;
import java.util.concurrent.TimeUnit;
import javax.annotation.Generated;
// AUTO-GENERATED DOCUMENTATION AND CLASS
/**
* gRPC stub implementation for Dialogflow API.
*
* This class is for advanced usage and reflects the underlying API directly.
*/
@Generated("by gapic-generator")
@BetaApi("A restructuring of stub classes is planned, so this may break in the future")
public class GrpcAgentsStub extends AgentsStub {
private static final MethodDescriptor getAgentMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2.Agents/GetAgent")
.setRequestMarshaller(ProtoUtils.marshaller(GetAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Agent.getDefaultInstance()))
.build();
private static final MethodDescriptor
searchAgentsMethodDescriptor =
MethodDescriptor.newBuilder()
.setType(MethodDescriptor.MethodType.UNARY)
.setFullMethodName("google.cloud.dialogflow.v2.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.v2.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.v2.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.v2.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.v2.Agents/RestoreAgent")
.setRequestMarshaller(ProtoUtils.marshaller(RestoreAgentRequest.getDefaultInstance()))
.setResponseMarshaller(ProtoUtils.marshaller(Operation.getDefaultInstance()))
.build();
private final BackgroundResource backgroundResources;
private final GrpcOperationsStub operationsStub;
private final UnaryCallable getAgentCallable;
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 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(
new RequestParamsExtractor() {
@Override
public Map extract(GetAgentRequest request) {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
}
})
.build();
GrpcCallSettings searchAgentsTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(searchAgentsMethodDescriptor)
.setParamsExtractor(
new RequestParamsExtractor() {
@Override
public Map extract(SearchAgentsRequest request) {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
}
})
.build();
GrpcCallSettings trainAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(trainAgentMethodDescriptor)
.setParamsExtractor(
new RequestParamsExtractor() {
@Override
public Map extract(TrainAgentRequest request) {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
}
})
.build();
GrpcCallSettings exportAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(exportAgentMethodDescriptor)
.setParamsExtractor(
new RequestParamsExtractor() {
@Override
public Map extract(ExportAgentRequest request) {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
}
})
.build();
GrpcCallSettings importAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(importAgentMethodDescriptor)
.setParamsExtractor(
new RequestParamsExtractor() {
@Override
public Map extract(ImportAgentRequest request) {
ImmutableMap.Builder params = ImmutableMap.builder();
params.put("parent", String.valueOf(request.getParent()));
return params.build();
}
})
.build();
GrpcCallSettings restoreAgentTransportSettings =
GrpcCallSettings.newBuilder()
.setMethodDescriptor(restoreAgentMethodDescriptor)
.setParamsExtractor(
new RequestParamsExtractor() {
@Override
public Map extract(RestoreAgentRequest 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.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,
this.operationsStub);
this.exportAgentCallable =
callableFactory.createUnaryCallable(
exportAgentTransportSettings, settings.exportAgentSettings(), clientContext);
this.exportAgentOperationCallable =
callableFactory.createOperationCallable(
exportAgentTransportSettings,
settings.exportAgentOperationSettings(),
clientContext,
this.operationsStub);
this.importAgentCallable =
callableFactory.createUnaryCallable(
importAgentTransportSettings, settings.importAgentSettings(), clientContext);
this.importAgentOperationCallable =
callableFactory.createOperationCallable(
importAgentTransportSettings,
settings.importAgentOperationSettings(),
clientContext,
this.operationsStub);
this.restoreAgentCallable =
callableFactory.createUnaryCallable(
restoreAgentTransportSettings, settings.restoreAgentSettings(), clientContext);
this.restoreAgentOperationCallable =
callableFactory.createOperationCallable(
restoreAgentTransportSettings,
settings.restoreAgentOperationSettings(),
clientContext,
this.operationsStub);
backgroundResources = new BackgroundResourceAggregation(clientContext.getBackgroundResources());
}
@BetaApi("The surface for use by generated code is not stable yet and may change in the future.")
public GrpcOperationsStub getOperationsStub() {
return operationsStub;
}
public UnaryCallable getAgentCallable() {
return getAgentCallable;
}
public UnaryCallable searchAgentsPagedCallable() {
return searchAgentsPagedCallable;
}
public UnaryCallable searchAgentsCallable() {
return searchAgentsCallable;
}
@BetaApi("The surface for use by generated code is not stable yet and may change in the future.")
public OperationCallable trainAgentOperationCallable() {
return trainAgentOperationCallable;
}
public UnaryCallable trainAgentCallable() {
return trainAgentCallable;
}
@BetaApi("The surface for use by generated code is not stable yet and may change in the future.")
public OperationCallable
exportAgentOperationCallable() {
return exportAgentOperationCallable;
}
public UnaryCallable exportAgentCallable() {
return exportAgentCallable;
}
@BetaApi("The surface for use by generated code is not stable yet and may change in the future.")
public OperationCallable importAgentOperationCallable() {
return importAgentOperationCallable;
}
public UnaryCallable importAgentCallable() {
return importAgentCallable;
}
@BetaApi("The surface for use by generated code is not stable yet and may change in the future.")
public OperationCallable restoreAgentOperationCallable() {
return restoreAgentOperationCallable;
}
public UnaryCallable restoreAgentCallable() {
return restoreAgentCallable;
}
@Override
public final void close() {
shutdown();
}
@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);
}
}