167 lines
6.1 KiB
C++
167 lines
6.1 KiB
C++
/**
|
|
* Copyright (c) 2023 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
#define USING_LOG_PREFIX STORAGE
|
|
#include <gtest/gtest.h>
|
|
#include <gmock/gmock.h>
|
|
#define private public
|
|
#define protected public
|
|
#include <string.h>
|
|
#include "storage/ls/ob_ls_reserved_snapshot_mgr.h"
|
|
#include "mittest/mtlenv/mock_tenant_module_env.h"
|
|
#include "unittest/storage/test_dml_common.h"
|
|
namespace oceanbase
|
|
{
|
|
using namespace common;
|
|
using namespace compaction;
|
|
using namespace storage;
|
|
|
|
namespace storage
|
|
{
|
|
int ObLSReservedSnapshotMgr::sync_clog(const int64_t new_reserved_snapshot)
|
|
{
|
|
LOG_INFO("mock sync clog", K(new_reserved_snapshot));
|
|
return OB_SUCCESS;
|
|
}
|
|
}
|
|
|
|
namespace unittest
|
|
{
|
|
|
|
class TestLSReservedSnapshotMgr : public ::testing::Test
|
|
{
|
|
public:
|
|
TestLSReservedSnapshotMgr()
|
|
: ls_id_(TEST_LS_ID)
|
|
{}
|
|
~TestLSReservedSnapshotMgr() = default;
|
|
static void SetUpTestCase();
|
|
static void TearDownTestCase();
|
|
static constexpr int64_t TEST_TENANT_ID = 1;
|
|
static constexpr int64_t TEST_LS_ID = 9001;
|
|
share::ObLSID ls_id_;
|
|
};
|
|
|
|
void TestLSReservedSnapshotMgr::SetUpTestCase()
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ret = MockTenantModuleEnv::get_instance().init();
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
|
|
// ls service cannot service before ObServerCheckpointSlogHandler starts
|
|
// running
|
|
ObServerCheckpointSlogHandler::get_instance().is_started_ = true;
|
|
// create ls
|
|
ObLSHandle ls_handle;
|
|
ret = TestDmlCommon::create_ls(TEST_TENANT_ID, ObLSID(TEST_LS_ID), ls_handle);
|
|
if (OB_SUCCESS != ret) {
|
|
LOG_ERROR("[FATAL ERROR] failed to create ls", K(ret));
|
|
}
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
}
|
|
|
|
void TestLSReservedSnapshotMgr::TearDownTestCase()
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ret = MTL(ObLSService*)->remove_ls(ObLSID(TEST_LS_ID));
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
|
|
MockTenantModuleEnv::get_instance().destroy();
|
|
}
|
|
|
|
TEST_F(TestLSReservedSnapshotMgr, test_add_del_dependent_tablet)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObLSHandle ls_handle;
|
|
ObLSService *ls_svr = MTL(ObLSService*);
|
|
ret = ls_svr->get_ls(ls_id_, ls_handle, ObLSGetMod::STORAGE_MOD);
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
|
|
ObLSReservedSnapshotMgr &reserved_snapshot_mgr = ls_handle.get_ls()->reserved_snapshot_mgr_;
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.try_sync_reserved_snapshot(100, true/*update_flag*/));
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
|
|
const ObTabletID tablet_id(100);
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.add_dependent_medium_tablet(tablet_id));
|
|
ASSERT_EQ(OB_ENTRY_EXIST, reserved_snapshot_mgr.add_dependent_medium_tablet(tablet_id));
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.del_dependent_medium_tablet(tablet_id));
|
|
ASSERT_EQ(OB_HASH_NOT_EXIST, reserved_snapshot_mgr.del_dependent_medium_tablet(tablet_id));
|
|
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
|
|
const ObTabletID tablet_id2(200);
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.add_dependent_medium_tablet(tablet_id));
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.add_dependent_medium_tablet(tablet_id2));
|
|
|
|
// have dependent tablet, can not push min_reserved_snapshot_
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.try_sync_reserved_snapshot(200, true/*update_flag*/));
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
ASSERT_EQ(200, reserved_snapshot_mgr.next_reserved_snapshot_);
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.try_sync_reserved_snapshot(300, true/*update_flag*/));
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
ASSERT_EQ(300, reserved_snapshot_mgr.next_reserved_snapshot_);
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.del_dependent_medium_tablet(tablet_id));
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
ASSERT_EQ(300, reserved_snapshot_mgr.next_reserved_snapshot_);
|
|
// no dependent tablet, push min_reserved_snapshot_
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.del_dependent_medium_tablet(tablet_id2));
|
|
ASSERT_EQ(300, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
ASSERT_EQ(0, reserved_snapshot_mgr.next_reserved_snapshot_);
|
|
|
|
reserved_snapshot_mgr.min_reserved_snapshot_ = 0;
|
|
reserved_snapshot_mgr.next_reserved_snapshot_ = 0;
|
|
}
|
|
|
|
TEST_F(TestLSReservedSnapshotMgr, test_update_with_smaller_snapshot)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObLSHandle ls_handle;
|
|
ObLSService *ls_svr = MTL(ObLSService*);
|
|
ret = ls_svr->get_ls(ls_id_, ls_handle, ObLSGetMod::STORAGE_MOD);
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
|
|
ObLSReservedSnapshotMgr &reserved_snapshot_mgr = ls_handle.get_ls()->reserved_snapshot_mgr_;
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.try_sync_reserved_snapshot(100, true/*update_flag*/));
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
|
|
// can't update with smaller snapshot
|
|
ASSERT_EQ(OB_SNAPSHOT_DISCARDED, reserved_snapshot_mgr.try_sync_reserved_snapshot(50, true/*update_flag*/));
|
|
|
|
const ObTabletID tablet_id(100);
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.add_dependent_medium_tablet(tablet_id));
|
|
ASSERT_EQ(OB_SNAPSHOT_DISCARDED, reserved_snapshot_mgr.try_sync_reserved_snapshot(50, true/*update_flag*/));
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.del_dependent_medium_tablet(tablet_id));
|
|
ASSERT_EQ(OB_SNAPSHOT_DISCARDED, reserved_snapshot_mgr.try_sync_reserved_snapshot(50, true/*update_flag*/));
|
|
|
|
ASSERT_EQ(100, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
|
|
ASSERT_EQ(OB_SUCCESS, reserved_snapshot_mgr.try_sync_reserved_snapshot(150, true/*update_flag*/));
|
|
ASSERT_EQ(150, reserved_snapshot_mgr.min_reserved_snapshot_);
|
|
}
|
|
|
|
}//end namespace unittest
|
|
}//end namespace oceanbase
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
system("rm -f test_ls_reserved_snapshot.log*");
|
|
OB_LOGGER.set_file_name("test_ls_reserved_snapshot.log");
|
|
OB_LOGGER.set_log_level("DEBUG");
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|