|
|
@ -45,60 +45,78 @@ |
|
|
|
|
|
|
|
|
|
|
|
namespace grpc { |
|
|
|
namespace grpc { |
|
|
|
|
|
|
|
|
|
|
|
// Common interface for all client side streaming.
|
|
|
|
/// Common interface for all synchronous client side streaming.
|
|
|
|
class ClientStreamingInterface { |
|
|
|
class ClientStreamingInterface { |
|
|
|
public: |
|
|
|
public: |
|
|
|
virtual ~ClientStreamingInterface() {} |
|
|
|
virtual ~ClientStreamingInterface() {} |
|
|
|
|
|
|
|
|
|
|
|
// Wait until the stream finishes, and return the final status. When the
|
|
|
|
/// Wait until the stream finishes, and return the final status. When the
|
|
|
|
// client side declares it has no more message to send, either implicitly or
|
|
|
|
/// client side declares it has no more message to send, either implicitly or
|
|
|
|
// by calling WritesDone, it needs to make sure there is no more message to
|
|
|
|
/// by calling \a WritesDone(), it needs to make sure there is no more message
|
|
|
|
// be received from the server, either implicitly or by getting a false from
|
|
|
|
/// to be received from the server, either implicitly or by getting a false
|
|
|
|
// a Read().
|
|
|
|
/// from a \a Read().
|
|
|
|
// This function will return either:
|
|
|
|
///
|
|
|
|
// - when all incoming messages have been read and the server has returned
|
|
|
|
/// This function will return either:
|
|
|
|
// status
|
|
|
|
/// - when all incoming messages have been read and the server has returned
|
|
|
|
// - OR when the server has returned a non-OK status
|
|
|
|
/// status.
|
|
|
|
|
|
|
|
/// - OR when the server has returned a non-OK status.
|
|
|
|
virtual Status Finish() = 0; |
|
|
|
virtual Status Finish() = 0; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
// An interface that yields a sequence of R messages.
|
|
|
|
/// An interface that yields a sequence of messages of type \a R.
|
|
|
|
template <class R> |
|
|
|
template <class R> |
|
|
|
class ReaderInterface { |
|
|
|
class ReaderInterface { |
|
|
|
public: |
|
|
|
public: |
|
|
|
virtual ~ReaderInterface() {} |
|
|
|
virtual ~ReaderInterface() {} |
|
|
|
|
|
|
|
|
|
|
|
// Blocking read a message and parse to msg. Returns true on success.
|
|
|
|
/// Blocking read a message and parse to \a msg. Returns \a true on success.
|
|
|
|
// The method returns false when there will be no more incoming messages,
|
|
|
|
///
|
|
|
|
// either because the other side has called WritesDone or the stream has
|
|
|
|
/// \param[out] msg The read message.
|
|
|
|
// failed (or been cancelled).
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \return \a false when there will be no more incoming messages, either
|
|
|
|
|
|
|
|
/// because the other side has called \a WritesDone() or the stream has failed
|
|
|
|
|
|
|
|
/// (or been cancelled).
|
|
|
|
virtual bool Read(R* msg) = 0; |
|
|
|
virtual bool Read(R* msg) = 0; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
// An interface that can be fed a sequence of W messages.
|
|
|
|
/// An interface that can be fed a sequence of messages of type \a W.
|
|
|
|
template <class W> |
|
|
|
template <class W> |
|
|
|
class WriterInterface { |
|
|
|
class WriterInterface { |
|
|
|
public: |
|
|
|
public: |
|
|
|
virtual ~WriterInterface() {} |
|
|
|
virtual ~WriterInterface() {} |
|
|
|
|
|
|
|
|
|
|
|
// Blocking write msg to the stream. Returns true on success.
|
|
|
|
/// Blocking write \a msg to the stream with options.
|
|
|
|
// Returns false when the stream has been closed.
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \param msg The message to be written to the stream.
|
|
|
|
|
|
|
|
/// \param options Options affecting the write operation.
|
|
|
|
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \return \a true on success, \a false when the stream has been closed.
|
|
|
|
virtual bool Write(const W& msg, const WriteOptions& options) = 0; |
|
|
|
virtual bool Write(const W& msg, const WriteOptions& options) = 0; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Blocking write \a msg to the stream with default options.
|
|
|
|
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \param msg The message to be written to the stream.
|
|
|
|
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \return \a true on success, \a false when the stream has been closed.
|
|
|
|
inline bool Write(const W& msg) { return Write(msg, WriteOptions()); } |
|
|
|
inline bool Write(const W& msg) { return Write(msg, WriteOptions()); } |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Client-side interface for streaming reads of message of type \a R.
|
|
|
|
template <class R> |
|
|
|
template <class R> |
|
|
|
class ClientReaderInterface : public ClientStreamingInterface, |
|
|
|
class ClientReaderInterface : public ClientStreamingInterface, |
|
|
|
public ReaderInterface<R> { |
|
|
|
public ReaderInterface<R> { |
|
|
|
public: |
|
|
|
public: |
|
|
|
|
|
|
|
/// Blocking wait for initial metadata from server. The received metadata
|
|
|
|
|
|
|
|
/// can only be accessed after this call returns. Should only be called before
|
|
|
|
|
|
|
|
/// the first read. Calling this method is optional, and if it is not called
|
|
|
|
|
|
|
|
/// the metadata will be available in ClientContext after the first read.
|
|
|
|
virtual void WaitForInitialMetadata() = 0; |
|
|
|
virtual void WaitForInitialMetadata() = 0; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
template <class R> |
|
|
|
template <class R> |
|
|
|
class ClientReader GRPC_FINAL : public ClientReaderInterface<R> { |
|
|
|
class ClientReader GRPC_FINAL : public ClientReaderInterface<R> { |
|
|
|
public: |
|
|
|
public: |
|
|
|
// Blocking create a stream and write the first request out.
|
|
|
|
/// Blocking create a stream and write the first request out.
|
|
|
|
template <class W> |
|
|
|
template <class W> |
|
|
|
ClientReader(Channel* channel, const RpcMethod& method, |
|
|
|
ClientReader(Channel* channel, const RpcMethod& method, |
|
|
|
ClientContext* context, const W& request) |
|
|
|
ClientContext* context, const W& request) |
|
|
@ -113,17 +131,13 @@ class ClientReader GRPC_FINAL : public ClientReaderInterface<R> { |
|
|
|
cq_.Pluck(&ops); |
|
|
|
cq_.Pluck(&ops); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// Blocking wait for initial metadata from server. The received metadata
|
|
|
|
|
|
|
|
// can only be accessed after this call returns. Should only be called before
|
|
|
|
|
|
|
|
// the first read. Calling this method is optional, and if it is not called
|
|
|
|
|
|
|
|
// the metadata will be available in ClientContext after the first read.
|
|
|
|
|
|
|
|
void WaitForInitialMetadata() { |
|
|
|
void WaitForInitialMetadata() { |
|
|
|
GPR_ASSERT(!context_->initial_metadata_received_); |
|
|
|
GPR_ASSERT(!context_->initial_metadata_received_); |
|
|
|
|
|
|
|
|
|
|
|
CallOpSet<CallOpRecvInitialMetadata> ops; |
|
|
|
CallOpSet<CallOpRecvInitialMetadata> ops; |
|
|
|
ops.RecvInitialMetadata(context_); |
|
|
|
ops.RecvInitialMetadata(context_); |
|
|
|
call_.PerformOps(&ops); |
|
|
|
call_.PerformOps(&ops); |
|
|
|
cq_.Pluck(&ops); // status ignored
|
|
|
|
cq_.Pluck(&ops); /// status ignored
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
bool Read(R* msg) GRPC_OVERRIDE { |
|
|
|
bool Read(R* msg) GRPC_OVERRIDE { |
|
|
@ -151,17 +165,22 @@ class ClientReader GRPC_FINAL : public ClientReaderInterface<R> { |
|
|
|
Call call_; |
|
|
|
Call call_; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Client-side interface for streaming writes of message of type \a W.
|
|
|
|
template <class W> |
|
|
|
template <class W> |
|
|
|
class ClientWriterInterface : public ClientStreamingInterface, |
|
|
|
class ClientWriterInterface : public ClientStreamingInterface, |
|
|
|
public WriterInterface<W> { |
|
|
|
public WriterInterface<W> { |
|
|
|
public: |
|
|
|
public: |
|
|
|
|
|
|
|
/// Half close writing from the client.
|
|
|
|
|
|
|
|
/// Block until writes are completed.
|
|
|
|
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \return Whether the writes were successful.
|
|
|
|
virtual bool WritesDone() = 0; |
|
|
|
virtual bool WritesDone() = 0; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
template <class W> |
|
|
|
template <class W> |
|
|
|
class ClientWriter : public ClientWriterInterface<W> { |
|
|
|
class ClientWriter : public ClientWriterInterface<W> { |
|
|
|
public: |
|
|
|
public: |
|
|
|
// Blocking create a stream.
|
|
|
|
/// Blocking create a stream.
|
|
|
|
template <class R> |
|
|
|
template <class R> |
|
|
|
ClientWriter(Channel* channel, const RpcMethod& method, |
|
|
|
ClientWriter(Channel* channel, const RpcMethod& method, |
|
|
|
ClientContext* context, R* response) |
|
|
|
ClientContext* context, R* response) |
|
|
@ -191,7 +210,7 @@ class ClientWriter : public ClientWriterInterface<W> { |
|
|
|
return cq_.Pluck(&ops); |
|
|
|
return cq_.Pluck(&ops); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// Read the final response and wait for the final status.
|
|
|
|
/// Read the final response and wait for the final status.
|
|
|
|
Status Finish() GRPC_OVERRIDE { |
|
|
|
Status Finish() GRPC_OVERRIDE { |
|
|
|
Status status; |
|
|
|
Status status; |
|
|
|
finish_ops_.ClientRecvStatus(context_, &status); |
|
|
|
finish_ops_.ClientRecvStatus(context_, &status); |
|
|
@ -207,20 +226,28 @@ class ClientWriter : public ClientWriterInterface<W> { |
|
|
|
Call call_; |
|
|
|
Call call_; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
// Client-side interface for bi-directional streaming.
|
|
|
|
/// Client-side interface for bi-directional streaming.
|
|
|
|
template <class W, class R> |
|
|
|
template <class W, class R> |
|
|
|
class ClientReaderWriterInterface : public ClientStreamingInterface, |
|
|
|
class ClientReaderWriterInterface : public ClientStreamingInterface, |
|
|
|
public WriterInterface<W>, |
|
|
|
public WriterInterface<W>, |
|
|
|
public ReaderInterface<R> { |
|
|
|
public ReaderInterface<R> { |
|
|
|
public: |
|
|
|
public: |
|
|
|
|
|
|
|
/// Blocking wait for initial metadata from server. The received metadata
|
|
|
|
|
|
|
|
/// can only be accessed after this call returns. Should only be called before
|
|
|
|
|
|
|
|
/// the first read. Calling this method is optional, and if it is not called
|
|
|
|
|
|
|
|
/// the metadata will be available in ClientContext after the first read.
|
|
|
|
virtual void WaitForInitialMetadata() = 0; |
|
|
|
virtual void WaitForInitialMetadata() = 0; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// Block until writes are completed.
|
|
|
|
|
|
|
|
///
|
|
|
|
|
|
|
|
/// \return Whether the writes were successful.
|
|
|
|
virtual bool WritesDone() = 0; |
|
|
|
virtual bool WritesDone() = 0; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
template <class W, class R> |
|
|
|
template <class W, class R> |
|
|
|
class ClientReaderWriter GRPC_FINAL : public ClientReaderWriterInterface<W, R> { |
|
|
|
class ClientReaderWriter GRPC_FINAL : public ClientReaderWriterInterface<W, R> { |
|
|
|
public: |
|
|
|
public: |
|
|
|
// Blocking create a stream.
|
|
|
|
/// Blocking create a stream.
|
|
|
|
ClientReaderWriter(Channel* channel, const RpcMethod& method, |
|
|
|
ClientReaderWriter(Channel* channel, const RpcMethod& method, |
|
|
|
ClientContext* context) |
|
|
|
ClientContext* context) |
|
|
|
: context_(context), call_(channel->CreateCall(method, context, &cq_)) { |
|
|
|
: context_(context), call_(channel->CreateCall(method, context, &cq_)) { |
|
|
@ -230,10 +257,6 @@ class ClientReaderWriter GRPC_FINAL : public ClientReaderWriterInterface<W, R> { |
|
|
|
cq_.Pluck(&ops); |
|
|
|
cq_.Pluck(&ops); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// Blocking wait for initial metadata from server. The received metadata
|
|
|
|
|
|
|
|
// can only be accessed after this call returns. Should only be called before
|
|
|
|
|
|
|
|
// the first read. Calling this method is optional, and if it is not called
|
|
|
|
|
|
|
|
// the metadata will be available in ClientContext after the first read.
|
|
|
|
|
|
|
|
void WaitForInitialMetadata() { |
|
|
|
void WaitForInitialMetadata() { |
|
|
|
GPR_ASSERT(!context_->initial_metadata_received_); |
|
|
|
GPR_ASSERT(!context_->initial_metadata_received_); |
|
|
|
|
|
|
|
|
|
|
@ -344,7 +367,7 @@ class ServerWriter GRPC_FINAL : public WriterInterface<W> { |
|
|
|
ServerContext* const ctx_; |
|
|
|
ServerContext* const ctx_; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
// Server-side interface for bi-directional streaming.
|
|
|
|
/// Server-side interface for bi-directional streaming.
|
|
|
|
template <class W, class R> |
|
|
|
template <class W, class R> |
|
|
|
class ServerReaderWriter GRPC_FINAL : public WriterInterface<W>, |
|
|
|
class ServerReaderWriter GRPC_FINAL : public WriterInterface<W>, |
|
|
|
public ReaderInterface<R> { |
|
|
|
public ReaderInterface<R> { |
|
|
|