bahman/scylla
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

tablets: Introduce tablet_map::for_each_tablet()

Browse Source
This commit is contained in:
Tomasz Grabiec
2023-07-19 23:33:34 +02:00
parent 0e3eac29d0
commit d59b8d316c
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
locator/tablets.cc
View File

@@ -153,6 +153,15 @@ void tablet_map::set_tablet_transition_info(tablet_id id, tablet_transition_info
_transitions.insert_or_assign(id, std::move(info)); _transitions.insert_or_assign(id, std::move(info));
} }
future<> tablet_map::for_each_tablet(seastar::noncopyable_function<void(tablet_id, const tablet_info&)> func) const {
std::optional<tablet_id> tid = first_tablet();
for (const tablet_info& ti : tablets()) {
co_await coroutine::maybe_yield();
func(*tid, ti);
tid = next_tablet(*tid);
}
}
std::optional<shard_id> tablet_map::get_shard(tablet_id tid, host_id host) const { std::optional<shard_id> tablet_map::get_shard(tablet_id tid, host_id host) const {
auto&& info = get_tablet_info(tid); auto&& info = get_tablet_info(tid);

6
locator/tablets.hh
View File

@@ -19,6 +19,9 @@
#include <boost/range/adaptor/transformed.hpp> #include <boost/range/adaptor/transformed.hpp>
#include <seastar/core/reactor.hh> #include <seastar/core/reactor.hh>
#include <seastar/util/log.hh> #include <seastar/util/log.hh>
#include <seastar/core/coroutine.hh>
#include <seastar/util/noncopyable_function.hh>
#include <seastar/coroutine/maybe_yield.hh>
#include <vector> #include <vector>
@@ -216,6 +219,9 @@ public:
return _tablets; return _tablets;
} }
/// Calls a given function for each tablet in the map in token ownership order.
future<> for_each_tablet(seastar::noncopyable_function<void(tablet_id, const tablet_info&)> func) const;
const auto& transitions() const { const auto& transitions() const {
return _transitions; return _transitions;
} }