mirror of https://github.com/grpc/grpc.git
Async client api change. Add a ClientAsyncResponseReader. Make the api similar to streaming and symmetric to server side.
Yang Gao
10 years ago
parent
d0d007e6d1
commit
3a5e5495e5
9 changed files with 213 additions and 128 deletions
@ -0,0 +1,144 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2014, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef __GRPCPP_ASYNC_UNARY_CALL_H__ |
||||
#define __GRPCPP_ASYNC_UNARY_CALL_H__ |
||||
|
||||
#include <grpc++/channel_interface.h> |
||||
#include <grpc++/client_context.h> |
||||
#include <grpc++/completion_queue.h> |
||||
#include <grpc++/server_context.h> |
||||
#include <grpc++/impl/call.h> |
||||
#include <grpc++/impl/service_type.h> |
||||
#include <grpc++/status.h> |
||||
#include <grpc/support/log.h> |
||||
|
||||
namespace grpc { |
||||
template <class R> |
||||
class ClientAsyncResponseReader final { |
||||
public: |
||||
ClientAsyncResponseReader(ChannelInterface* channel, CompletionQueue* cq, |
||||
const RpcMethod& method, ClientContext* context, |
||||
const google::protobuf::Message& request, void* tag) |
||||
: context_(context), |
||||
call_(channel->CreateCall(method, context, cq)) { |
||||
init_buf_.Reset(tag); |
||||
init_buf_.AddSendInitialMetadata(&context->send_initial_metadata_); |
||||
init_buf_.AddSendMessage(request); |
||||
init_buf_.AddClientSendClose(); |
||||
call_.PerformOps(&init_buf_); |
||||
} |
||||
|
||||
void ReadInitialMetadata(void* tag) { |
||||
GPR_ASSERT(!context_->initial_metadata_received_); |
||||
|
||||
meta_buf_.Reset(tag); |
||||
meta_buf_.AddRecvInitialMetadata(context_); |
||||
call_.PerformOps(&meta_buf_); |
||||
} |
||||
|
||||
void Finish(R* msg, Status* status, void* tag) { |
||||
finish_buf_.Reset(tag); |
||||
if (!context_->initial_metadata_received_) { |
||||
finish_buf_.AddRecvInitialMetadata(context_); |
||||
} |
||||
finish_buf_.AddRecvMessage(msg); |
||||
finish_buf_.AddClientRecvStatus(context_, status); |
||||
call_.PerformOps(&finish_buf_); |
||||
} |
||||
|
||||
|
||||
private: |
||||
ClientContext* context_ = nullptr; |
||||
Call call_; |
||||
CallOpBuffer init_buf_; |
||||
CallOpBuffer meta_buf_; |
||||
CallOpBuffer finish_buf_; |
||||
}; |
||||
|
||||
template <class W> |
||||
class ServerAsyncResponseWriter final : public ServerAsyncStreamingInterface { |
||||
public: |
||||
explicit ServerAsyncResponseWriter(ServerContext* ctx) |
||||
: call_(nullptr, nullptr, nullptr), ctx_(ctx) {} |
||||
|
||||
void SendInitialMetadata(void* tag) { |
||||
GPR_ASSERT(!ctx_->sent_initial_metadata_); |
||||
|
||||
meta_buf_.Reset(tag); |
||||
meta_buf_.AddSendInitialMetadata(&ctx_->initial_metadata_); |
||||
ctx_->sent_initial_metadata_ = true; |
||||
call_.PerformOps(&meta_buf_); |
||||
} |
||||
|
||||
void Finish(const W& msg, const Status& status, void* tag) { |
||||
finish_buf_.Reset(tag); |
||||
if (!ctx_->sent_initial_metadata_) { |
||||
finish_buf_.AddSendInitialMetadata(&ctx_->initial_metadata_); |
||||
ctx_->sent_initial_metadata_ = true; |
||||
} |
||||
// The response is dropped if the status is not OK.
|
||||
if (status.IsOk()) { |
||||
finish_buf_.AddSendMessage(msg); |
||||
} |
||||
bool cancelled = false; |
||||
finish_buf_.AddServerRecvClose(&cancelled); |
||||
finish_buf_.AddServerSendStatus(&ctx_->trailing_metadata_, status); |
||||
call_.PerformOps(&finish_buf_); |
||||
} |
||||
|
||||
void FinishWithError(const Status& status, void* tag) { |
||||
GPR_ASSERT(!status.IsOk()); |
||||
finish_buf_.Reset(tag); |
||||
if (!ctx_->sent_initial_metadata_) { |
||||
finish_buf_.AddSendInitialMetadata(&ctx_->initial_metadata_); |
||||
ctx_->sent_initial_metadata_ = true; |
||||
} |
||||
bool cancelled = false; |
||||
finish_buf_.AddServerRecvClose(&cancelled); |
||||
finish_buf_.AddServerSendStatus(&ctx_->trailing_metadata_, status); |
||||
call_.PerformOps(&finish_buf_); |
||||
} |
||||
|
||||
private: |
||||
void BindCall(Call* call) override { call_ = *call; } |
||||
|
||||
Call call_; |
||||
ServerContext* ctx_; |
||||
CallOpBuffer meta_buf_; |
||||
CallOpBuffer finish_buf_; |
||||
}; |
||||
|
||||
} // namespace grpc
|
||||
|
||||
#endif // __GRPCPP_ASYNC_UNARY_CALL_H__
|
||||
Loading…
Reference in new issue