Files
scylladb/mutation_partition_view.hh
Benny Halevy 4963eb73b5 mutation_partition_view: add accept_gently methods
Allow yielding when consuming mutation_partition_view.
To be used in later patches by a new unfreeze_gently function
and frozen_mutation::consume.

Signed-off-by: Benny Halevy <bhalevy@scylladb.com>
2022-05-05 13:32:25 +03:00

80 lines
3.1 KiB
C++

/*
* Copyright (C) 2015-present ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#pragma once
#include "replica/database_fwd.hh"
#include "mutation_partition_visitor.hh"
#include "utils/input_stream.hh"
#include "atomic_cell.hh"
namespace ser {
class mutation_partition_view;
}
class partition_builder;
class converting_mutation_partition_applier;
template<typename T>
concept MutationViewVisitor = requires (T& visitor, tombstone t, atomic_cell ac,
collection_mutation_view cmv, range_tombstone rt,
position_in_partition_view pipv, row_tombstone row_tomb,
row_marker rm) {
visitor.accept_partition_tombstone(t);
visitor.accept_static_cell(column_id(), std::move(ac));
visitor.accept_static_cell(column_id(), cmv);
visitor.accept_row_tombstone(rt);
visitor.accept_row(pipv, row_tomb, rm,
is_dummy::no, is_continuous::yes);
visitor.accept_row_cell(column_id(), std::move(ac));
visitor.accept_row_cell(column_id(), cmv);
};
class mutation_partition_view_virtual_visitor {
public:
virtual ~mutation_partition_view_virtual_visitor();
virtual void accept_partition_tombstone(tombstone t) = 0;
virtual void accept_static_cell(column_id, atomic_cell ac) = 0;
virtual void accept_static_cell(column_id, collection_mutation_view cmv) = 0;
virtual void accept_row_tombstone(range_tombstone rt) = 0;
virtual void accept_row(position_in_partition_view pipv, row_tombstone rt, row_marker rm, is_dummy, is_continuous) = 0;
virtual void accept_row_cell(column_id, atomic_cell ac) = 0;
virtual void accept_row_cell(column_id, collection_mutation_view cmv) = 0;
};
// View on serialized mutation partition. See mutation_partition_serializer.
class mutation_partition_view {
utils::input_stream _in;
private:
mutation_partition_view(utils::input_stream v)
: _in(v)
{ }
template<typename Visitor>
requires MutationViewVisitor<Visitor>
void do_accept(const column_mapping&, Visitor& visitor) const;
template<typename Visitor>
requires MutationViewVisitor<Visitor>
future<> do_accept_gently(const column_mapping&, Visitor& visitor) const;
public:
static mutation_partition_view from_stream(utils::input_stream v) {
return { v };
}
static mutation_partition_view from_view(ser::mutation_partition_view v);
void accept(const schema& schema, partition_builder& visitor) const;
future<> accept_gently(const schema& schema, partition_builder& visitor) const;
void accept(const column_mapping&, converting_mutation_partition_applier& visitor) const;
future<> accept_gently(const column_mapping&, converting_mutation_partition_applier& visitor) const;
void accept(const column_mapping&, mutation_partition_view_virtual_visitor& mpvvv) const;
future<> accept_gently(const column_mapping&, mutation_partition_view_virtual_visitor& mpvvv) const;
std::optional<clustering_key> first_row_key() const;
std::optional<clustering_key> last_row_key() const;
};