mirrors/scylladb
1
0
Fork 0
mirror of https://github.com/scylladb/scylladb.git synced 2026-06-06 06:53:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

mutation_reader: Make consume() work with deferring callbacks too

Browse Source
This commit is contained in:
Tomasz Grabiec
2015-07-03 12:31:22 +02:00
parent 59a95f8237
commit 01c698def4
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mutation_reader.hh
View File

@@ -36,15 +36,16 @@ mutation_reader make_empty_reader();
template <typename Consumer>
inline
future<> consume(mutation_reader& reader, Consumer consumer) {
static_assert(std::is_same<stop_iteration, std::result_of_t<Consumer(mutation&&)>>::value, "bad Consumer signature");
static_assert(std::is_same<future<stop_iteration>, futurize_t<std::result_of_t<Consumer(mutation&&)>>>::value, "bad Consumer signature");
using futurator = futurize<std::result_of_t<Consumer(mutation&&)>>;
return do_with(std::move(consumer), [&reader] (Consumer& c) -> future<> {
return repeat([&reader, &c] () {
return reader().then([&c] (mutation_opt&& mo) {
return reader().then([&c] (mutation_opt&& mo) -> future<stop_iteration> {
if (!mo) {
return stop_iteration::yes;
return make_ready_future<stop_iteration>(stop_iteration::yes);
}
return c(std::move(*mo));
return futurator::apply(c, std::move(*mo));
});
});
});