TEST(ObMultiVersionTabletImage, test_write_null)
{
FileInfoCache fic;
ObMultiVersionTabletImage image(fic);
fic.init(100);
CharArena allocator;
ObRange r1;
ObBorderFlag flag;
flag.set_min_value();
flag.set_max_value();
create_range(allocator, r1, 1, flag.get_data(), "", "");
int ret = 0;
ObTablet* tablet = NULL;
ObSSTableId id;
id.sstable_file_id_ = 1;
id.sstable_file_offset_ = 0;
ret = image.alloc_tablet_object(r1, VERSION_1, tablet);
ASSERT_EQ(0, ret);
tablet->set_disk_no(1);
ret = tablet->add_sstable_by_id(id);
ASSERT_EQ(0, ret);
ret = image.add_tablet(tablet, false);
ASSERT_EQ(0, ret);
image.write(VERSION_1, 1);
fic.destroy();
}
TEST(ObMultiVersionTabletImage, test_service)
{
FileInfoCache fic;
ObMultiVersionTabletImage image(fic);
fic.init(100);
CharArena allocator;
ObRange r1,r2,r3;
create_range(allocator, r1, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "aoo", "foo");
create_range(allocator, r2, 1, ObBorderFlag::INCLUSIVE_END, "foo", "mj");
create_range(allocator, r3, 1, ObBorderFlag::INCLUSIVE_END, "mj", "oi");
int ret = 0;
ObTablet* tablet = NULL;
ObSSTableId id;
id.sstable_file_id_ = 1;
id.sstable_file_offset_ = 0;
ret = image.alloc_tablet_object(r1, VERSION_1, tablet);
ASSERT_EQ(0, ret);
tablet->set_disk_no(1);
ret = tablet->add_sstable_by_id(id);
ASSERT_EQ(0, ret);
ret = image.add_tablet(tablet, false);
ASSERT_EQ(0, ret);
//image.initialize_service_index();
/*
ret = image.alloc_tablet_object(r1, VERSION_2, tablet);
ASSERT_EQ(0, ret);
tablet->set_disk_no(1);
ret = tablet->add_sstable_by_id(id);
ASSERT_EQ(0, ret);
ret = image.add_tablet(tablet, false);
ASSERT_EQ(0, ret);
ret = image.alloc_tablet_object(r2, VERSION_2, tablet);
ASSERT_EQ(0, ret);
id.sstable_file_id_ = 2;
id.sstable_file_offset_ = 0;
tablet->set_disk_no(2);
tablet->add_sstable_by_id(id);
ret = image.add_tablet(tablet, false);
ASSERT_EQ(0, ret);
ret = image.alloc_tablet_object(r3, VERSION_2, tablet);
ASSERT_EQ(0, ret);
id.sstable_file_id_ = 3;
id.sstable_file_offset_ = 0;
tablet->set_disk_no(3);
tablet->add_sstable_by_id(id);
ret = image.add_tablet(tablet, false);
ASSERT_EQ(0, ret);
ret = write_all(image);
ASSERT_EQ(0, ret);
*/
fic.destroy();
}
TEST(ObMultiVersionTabletImage, test_remove)
{
FileInfoCache fic;
ObMultiVersionTabletImage image(fic);
fic.init(100);
CharArena allocator;
ObRange r1,r2,r3;
create_range(allocator, r1, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "aoo", "foo");
create_range(allocator, r2, 1, ObBorderFlag::INCLUSIVE_END, "foo", "mj");
create_range(allocator, r3, 1, ObBorderFlag::INCLUSIVE_END, "mj", "oi");
int ret = read_all(image);
ASSERT_EQ(0, ret);
int disk_no = 0;
ObTablet *tablet= NULL;
ret = image.remove_tablet(r1, VERSION_2, disk_no);
ASSERT_EQ(0, ret);
ASSERT_EQ(1, disk_no);
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, VERSION_2, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ASSERT_EQ(VERSION_1, tablet->get_data_version());
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
fic.destroy();
}
TEST(ObMultiVersionTabletImage, test_query_min_max)
{
FileInfoCache fic;
ObMultiVersionTabletImage image(fic);
fic.init(100);
CharArena allocator;
ObRange r1,r2,r3;
create_range(allocator, r1, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "aoo", "foo");
create_range(allocator, r2, 1, ObBorderFlag::INCLUSIVE_END, "foo", "mj");
create_range(allocator, r3, 1, ObBorderFlag::INCLUSIVE_END, "mj", "oi");
ObRange rall;
ObBorderFlag flag;
flag.set_min_value();
flag.set_max_value();
create_range(allocator, rall, 1, flag.get_data(), "aoo", "zoo");
int ret = 0;
ObTablet* tablet = NULL;
ObSSTableId id;
id.sstable_file_id_ = 1;
id.sstable_file_offset_ = 0;
ret = image.alloc_tablet_object(rall, VERSION_1, tablet);
ASSERT_EQ(0, ret);
tablet->set_disk_no(1);
ret = tablet->add_sstable_by_id(id);
ASSERT_EQ(0, ret);
ret = image.add_tablet(tablet, false, true);
ASSERT_EQ(0, ret);
image.dump();
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_1, tablet->get_data_version());
image.release_tablet(tablet);
ret = image.acquire_tablet(rall, ObMultiVersionTabletImage::SCAN_FORWARD, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_1, tablet->get_data_version());
tablet->get_range().hex_dump();
image.release_tablet(tablet);
fic.destroy();
}
TEST(ObMultiVersionTabletImage, test_upgrade)
{
FileInfoCache fic;
ObMultiVersionTabletImage image(fic);
fic.init(100);
CharArena allocator;
ObRange r1,r2,r3;
create_range(allocator, r1, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "aoo", "foo");
create_range(allocator, r2, 1, ObBorderFlag::INCLUSIVE_END, "foo", "mj");
create_range(allocator, r3, 1, ObBorderFlag::INCLUSIVE_END, "mj", "oi");
int ret = read_all(image);
ASSERT_EQ(0, ret);
int disk_no = 0;
ObTablet *tablet= NULL;
ret = image.remove_tablet(r1, VERSION_2, disk_no);
ASSERT_EQ(0, ret);
ASSERT_EQ(1, disk_no);
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, VERSION_1, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ASSERT_EQ(VERSION_1, tablet->get_data_version());
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
// get tablet version1 r1 for merge
int num = 2;
ObTablet *tablets[num];
ret = image.get_tablets_for_merge(VERSION_2, num, tablets);
ASSERT_EQ(0, ret);
ASSERT_EQ(1, num);
tablet = tablets[0];
ASSERT_EQ(VERSION_1, tablet->get_data_version());
ASSERT_EQ(true, tablet->get_range().equal(r1));
ret = image.release_tablet(tablet);
// upgrade r1 v1 to r1 v2
ObTablet *new_tablet = NULL;
ret = image.alloc_tablet_object(r1, VERSION_2, new_tablet);
ASSERT_EQ(0, ret);
new_tablet->set_disk_no(2);
ObSSTableId id;
id.sstable_file_id_ = 15 << 8 | 2;
id.sstable_file_offset_ = 0;
ret = new_tablet->add_sstable_by_id(id);
ASSERT_EQ(0, ret);
ret = image.upgrade_tablet(tablet, new_tablet, false);
ASSERT_EQ(0, ret);
ret = image.upgrade_service();
ASSERT_EQ(0, ret);
// query r1 v2
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ASSERT_EQ(VERSION_2, tablet->get_data_version());
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
// query tablets for merge, should be null
ret = image.get_tablets_for_merge(VERSION_2, num, tablets);
ASSERT_EQ(0, ret);
ASSERT_EQ(0, num);
// query tablets v2 for merge to v3
num = 2;
ret = image.get_tablets_for_merge(VERSION_3, num, tablets);
ASSERT_EQ(0, ret);
ASSERT_EQ(2, num);
tablet = tablets[0];
ret = image.alloc_tablet_object(tablet->get_range(), VERSION_3, new_tablet);
ASSERT_EQ(0, ret);
new_tablet->set_disk_no(10);
id.sstable_file_id_ = 15 << 8 | 10;
id.sstable_file_offset_ = 0;
ret = new_tablet->add_sstable_by_id(id);
ASSERT_EQ(0, ret);
ret = image.upgrade_tablet(tablet, new_tablet, false);
ASSERT_EQ(0, ret);
for (int i = 0; i < num; ++i)
{
image.release_tablet(tablets[i]);
}
ret = image.upgrade_service();
// query v3
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_3, tablet->get_data_version());
image.release_tablet(tablet);
fic.destroy();
}
TEST(ObMultiVersionTabletImage, test_query)
{
FileInfoCache fic;
ObMultiVersionTabletImage image(fic);
fic.init(100);
CharArena allocator;
ObRange r1,r2,r3;
create_range(allocator, r1, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "aoo", "foo");
create_range(allocator, r2, 1, ObBorderFlag::INCLUSIVE_END, "foo", "mj");
create_range(allocator, r3, 1, ObBorderFlag::INCLUSIVE_END, "mj", "oi");
int ret = read_all(image);
ASSERT_EQ(0, ret);
image.dump();
ObTablet *tablet = NULL;
ret = image.acquire_tablet(r2, ObMultiVersionTabletImage::SCAN_FORWARD, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r2));
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_2, tablet->get_data_version());
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_1, tablet->get_data_version());
ret = image.acquire_tablet(r1, ObMultiVersionTabletImage::SCAN_FORWARD, VERSION_1, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_1, tablet->get_data_version());
ObRange query_whole_range;
query_whole_range.table_id_ = 1;
query_whole_range.border_flag_.set_min_value();
query_whole_range.border_flag_.set_max_value();
ret = image.acquire_tablet(query_whole_range, ObMultiVersionTabletImage::SCAN_FORWARD, VERSION_1, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(VERSION_1, tablet->get_data_version());
ret = image.acquire_tablet_all_version(query_whole_range, ObMultiVersionTabletImage::SCAN_BACKWARD,
ObMultiVersionTabletImage::FROM_NEWEST_INDEX, VERSION_2, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r3));
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
ret = image.acquire_tablet(query_whole_range, ObMultiVersionTabletImage::SCAN_BACKWARD, VERSION_1, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
ObRange r4;
create_range(allocator, r4, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "foo", "koo");
ret = image.acquire_tablet_all_version(r4, ObMultiVersionTabletImage::SCAN_FORWARD,
ObMultiVersionTabletImage::FROM_NEWEST_INDEX, 0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r1));
ASSERT_EQ(VERSION_2, tablet->get_data_version());
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
create_range(allocator, r4, 1, ObBorderFlag::INCLUSIVE_START|ObBorderFlag::INCLUSIVE_END, "foz", "noo");
ret = image.acquire_tablet(r4, ObMultiVersionTabletImage::SCAN_FORWARD,0, tablet);
ASSERT_EQ(0, ret);
ASSERT_EQ(true, tablet->get_range().equal(r2));
ASSERT_EQ(VERSION_2, tablet->get_data_version());
ret = image.release_tablet(tablet);
ASSERT_EQ(0, ret);
fic.destroy();
}
static void TearDownTestCase()
{
fic.destroy();
}
