diff --git a/mutation_writer/shard_based_splitting_writer.cc b/mutation_writer/shard_based_splitting_writer.cc index 238a622f09..86a29dbf90 100644 --- a/mutation_writer/shard_based_splitting_writer.cc +++ b/mutation_writer/shard_based_splitting_writer.cc @@ -48,13 +48,7 @@ class shard_based_splitting_mutation_writer { return _handle.push(std::move(mf)); } future<> consume_end_of_stream() { - // consume_end_of_stream is always called from a finally block, - // and that's because we wait for _consume_fut to return. We - // don't want to generate another exception here if the read was - // aborted. - if (!_handle.is_terminated()) { - _handle.push_end_of_stream(); - } + _handle.push_end_of_stream(); return std::move(_consume_fut); } void abort(std::exception_ptr ep) { diff --git a/mutation_writer/timestamp_based_splitting_writer.cc b/mutation_writer/timestamp_based_splitting_writer.cc index 350f53f8b3..b6cce9f504 100644 --- a/mutation_writer/timestamp_based_splitting_writer.cc +++ b/mutation_writer/timestamp_based_splitting_writer.cc @@ -139,9 +139,7 @@ class timestamp_based_splitting_mutation_writer { return _handle.push(std::move(mf)); } future<> consume_end_of_stream() { - if (!_handle.is_terminated()) { - _handle.push_end_of_stream(); - } + _handle.push_end_of_stream(); return std::move(_consume_fut); } void abort(std::exception_ptr ep) {