mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
3.0 KiB
91 lines
3.0 KiB
7 years ago
|
/*
|
||
|
|
*
|
||
|
|
* Copyright 2015 gRPC authors.
|
||
|
|
*
|
||
|
|
* 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
|
||
|
|
*
|
||
|
|
* http://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.
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef GRPCPP_IMPL_CODEGEN_CLIENT_UNARY_CALL_H
|
||
|
|
#define GRPCPP_IMPL_CODEGEN_CLIENT_UNARY_CALL_H
|
||
|
|
|
||
|
|
#include <grpcpp/impl/codegen/call.h>
|
||
|
|
#include <grpcpp/impl/codegen/channel_interface.h>
|
||
|
|
#include <grpcpp/impl/codegen/config.h>
|
||
|
|
#include <grpcpp/impl/codegen/core_codegen_interface.h>
|
||
|
|
#include <grpcpp/impl/codegen/status.h>
|
||
|
|
|
||
|
|
namespace grpc {
|
||
|
|
|
||
|
|
class Channel;
|
||
|
|
class ClientContext;
|
||
|
|
class CompletionQueue;
|
||
|
|
|
||
|
|
namespace internal {
|
||
|
|
class RpcMethod;
|
||
|
|
/// Wrapper that performs a blocking unary call
|
||
|
|
template <class InputMessage, class OutputMessage>
|
||
|
|
Status BlockingUnaryCall(ChannelInterface* channel, const RpcMethod& method,
|
||
|
|
ClientContext* context, const InputMessage& request,
|
||
|
|
OutputMessage* result) {
|
||
|
|
return BlockingUnaryCallImpl<InputMessage, OutputMessage>(
|
||
|
|
channel, method, context, request, result)
|
||
|
|
.status();
|
||
|
|
}
|
||
|
|
|
||
|
|
template <class InputMessage, class OutputMessage>
|
||
|
|
class BlockingUnaryCallImpl {
|
||
|
|
public:
|
||
|
|
BlockingUnaryCallImpl(ChannelInterface* channel, const RpcMethod& method,
|
||
|
|
ClientContext* context, const InputMessage& request,
|
||
|
|
OutputMessage* result) {
|
||
|
|
CompletionQueue cq(grpc_completion_queue_attributes{
|
||
|
|
GRPC_CQ_CURRENT_VERSION, GRPC_CQ_PLUCK,
|
||
|
|
GRPC_CQ_DEFAULT_POLLING}); // Pluckable completion queue
|
||
|
|
Call call(channel->CreateCall(method, context, &cq));
|
||
|
|
CallOpSet<CallOpSendInitialMetadata, CallOpSendMessage,
|
||
|
|
CallOpRecvInitialMetadata, CallOpRecvMessage<OutputMessage>,
|
||
|
|
CallOpClientSendClose, CallOpClientRecvStatus>
|
||
|
|
ops;
|
||
|
|
status_ = ops.SendMessage(request);
|
||
|
|
if (!status_.ok()) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
ops.SendInitialMetadata(context->send_initial_metadata_,
|
||
|
|
context->initial_metadata_flags());
|
||
|
|
ops.RecvInitialMetadata(context);
|
||
|
|
ops.RecvMessage(result);
|
||
|
|
ops.AllowNoMessage();
|
||
|
|
ops.ClientSendClose();
|
||
|
|
ops.ClientRecvStatus(context, &status_);
|
||
|
|
call.PerformOps(&ops);
|
||
|
|
if (cq.Pluck(&ops)) {
|
||
|
|
if (!ops.got_message && status_.ok()) {
|
||
|
|
status_ = Status(StatusCode::UNIMPLEMENTED,
|
||
|
|
"No message returned for unary request");
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
GPR_CODEGEN_ASSERT(!status_.ok());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
Status status() { return status_; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
Status status_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace internal
|
||
|
|
} // namespace grpc
|
||
|
|
|
||
|
|
#endif // GRPCPP_IMPL_CODEGEN_CLIENT_UNARY_CALL_H
|