320 lines
11 KiB
C++
320 lines
11 KiB
C++
/**
|
|
* Copyright (c) 2021 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 SHARE
|
|
#define private public
|
|
#include "lib/restore/ob_storage.h"
|
|
#include "lib/restore/ob_storage_oss_base.cpp"
|
|
#undef private
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "lib/utility/ob_test_util.h"
|
|
#include "share/backup/ob_backup_io_adapter.h"
|
|
#include "share/backup/ob_backup_struct.h"
|
|
#define private public
|
|
#include "share/backup/ob_archive_checkpoint_mgr.h"
|
|
#undef private
|
|
#include "test_ob_backup_dest_config.h"
|
|
#include "lib/allocator/page_arena.h"
|
|
#include "lib/oblog/ob_log.h"
|
|
#include "lib/oblog/ob_log_module.h"
|
|
#include "share/ob_force_print_log.h"
|
|
#include "share/backup/ob_backup_path.h"
|
|
#include "share/ob_thread_pool.h"
|
|
#include "apr_allocator.h"
|
|
|
|
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::share;
|
|
|
|
const ObStorageType TEST_STORAGE_TYPES[] = {
|
|
OB_STORAGE_FILE,
|
|
OB_STORAGE_OSS,
|
|
};
|
|
|
|
class TestArchiveCheckpointMgr: public ::testing::Test
|
|
{
|
|
public:
|
|
TestArchiveCheckpointMgr() {}
|
|
virtual ~TestArchiveCheckpointMgr() {}
|
|
virtual void SetUp()
|
|
{
|
|
}
|
|
virtual void TearDown()
|
|
{
|
|
}
|
|
|
|
static void SetUpTestCase()
|
|
{
|
|
}
|
|
|
|
static void TearDownTestCase()
|
|
{
|
|
}
|
|
|
|
// function members
|
|
static int get_root_path(const ObStorageType &type, ObBackupPath &root_path);
|
|
static int get_storage_info(const ObStorageType &type, ObString &storage_info);
|
|
int clean_root_dir(const ObStorageType &type);
|
|
int clean_dir(const ObStorageType &type, const ObString &dir_uri);
|
|
int generate_simple_files(const ObStorageType &type);
|
|
|
|
// test cases
|
|
typedef int (TestArchiveCheckpointMgr::*TEST_FUNCTION) (const ObStorageType &type);
|
|
int run_test_func(TEST_FUNCTION function);
|
|
int test_write_and_read_checkpoint(const ObStorageType &type);
|
|
|
|
ObBackupStorageInfo storage_info_;
|
|
|
|
private:
|
|
// disallow copy
|
|
DISALLOW_COPY_AND_ASSIGN(TestArchiveCheckpointMgr);
|
|
};
|
|
|
|
int TestArchiveCheckpointMgr::run_test_func(TEST_FUNCTION f)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const int64_t type_count = sizeof(TEST_STORAGE_TYPES) / sizeof(ObStorageType);
|
|
ObBackupPath path;
|
|
ObString storage_info;
|
|
for (int64_t i = 0; OB_SUCC(ret) && i < type_count; ++i) {
|
|
const ObStorageType type = TEST_STORAGE_TYPES[i];
|
|
const int64_t start_ts = ObTimeUtility::current_time();
|
|
storage_info_.reset();
|
|
FLOG_INFO("start run_test_func", K(type));
|
|
if (OB_FAIL(get_root_path(type, path))) {
|
|
LOG_WARN("failed to get root patch", K(ret), K(type));
|
|
} else if (path.is_empty()) {
|
|
LOG_INFO("path is not set, skip unittest", K(type));
|
|
} else if (OB_FAIL(get_storage_info(type, storage_info))) {
|
|
LOG_WARN("failed to get storage info", K(ret), K(type));
|
|
} else if (OB_FAIL(storage_info_.set(type, storage_info.ptr()))) {
|
|
LOG_INFO("path is not set, skip unittest", K(type));
|
|
} else if (OB_FAIL(clean_root_dir(type))) {
|
|
LOG_WARN("failed to clean root dir", K(ret), K(type));
|
|
} else if (OB_FAIL((this->*f)(type))) {
|
|
LOG_WARN("failed to to test function", K(ret), K(type));
|
|
} else if (OB_FAIL(clean_root_dir(type))) {
|
|
LOG_WARN("failed to clean root dir", K(ret), K(type));
|
|
}
|
|
const int64_t cost_ts = ObTimeUtility::current_time() - start_ts;
|
|
FLOG_INFO("end run_test_func", K(ret), K(type), K(cost_ts));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int TestArchiveCheckpointMgr::get_root_path(const ObStorageType &type, ObBackupPath &root_path)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
root_path.reset();
|
|
|
|
if (OB_STORAGE_OSS == type) {
|
|
if (OB_FAIL(root_path.init(oss_root_path))) {
|
|
LOG_WARN("failed to init oss root path", K(ret));
|
|
}
|
|
} else if (OB_STORAGE_FILE == type) {
|
|
if (OB_FAIL(root_path.init(file_root_path))) {
|
|
LOG_WARN("failed to init file root path", K(ret));
|
|
}
|
|
} else {
|
|
ret = OB_ERROR;
|
|
LOG_ERROR("unknown type", K(type));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int TestArchiveCheckpointMgr::get_storage_info(const ObStorageType &type, ObString &storage_info)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
storage_info.reset();
|
|
if (OB_STORAGE_OSS == type) {
|
|
storage_info = ObString(oss_storage_info);
|
|
} else if (OB_STORAGE_FILE == type) {
|
|
storage_info = ObString(file_storage_info);
|
|
} else {
|
|
ret = OB_ERROR;
|
|
LOG_ERROR("unknown type", K(type));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int TestArchiveCheckpointMgr::generate_simple_files(const ObStorageType &type)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObBackupPath root_path;
|
|
ObBackupPath tmp_root_path;
|
|
ObBackupIoAdapter util;
|
|
char buf[] = "test";
|
|
int64_t start_ts = ObTimeUtility::current_time();
|
|
|
|
LOG_INFO("start generate_simple_files", K(type));
|
|
if (OB_FAIL(get_root_path(type, root_path))) {
|
|
LOG_WARN("failed to get root path", K(ret), K(root_path));
|
|
}
|
|
|
|
//├── test_checkpoint
|
|
//│ ├── file1
|
|
//│ ├── file2
|
|
//│ ├── file3
|
|
|
|
for (int64_t i = 1; OB_SUCC(ret) && i <= 4; ++i) {
|
|
tmp_root_path.reset();
|
|
tmp_root_path = root_path;
|
|
if (OB_FAIL(tmp_root_path.join_checkpoint_info_file(OB_STR_CHECKPOINT_FILE_NAME, i, ObBackupFileSuffix::ARCHIVE))) {
|
|
LOG_WARN("failed to join path", K(ret));
|
|
} else if (OB_FAIL(util.mk_parent_dir(tmp_root_path.get_ptr(), &storage_info_))) {
|
|
LOG_WARN("failed to mk dir", K(ret));
|
|
} else if (OB_FAIL(util.write_single_file(tmp_root_path.get_ptr(), &storage_info_, buf, sizeof(buf),
|
|
ObStorageIdMod::get_default_archive_id_mod()))) {
|
|
LOG_WARN("failed to write dir less", K(ret));
|
|
}
|
|
}
|
|
|
|
const int64_t cost_ts = ObTimeUtility::current_time() - start_ts;
|
|
LOG_INFO("finish prepare simple dir", K(type), K(cost_ts), K(ret));
|
|
return ret;
|
|
}
|
|
|
|
int TestArchiveCheckpointMgr::clean_root_dir(const ObStorageType &type)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObBackupPath root_path;
|
|
if (OB_FAIL(get_root_path(type, root_path))) {
|
|
LOG_WARN("failed to get root path", K(ret), K(root_path));
|
|
} else if (OB_FAIL(clean_dir(type, root_path.get_ptr()))) {
|
|
LOG_WARN("failed to clean dir", K(ret), K(type), K(root_path));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int TestArchiveCheckpointMgr::clean_dir(const ObStorageType &type, const ObString &dir_uri)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
ObBackupIoAdapter util;
|
|
char path[OB_MAX_BACKUP_PATH_LENGTH] = "";
|
|
ObArray <ObString> file_names;
|
|
ObArenaAllocator allocator;
|
|
char uri_del[OB_MAX_URI_LENGTH];
|
|
ObFileListArrayOp op(file_names, allocator);
|
|
|
|
if (OB_FAIL(util.list_files(dir_uri, &storage_info_, op))) {
|
|
LOG_WARN("failed to list files", K(ret), K(dir_uri));
|
|
} else {
|
|
for (int64_t i = 0; OB_SUCC(ret) && i < file_names.count(); ++i) {
|
|
const ObString file = file_names.at(i);
|
|
if (OB_FAIL(databuff_printf(path, sizeof(path), "%.*s/%.*s",
|
|
dir_uri.length(), dir_uri.ptr(), file.length(), file.ptr()))) {
|
|
LOG_WARN("failed to join path", K(ret));
|
|
} else if (OB_FAIL(util.del_file(path, &storage_info_))) {
|
|
LOG_WARN("failed to clean dir", K(ret), K(type), K(path));
|
|
}
|
|
LOG_INFO("del_file", K(path));
|
|
}
|
|
}
|
|
|
|
if (OB_SUCC(ret)) {
|
|
if (OB_FAIL(util.del_dir(dir_uri, &storage_info_))) {
|
|
LOG_WARN("failed to del dir", K(ret), K(dir_uri));
|
|
}
|
|
LOG_INFO("del_dir", K(dir_uri));
|
|
}
|
|
|
|
return ret;
|
|
};
|
|
|
|
int TestArchiveCheckpointMgr::test_write_and_read_checkpoint(const ObStorageType &type)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
int64_t start_ts = ObTimeUtility::current_time();
|
|
ObBackupPath root_path;
|
|
ObBackupPath tmp_root_path;
|
|
ObBackupIoAdapter util;
|
|
if (OB_FAIL(get_root_path(type, root_path))) {
|
|
LOG_WARN("failed to get root path", K(ret), K(root_path));
|
|
} else {
|
|
ObArchiveCheckpointMgr mgr;
|
|
bool is_exist = true;
|
|
uint64_t checkpoint = 0;
|
|
if (OB_FAIL(mgr.init(root_path, OB_STR_CHECKPOINT_FILE_NAME, ObBackupFileSuffix::ARCHIVE, &storage_info_))) {
|
|
LOG_WARN("failed to init checkpoint mgr", K(ret), K(root_path));
|
|
} else if (OB_FAIL(mgr.write(10, 9))) {
|
|
LOG_WARN("failed to write files", K(ret), K(root_path));
|
|
} else if (OB_FAIL(mgr.write(9, 8))) {
|
|
LOG_WARN("failed to write files", K(ret), K(root_path));
|
|
} else if (OB_FAIL(mgr.read(checkpoint))) {
|
|
LOG_WARN("failed to read files", K(ret));
|
|
} else if (checkpoint != 10) {
|
|
ret = OB_ERROR;
|
|
} else if (OB_FAIL(mgr.write(100, 99))) {
|
|
LOG_WARN("failed to write files", K(ret), K(root_path));
|
|
} else if (OB_FAIL(mgr.write(50, 49))) {
|
|
LOG_WARN("failed to write files", K(ret), K(root_path));
|
|
} else if (OB_FAIL(mgr.read(checkpoint))) {
|
|
LOG_WARN("failed to read files", K(ret));
|
|
} else if (checkpoint != 100) {
|
|
ret = OB_ERROR;
|
|
} else {
|
|
tmp_root_path.reset();
|
|
tmp_root_path = root_path;
|
|
if (OB_FAIL(tmp_root_path.join_checkpoint_info_file(OB_STR_CHECKPOINT_FILE_NAME, 10, ObBackupFileSuffix::ARCHIVE))) {
|
|
LOG_WARN("failed to join path", K(ret));
|
|
} else if (OB_FAIL(util.is_exist(tmp_root_path.get_ptr(), &storage_info_, is_exist))) {
|
|
LOG_WARN("failed to judge file exist", K(ret), K(tmp_root_path));
|
|
} else if (is_exist) {
|
|
ret = OB_ERROR;
|
|
}
|
|
tmp_root_path.reset();
|
|
tmp_root_path = root_path;
|
|
if (OB_SUCC(ret)) {
|
|
if (OB_FAIL(tmp_root_path.join_checkpoint_info_file(OB_STR_CHECKPOINT_FILE_NAME, 9, ObBackupFileSuffix::ARCHIVE))) {
|
|
LOG_WARN("failed to join path", K(ret));
|
|
} else if (OB_FAIL(util.is_exist(tmp_root_path.get_ptr(), &storage_info_, is_exist))) {
|
|
LOG_WARN("failed to judge file exist", K(ret), K(tmp_root_path));
|
|
} else if (is_exist) {
|
|
ret = OB_ERROR;
|
|
}
|
|
}
|
|
tmp_root_path.reset();
|
|
tmp_root_path = root_path;
|
|
if (OB_SUCC(ret)) {
|
|
if (OB_FAIL(tmp_root_path.join_checkpoint_info_file(OB_STR_CHECKPOINT_FILE_NAME, 50, ObBackupFileSuffix::ARCHIVE))) {
|
|
LOG_WARN("failed to join path", K(ret));
|
|
} else if (OB_FAIL(util.is_exist(tmp_root_path.get_ptr(), &storage_info_, is_exist))) {
|
|
LOG_WARN("failed to judge file exist", K(ret), K(tmp_root_path));
|
|
} else if (!is_exist) {
|
|
ret = OB_ERROR;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
const int64_t cost_ts = ObTimeUtility::current_time() - start_ts;
|
|
LOG_INFO("finish test_list_util", K(type), K(cost_ts), K(ret));
|
|
return ret;
|
|
}
|
|
|
|
TEST_F(TestArchiveCheckpointMgr, test_write_read_op)
|
|
{
|
|
TEST_FUNCTION f = &TestArchiveCheckpointMgr::test_write_and_read_checkpoint;
|
|
int ret = run_test_func(f);
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
system("rm -f test_archive_checkpoint_mgr.log");
|
|
OB_LOGGER.set_file_name("test_archive_checkpoint_mgr.log");
|
|
OB_LOGGER.set_log_level("INFO");
|
|
::testing::InitGoogleTest(&argc,argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|