From 1e2a598fe7f445a83d9395eb2cdd7e4504745813 Mon Sep 17 00:00:00 2001
From: Asias He <asias@scylladb.com>
Date: Tue, 25 Jun 2019 10:10:42 +0800
Subject: [PATCH] repair: Add get_full_row_hashes_sink_op

It is a helper that works on the sink() of the get_full_row_hashes
rpc stream verb.
---
 repair/row_level.cc | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/repair/row_level.cc b/repair/row_level.cc
index cebc617479..04cda5fa8c 100644
--- a/repair/row_level.cc
+++ b/repair/row_level.cc
@@ -1227,6 +1227,16 @@ private:
         });
     }
 
+    future<> get_full_row_hashes_sink_op(rpc::sink<repair_stream_cmd>& sink) {
+        return sink(repair_stream_cmd::get_full_row_hashes).then([&sink] {
+            return sink.flush();
+        }).handle_exception([&sink] (std::exception_ptr ep) {
+            return sink.close().then([ep = std::move(ep)] () mutable {
+                return make_exception_future<>(std::move(ep));
+            });
+        });
+    }
+
 public:
     // RPC handler
     future<std::unordered_set<repair_hash>>