101 lines
3.5 KiB
C++
101 lines
3.5 KiB
C++
// owner: shanyan.g
|
|
// owner group: transaction
|
|
|
|
/**
|
|
* 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.
|
|
*/
|
|
#include <gmock/gmock.h>
|
|
|
|
#define USING_LOG_PREFIX SHARE
|
|
#define protected public
|
|
#define private public
|
|
|
|
#include "env/ob_simple_cluster_test_base.h"
|
|
#include "storage/tx_storage/ob_ls_service.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
using namespace unittest;
|
|
namespace share
|
|
{
|
|
using namespace common;
|
|
using namespace transaction;
|
|
|
|
class TestTabletToLSCache : public unittest::ObSimpleClusterTestBase
|
|
{
|
|
public:
|
|
TestTabletToLSCache() : unittest::ObSimpleClusterTestBase("test_tablet_to_ls_cache") {}
|
|
};
|
|
|
|
TEST_F(TestTabletToLSCache, tablet_to_ls_cache)
|
|
{
|
|
// 0. init
|
|
int ret = OB_SUCCESS;
|
|
uint64_t tenant_id = OB_INVALID_TENANT_ID;
|
|
const int64_t TOTAL_NUM = 10;
|
|
ObSEArray<ObTabletLSPair, TOTAL_NUM> tablet_ls_pairs;
|
|
ASSERT_EQ(OB_SUCCESS, create_tenant());
|
|
ASSERT_EQ(OB_SUCCESS, get_tenant_id(tenant_id));
|
|
ASSERT_EQ(true, is_valid_tenant_id(tenant_id));
|
|
ASSERT_EQ(OB_SUCCESS, get_curr_simple_server().init_sql_proxy2());
|
|
ObMySQLProxy &sql_proxy = get_curr_simple_server().get_sql_proxy2();
|
|
|
|
MTL_SWITCH(tenant_id) {
|
|
ObLSService *ls_service = MTL(ObLSService *);
|
|
ASSERT_EQ(true, OB_NOT_NULL(ls_service));
|
|
ObTransService *tx_service = MTL(ObTransService *);
|
|
ASSERT_EQ(true, OB_NOT_NULL(tx_service));
|
|
int64_t base_size = tx_service->tablet_to_ls_cache_.size();
|
|
|
|
// 1. create tablet
|
|
ASSERT_EQ(OB_SUCCESS, batch_create_table(tenant_id, sql_proxy, TOTAL_NUM, tablet_ls_pairs));
|
|
ASSERT_EQ(TOTAL_NUM, tablet_ls_pairs.count());
|
|
ASSERT_EQ(base_size + TOTAL_NUM, tx_service->tablet_to_ls_cache_.size());
|
|
|
|
// 2. check and get ls
|
|
ObLSID ls_id;
|
|
ARRAY_FOREACH(tablet_ls_pairs, i) {
|
|
const ObTabletLSPair &pair = tablet_ls_pairs.at(i);
|
|
bool is_local = false;
|
|
ls_id.reset();
|
|
ASSERT_EQ(OB_SUCCESS, tx_service->check_and_get_ls_info(pair.get_tablet_id(), ls_id, is_local));
|
|
ASSERT_EQ(pair.get_ls_id().id(), ls_id.id());
|
|
ASSERT_EQ(true, is_local);
|
|
}
|
|
|
|
// 3. drop table
|
|
ObArray<ObTabletID> remove_tablet_ids;
|
|
ObLSHandle ls_handle;
|
|
ls_id = tablet_ls_pairs.at(0).get_ls_id();
|
|
ASSERT_EQ(OB_SUCCESS, ls_service->get_ls(ls_id, ls_handle, ObLSGetMod::STORAGE_MOD));
|
|
ObLSTabletService *ls_tablet_svr = ls_handle.get_ls()->get_tablet_svr();
|
|
ARRAY_FOREACH(tablet_ls_pairs, i) {
|
|
const ObTabletLSPair &pair = tablet_ls_pairs.at(i);
|
|
remove_tablet_ids.push_back(pair.get_tablet_id());
|
|
}
|
|
ASSERT_EQ(OB_SUCCESS, batch_drop_table(tenant_id, sql_proxy, TOTAL_NUM));
|
|
// after drop table, the tablets won't be removed immediately, so need to call remove_tablets
|
|
ASSERT_EQ(OB_SUCCESS, ls_tablet_svr->remove_tablets(remove_tablet_ids));
|
|
ASSERT_EQ(base_size, tx_service->tablet_to_ls_cache_.size());
|
|
}
|
|
}
|
|
|
|
} // namespace share
|
|
} // namespace oceanbase
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
oceanbase::unittest::init_log_and_gtest(argc, argv);
|
|
OB_LOGGER.set_log_level("INFO");
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|