+++ /dev/null
-// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:nil -*-
-// vim: ts=8 sw=2 smarttab
-
-#include "tree.h"
-
-#include "node.h"
-#include "node_extent_manager.h"
-#include "stages/key_layout.h"
-#include "super.h"
-
-namespace crimson::os::seastore::onode {
-
-using btree_ertr = Btree::btree_ertr;
-template <class ValueT=void>
-using btree_future = Btree::btree_future<ValueT>;
-using Cursor = Btree::Cursor;
-
-Cursor::Cursor(Btree* p_tree, Ref<tree_cursor_t> _p_cursor)
- : p_tree(p_tree) {
- if (_p_cursor->is_end()) {
- // no need to hold the leaf node
- } else {
- p_cursor = _p_cursor;
- }
-}
-Cursor::Cursor(Btree* p_tree) : p_tree{p_tree} {}
-Cursor::Cursor(const Cursor&) = default;
-Cursor::Cursor(Cursor&&) noexcept = default;
-Cursor& Cursor::operator=(const Cursor&) = default;
-Cursor& Cursor::operator=(Cursor&&) = default;
-Cursor::~Cursor() = default;
-
-bool Cursor::is_end() const {
- if (p_cursor) {
- assert(!p_cursor->is_end());
- return false;
- } else {
- return true;
- }
-}
-
-ghobject_t Cursor::get_ghobj() const {
- return p_cursor->get_key_view().to_ghobj();
-}
-
-const onode_t* Cursor::value() const {
- return p_cursor->get_p_value();
-}
-
-bool Cursor::operator==(const Cursor& x) const {
- return p_cursor == x.p_cursor;
-}
-
-Cursor& Cursor::operator++() {
- // TODO
- return *this;
-}
-
-Cursor Cursor::operator++(int) {
- Cursor tmp = *this;
- ++*this;
- return tmp;
-}
-
-Cursor Cursor::make_end(Btree* p_tree) {
- return {p_tree};
-}
-
-Btree::Btree(NodeExtentManagerURef&& _nm)
- : nm{std::move(_nm)},
- root_tracker{RootNodeTracker::create(nm->is_read_isolated())} {}
-
-Btree::~Btree() { assert(root_tracker->is_clean()); }
-
-btree_future<> Btree::mkfs(Transaction& t) {
- return Node::mkfs(get_context(t), *root_tracker);
-}
-
-btree_future<Cursor> Btree::begin(Transaction& t) {
- return get_root(t).safe_then([this, &t](auto root) {
- return root->lookup_smallest(get_context(t));
- }).safe_then([this](auto cursor) {
- return Cursor{this, cursor};
- });
-}
-
-btree_future<Cursor> Btree::last(Transaction& t) {
- return get_root(t).safe_then([this, &t](auto root) {
- return root->lookup_largest(get_context(t));
- }).safe_then([this](auto cursor) {
- return Cursor(this, cursor);
- });
-}
-
-Cursor Btree::end() {
- return Cursor::make_end(this);
-}
-
-btree_future<bool>
-Btree::contains(Transaction& t, const ghobject_t& obj) {
- return seastar::do_with(
- full_key_t<KeyT::HOBJ>(obj),
- [this, &t](auto& key) -> btree_future<bool> {
- return get_root(t).safe_then([this, &t, &key](auto root) {
- // TODO: improve lower_bound()
- return root->lower_bound(get_context(t), key);
- }).safe_then([](auto result) {
- return MatchKindBS::EQ == result.match();
- });
- }
- );
-}
-
-btree_future<Cursor>
-Btree::find(Transaction& t, const ghobject_t& obj) {
- return seastar::do_with(
- full_key_t<KeyT::HOBJ>(obj),
- [this, &t](auto& key) -> btree_future<Cursor> {
- return get_root(t).safe_then([this, &t, &key](auto root) {
- // TODO: improve lower_bound()
- return root->lower_bound(get_context(t), key);
- }).safe_then([this](auto result) {
- if (result.match() == MatchKindBS::EQ) {
- return Cursor(this, result.p_cursor);
- } else {
- return Cursor::make_end(this);
- }
- });
- }
- );
-}
-
-btree_future<Cursor>
-Btree::lower_bound(Transaction& t, const ghobject_t& obj) {
- return seastar::do_with(
- full_key_t<KeyT::HOBJ>(obj),
- [this, &t](auto& key) -> btree_future<Cursor> {
- return get_root(t).safe_then([this, &t, &key](auto root) {
- return root->lower_bound(get_context(t), key);
- }).safe_then([this](auto result) {
- return Cursor(this, result.p_cursor);
- });
- }
- );
-}
-
-btree_future<std::pair<Cursor, bool>>
-Btree::insert(Transaction& t, const ghobject_t& obj, const onode_t& value) {
- return seastar::do_with(
- full_key_t<KeyT::HOBJ>(obj),
- [this, &t, &value](auto& key) -> btree_future<std::pair<Cursor, bool>> {
- return get_root(t).safe_then([this, &t, &key, &value](auto root) {
- return root->insert(get_context(t), key, value);
- }).safe_then([this](auto ret) {
- auto& [cursor, success] = ret;
- return std::make_pair(Cursor(this, cursor), success);
- });
- }
- );
-}
-
-btree_future<size_t> Btree::erase(Transaction& t, const ghobject_t& obj) {
- // TODO
- return btree_ertr::make_ready_future<size_t>(0u);
-}
-
-btree_future<Cursor> Btree::erase(Cursor& pos) {
- // TODO
- return btree_ertr::make_ready_future<Cursor>(
- Cursor::make_end(this));
-}
-
-btree_future<Cursor>
-Btree::erase(Cursor& first, Cursor& last) {
- // TODO
- return btree_ertr::make_ready_future<Cursor>(
- Cursor::make_end(this));
-}
-
-btree_future<size_t> Btree::height(Transaction& t) {
- return get_root(t).safe_then([](auto root) {
- return size_t(root->level() + 1);
- });
-}
-
-btree_future<tree_stats_t> Btree::get_stats_slow(Transaction& t) {
- return get_root(t).safe_then([this, &t](auto root) {
- unsigned height = root->level() + 1;
- return root->get_tree_stats(get_context(t)
- ).safe_then([height](auto stats) {
- stats.height = height;
- return btree_ertr::make_ready_future<tree_stats_t>(stats);
- });
- });
-}
-
-std::ostream& Btree::dump(Transaction& t, std::ostream& os) {
- auto root = root_tracker->get_root(t);
- if (root) {
- root->dump(os);
- } else {
- os << "empty tree!";
- }
- return os;
-}
-
-std::ostream& Btree::print(std::ostream& os) const {
- return os << "BTree-" << *nm;
-}
-
-btree_future<Ref<Node>> Btree::get_root(Transaction& t) {
- auto root = root_tracker->get_root(t);
- if (root) {
- return btree_ertr::make_ready_future<Ref<Node>>(root);
- } else {
- return Node::load_root(get_context(t), *root_tracker);
- }
-}
-
-bool Btree::test_is_clean() const {
- return root_tracker->is_clean();
-}
-
-btree_future<> Btree::test_clone_from(
- Transaction& t, Transaction& t_from, Btree& from) {
- // Note: assume the tree to clone is tracked correctly in memory.
- // In some unit tests, parts of the tree are stubbed out that they
- // should not be loaded from NodeExtentManager.
- return from.get_root(t_from
- ).safe_then([this, &t](auto root_from) {
- return root_from->test_clone_root(get_context(t), *root_tracker);
- });
-}
-
-}