1d84a254c0
The flat_mutation_reader files were conflated and contained multiple readers, which were not strictly necessary. Splitting optimizes both iterative compilation times, as touching rarely used readers doesn't recompile large chunks of codebase. Total compilation times are also improved, as the size of flat_mutation_reader.hh and flat_mutation_reader_v2.hh have been reduced and those files are included by many file in the codebase. With changes real 29m14.051s user 168m39.071s sys 5m13.443s Without changes real 30m36.203s user 175m43.354s sys 5m26.376s Closes #10194
75 lines
2.2 KiB
C++
75 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2016-present ScyllaDB
|
|
*
|
|
* Modified by ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
#include "readers/flat_mutation_reader_v2.hh"
|
|
#include "readers/flat_mutation_reader.hh"
|
|
#include "db/system_keyspace.hh"
|
|
|
|
namespace replica {
|
|
class database;
|
|
}
|
|
|
|
namespace db {
|
|
|
|
namespace size_estimates {
|
|
|
|
struct token_range {
|
|
bytes start;
|
|
bytes end;
|
|
};
|
|
|
|
class size_estimates_mutation_reader final : public flat_mutation_reader_v2::impl {
|
|
replica::database& _db;
|
|
const dht::partition_range* _prange;
|
|
const query::partition_slice& _slice;
|
|
using ks_range = std::vector<sstring>;
|
|
std::optional<ks_range> _keyspaces;
|
|
ks_range::const_iterator _current_partition;
|
|
streamed_mutation::forwarding _fwd;
|
|
flat_mutation_reader_v2_opt _partition_reader;
|
|
public:
|
|
size_estimates_mutation_reader(replica::database& db, schema_ptr, reader_permit, const dht::partition_range&, const query::partition_slice&, streamed_mutation::forwarding);
|
|
|
|
virtual future<> fill_buffer() override;
|
|
virtual future<> next_partition() override;
|
|
virtual future<> fast_forward_to(const dht::partition_range&) override;
|
|
virtual future<> fast_forward_to(position_range) override;
|
|
virtual future<> close() noexcept override;
|
|
private:
|
|
future<> get_next_partition();
|
|
future<> close_partition_reader() noexcept;
|
|
|
|
std::vector<db::system_keyspace::range_estimates>
|
|
estimates_for_current_keyspace(std::vector<token_range> local_ranges) const;
|
|
};
|
|
|
|
struct virtual_reader {
|
|
replica::database& db;
|
|
|
|
flat_mutation_reader_v2 operator()(schema_ptr schema,
|
|
reader_permit permit,
|
|
const dht::partition_range& range,
|
|
const query::partition_slice& slice,
|
|
const io_priority_class& pc,
|
|
tracing::trace_state_ptr trace_state,
|
|
streamed_mutation::forwarding fwd,
|
|
mutation_reader::forwarding fwd_mr) {
|
|
return make_flat_mutation_reader_v2<size_estimates_mutation_reader>(db, std::move(schema), std::move(permit), range, slice, fwd);
|
|
}
|
|
|
|
virtual_reader(replica::database& db_) noexcept : db(db_) {}
|
|
};
|
|
|
|
future<std::vector<token_range>> test_get_local_ranges(replica::database& db);
|
|
|
|
} // namespace size_estimates
|
|
|
|
} // namespace db
|