120 lines
2.8 KiB
C++
120 lines
2.8 KiB
C++
#ifndef MAPPING_TREE_H
|
|
#define MAPPING_TREE_H
|
|
|
|
#include "persistent-data/data-structures/btree.h"
|
|
#include "persistent-data/run.h"
|
|
|
|
//----------------------------------------------------------------
|
|
|
|
namespace thin_provisioning {
|
|
namespace mapping_tree_detail {
|
|
class space_map_ref_counter {
|
|
public:
|
|
space_map_ref_counter(space_map::ptr sm);
|
|
|
|
void inc(block_address b);
|
|
void dec(block_address b);
|
|
|
|
private:
|
|
space_map::ptr sm_;
|
|
};
|
|
|
|
struct block_time {
|
|
uint64_t block_;
|
|
uint32_t time_;
|
|
};
|
|
|
|
class block_time_ref_counter {
|
|
public:
|
|
block_time_ref_counter(space_map::ptr sm);
|
|
void inc(block_time bt);
|
|
void dec(block_time bt);
|
|
|
|
private:
|
|
space_map::ptr sm_;
|
|
};
|
|
|
|
struct block_traits {
|
|
typedef base::le64 disk_type;
|
|
typedef block_time value_type;
|
|
typedef block_time_ref_counter ref_counter;
|
|
|
|
static void unpack(disk_type const &disk, value_type &value);
|
|
static void pack(value_type const &value, disk_type &disk);
|
|
};
|
|
|
|
class mtree_ref_counter {
|
|
public:
|
|
mtree_ref_counter(transaction_manager::ptr tm);
|
|
|
|
void inc(block_address b);
|
|
void dec(block_address b);
|
|
|
|
private:
|
|
transaction_manager::ptr tm_;
|
|
};
|
|
|
|
struct mtree_traits {
|
|
typedef base::le64 disk_type;
|
|
typedef uint64_t value_type;
|
|
typedef mtree_ref_counter ref_counter;
|
|
|
|
static void unpack(disk_type const &disk, value_type &value);
|
|
static void pack(value_type const &value, disk_type &disk);
|
|
};
|
|
|
|
//--------------------------------
|
|
|
|
class damage_visitor;
|
|
|
|
struct damage {
|
|
virtual ~damage() {}
|
|
virtual void visit(damage_visitor &v) const = 0;
|
|
};
|
|
|
|
struct missing_devices : public damage {
|
|
missing_devices(std::string const &desc, run<uint64_t> const &keys);
|
|
virtual void visit(damage_visitor &v) const;
|
|
|
|
std::string desc_;
|
|
run<uint64_t> keys_;
|
|
};
|
|
|
|
struct missing_mappings : public damage {
|
|
missing_mappings(std::string const &desc, uint64_t thin_dev,
|
|
run<uint64_t> const &keys);
|
|
virtual void visit(damage_visitor &v) const;
|
|
|
|
std::string desc_;
|
|
uint64_t thin_dev_;
|
|
run<uint64_t> keys_;
|
|
};
|
|
|
|
class damage_visitor {
|
|
public:
|
|
virtual ~damage_visitor() {}
|
|
|
|
void visit(damage const &d) {
|
|
d.visit(*this);
|
|
}
|
|
|
|
virtual void visit(missing_devices const &d) = 0;
|
|
virtual void visit(missing_mappings const &d) = 0;
|
|
};
|
|
|
|
}
|
|
|
|
typedef persistent_data::btree<2, mapping_tree_detail::block_traits> mapping_tree;
|
|
typedef persistent_data::btree<1, mapping_tree_detail::mtree_traits> dev_tree;
|
|
typedef persistent_data::btree<1, mapping_tree_detail::block_traits> single_mapping_tree;
|
|
|
|
void check_mapping_tree(dev_tree const &tree,
|
|
mapping_tree_detail::damage_visitor &visitor);
|
|
void check_mapping_tree(mapping_tree const &tree,
|
|
mapping_tree_detail::damage_visitor &visitor);
|
|
}
|
|
|
|
//----------------------------------------------------------------
|
|
|
|
#endif
|