grpc: add RPCSingleStreamCall base for single-streaming wrappers

1 files changed, 13 insertions, 0 deletions

diff --git a/src/murmur/MurmurGRPCImpl.h b/src/murmur/MurmurGRPCImpl.h
index aacd8e4a8..6871603da 100644
--- a/src/murmur/MurmurGRPCImpl.h
+++ b/src/murmur/MurmurGRPCImpl.h
@@ -194,6 +194,19 @@ public:
 };
 
 template <class InType, class OutType>
+class RPCSingleStreamCall : public RPCCall {
+public:
+	InType request;
+	::grpc::ServerAsyncWriter < OutType > stream;
+	RPCSingleStreamCall(MurmurRPCImpl *rpcImpl) : RPCCall(rpcImpl), stream(&context) {
+	}
+
+	virtual void error(const ::grpc::Status &err) {
+		stream.Finish(err, this->done());
+	}
+};
+
+template <class InType, class OutType>
 class RPCStreamStreamCall : public RPCCall {
 public:
 	InType request;