276 lines
10 KiB
C++
276 lines
10 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.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
#define private public // 获取私有成员
|
|
#include "observer/table/ob_table_session_pool.h"
|
|
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::table;
|
|
using namespace oceanbase::sql;
|
|
|
|
class TestTableSessPool: public ::testing::Test
|
|
{
|
|
public:
|
|
const int64_t TENANT_CNT = 10;
|
|
const int64_t USER_CNT = 100;
|
|
const int64_t NODE_CNT = 100;
|
|
const int64_t SESS_CNT = 10;
|
|
public:
|
|
TestTableSessPool() {}
|
|
virtual ~TestTableSessPool() {}
|
|
void prepare_sess_pool(ObTableApiSessPoolMgr &mgr);
|
|
private:
|
|
// disallow copy
|
|
DISALLOW_COPY_AND_ASSIGN(TestTableSessPool);
|
|
};
|
|
|
|
// 10租户，100用户，每个用户下面挂10个session
|
|
void TestTableSessPool::prepare_sess_pool(ObTableApiSessPoolMgr &mgr)
|
|
{
|
|
ASSERT_EQ(OB_SUCCESS, mgr.init());
|
|
uint64_t tenant_ids[TENANT_CNT];
|
|
uint64_t user_ids[USER_CNT];
|
|
ObTableApiSessPoolGuard pool_guards[TENANT_CNT];
|
|
for (uint64_t i = 0; i < TENANT_CNT; i++) {
|
|
tenant_ids[i] = i + 1;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.extend_sess_pool(tenant_ids[i], pool_guards[i]));
|
|
ObTableApiSessPool *tmp_pool = pool_guards[i].get_sess_pool();
|
|
ASSERT_NE(nullptr, tmp_pool);
|
|
for (uint64_t j = 0; j < USER_CNT; j++) {
|
|
user_ids[j] = j;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.get_session_pool(tenant_ids[i], pool_guards[i]));
|
|
ObTableApiSessPool *pool = pool_guards[i].get_sess_pool();
|
|
ASSERT_NE(nullptr, pool);
|
|
ObTableApiCredential credential;
|
|
credential.tenant_id_ = tenant_ids[i];
|
|
credential.user_id_ = user_ids[j];
|
|
credential.hash_val_ = credential.hash();
|
|
ASSERT_EQ(OB_SUCCESS, pool->update_sess(credential));
|
|
ObTableApiSessNode *node = nullptr;
|
|
ASSERT_EQ(OB_SUCCESS, tmp_pool->get_sess_node(credential.hash_val_, node));
|
|
ASSERT_NE(nullptr, node);
|
|
for (int64_t k = 0; k < SESS_CNT; k++) {
|
|
void *buf = node->allocator_.alloc(sizeof(ObTableApiSessNodeVal));
|
|
ASSERT_NE(nullptr, buf);
|
|
ObTableApiSessNodeVal *val = new (buf) ObTableApiSessNodeVal(tenant_ids[i], node);
|
|
val->is_inited_ = true;
|
|
ASSERT_EQ(true, node->sess_lists_.free_list_.add_last(val));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST_F(TestTableSessPool, get_session)
|
|
{
|
|
uint64_t tenant_id = 1;
|
|
uint64_t user_id = 0;
|
|
uint64_t key = user_id;
|
|
ObTableApiSessPoolMgr mgr;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.init());
|
|
ObTableApiSessPoolGuard pool_guard;
|
|
ASSERT_EQ(OB_HASH_NOT_EXIST, mgr.get_session_pool(tenant_id, pool_guard));
|
|
ASSERT_EQ(nullptr, pool_guard.get_sess_pool());
|
|
ASSERT_EQ(OB_SUCCESS, mgr.extend_sess_pool(tenant_id, pool_guard));
|
|
ObTableApiSessPool *pool = pool_guard.get_sess_pool();
|
|
ASSERT_TRUE(nullptr != pool);
|
|
ASSERT_TRUE(pool->is_inited_);
|
|
ASSERT_EQ(1, pool->ref_count_);
|
|
ASSERT_EQ(false, pool->is_deleted_);
|
|
ASSERT_EQ(tenant_id, pool->tenant_id_);
|
|
ObTableApiSessGuard sess_guard;
|
|
ObTableApiCredential credential;
|
|
credential.tenant_id_ = tenant_id;
|
|
credential.user_id_ = user_id;
|
|
ASSERT_EQ(OB_SUCCESS, pool->update_sess(credential));
|
|
ObTableApiSessNode *node = nullptr;
|
|
ASSERT_EQ(OB_SUCCESS, pool->get_sess_node(key, node));
|
|
ASSERT_TRUE(nullptr != node);
|
|
ASSERT_TRUE(node->is_empty());
|
|
void *buf = node->allocator_.alloc(sizeof(ObTableApiSessNodeVal));
|
|
ASSERT_NE(nullptr, buf);
|
|
ObTableApiSessNodeVal *new_val = new (buf) ObTableApiSessNodeVal(tenant_id, node);
|
|
new_val->is_inited_ = true;
|
|
ASSERT_EQ(true, node->sess_lists_.free_list_.add_last(new_val));
|
|
}
|
|
|
|
TEST_F(TestTableSessPool, remove_session)
|
|
{
|
|
int64_t tenant_id = 1;
|
|
uint64_t user_id = 0;
|
|
ObTableApiCredential credential;
|
|
credential.tenant_id_ = tenant_id;
|
|
credential.user_id_ = user_id;
|
|
ObTableApiSessNode node(credential);
|
|
for (int64_t i = 0; i < SESS_CNT; i++) {
|
|
void *buf = node.allocator_.alloc(sizeof(ObTableApiSessNodeVal));
|
|
ASSERT_NE(nullptr, buf);
|
|
ObTableApiSessNodeVal *val = new (buf) ObTableApiSessNodeVal(tenant_id, &node);
|
|
val->is_inited_ = true;
|
|
ASSERT_EQ(true, node.sess_lists_.free_list_.add_last(val));
|
|
}
|
|
ASSERT_EQ(false, node.is_empty());
|
|
ASSERT_EQ(SESS_CNT, node.sess_lists_.free_list_.get_size());
|
|
node.remove_unused_sess();
|
|
ASSERT_EQ(0, node.sess_lists_.free_list_.get_size());
|
|
}
|
|
|
|
TEST_F(TestTableSessPool, retire_session)
|
|
{
|
|
int ret = 0;
|
|
ObTableApiSessPoolMgr mgr;
|
|
prepare_sess_pool(mgr);
|
|
ObTableApiSessPoolForeachOp op;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.sess_pool_map_.foreach_refactored(op));
|
|
const ObTableApiSessPoolForeachOp::TelantIdArray &tenant_ids = op.get_telant_id_array();
|
|
ASSERT_EQ(TENANT_CNT, tenant_ids.count());
|
|
const int64_t N = tenant_ids.count();
|
|
// 1. 标记淘汰
|
|
for (int64_t i = 0; i < N; i++) {
|
|
uint64_t tenant_id = tenant_ids.at(i);
|
|
ObTableApiSessPoolGuard pool_guard;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.get_session_pool(tenant_id, pool_guard));
|
|
ObTableApiSessPool *pool = pool_guard.get_sess_pool();
|
|
ASSERT_NE(nullptr, pool);
|
|
ObTableApiSessForeachOp op;
|
|
ASSERT_EQ(OB_SUCCESS, pool->key_node_map_.foreach_refactored(op));
|
|
const ObTableApiSessForeachOp::SessKvArray &kvs = op.get_key_value_array();
|
|
ASSERT_EQ(NODE_CNT, kvs.count());
|
|
for (int64_t j = 0; j < kvs.count(); j++) {
|
|
if (j % 2 == 0) {
|
|
const ObTableApiSessForeachOp::ObTableApiSessKV &kv = kvs.at(j);
|
|
ASSERT_EQ(OB_SUCCESS, pool->move_sess_to_retired_list(kv.key_));
|
|
}
|
|
}
|
|
}
|
|
// 2. 触发淘汰
|
|
ASSERT_EQ(OB_SUCCESS, mgr.elimination_task_.run_recycle_retired_sess_task());
|
|
// 3. 检查
|
|
for (int64_t i = 0; i < N; i++) {
|
|
uint64_t tenant_id = tenant_ids.at(i);
|
|
ObTableApiSessPoolGuard pool_guard;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.get_session_pool(tenant_id, pool_guard));
|
|
ObTableApiSessPool *pool = pool_guard.get_sess_pool();
|
|
ASSERT_NE(nullptr, pool);
|
|
ObTableApiSessForeachOp op;
|
|
ASSERT_EQ(OB_SUCCESS, pool->key_node_map_.foreach_refactored(op));
|
|
const ObTableApiSessForeachOp::SessKvArray &kvs = op.get_key_value_array();
|
|
ASSERT_EQ(NODE_CNT/2, kvs.count());
|
|
for (int64_t j = 0; j < kvs.count(); j++) {
|
|
const ObTableApiSessForeachOp::ObTableApiSessKV &kv = kvs.at(j);
|
|
ASSERT_EQ(SESS_CNT, kv.node_->sess_lists_.free_list_.get_size());
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST_F(TestTableSessPool, reference_session)
|
|
{
|
|
// prepare
|
|
ObTableApiSessPoolMgr mgr;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.init());
|
|
uint64_t tenant_id = 1;
|
|
uint64_t user_id = 0;
|
|
ObTableApiSessPoolGuard pool_guard;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.extend_sess_pool(tenant_id, pool_guard));
|
|
ObTableApiSessPool *pool = pool_guard.get_sess_pool();
|
|
ASSERT_NE(nullptr, pool);
|
|
ObTableApiCredential credential;
|
|
credential.tenant_id_ = tenant_id;
|
|
credential.user_id_ = user_id;
|
|
ASSERT_EQ(OB_SUCCESS, pool->update_sess(credential));
|
|
ObTableApiSessNode *node = nullptr;
|
|
ASSERT_EQ(OB_SUCCESS, pool->get_sess_node(user_id, node));
|
|
ASSERT_NE(nullptr, node);
|
|
void *buf = node->allocator_.alloc(sizeof(ObTableApiSessNodeVal));
|
|
ASSERT_NE(nullptr, buf);
|
|
ObTableApiSessNodeVal *val = new (buf) ObTableApiSessNodeVal(tenant_id, node);
|
|
val->is_inited_ = true;
|
|
ASSERT_EQ(true, node->sess_lists_.free_list_.add_last(val));
|
|
// get and retire
|
|
{
|
|
// get
|
|
ObTableApiSessGuard guard;
|
|
ASSERT_EQ(OB_SUCCESS, pool->get_sess_info(credential, guard));
|
|
ASSERT_NE(nullptr, guard.get_sess_node_val());
|
|
// mark retire
|
|
ObTableApiSessForeachOp op;
|
|
ASSERT_EQ(OB_SUCCESS, pool->key_node_map_.foreach_refactored(op));
|
|
const ObTableApiSessForeachOp::SessKvArray &kvs = op.get_key_value_array();
|
|
ASSERT_EQ(1, kvs.count());
|
|
const ObTableApiSessForeachOp::ObTableApiSessKV &kv = kvs.at(0);
|
|
// run retire task
|
|
ASSERT_EQ(OB_SUCCESS, mgr.elimination_task_.run_recycle_retired_sess_task());
|
|
// check
|
|
op.reset();
|
|
ASSERT_EQ(OB_SUCCESS, pool->key_node_map_.foreach_refactored(op));
|
|
const ObTableApiSessForeachOp::SessKvArray &new_kvs = op.get_key_value_array();
|
|
ASSERT_EQ(1, new_kvs.count());
|
|
}
|
|
// retire after def ref
|
|
ASSERT_EQ(OB_SUCCESS, mgr.elimination_task_.run_recycle_retired_sess_task());
|
|
// check
|
|
ObTableApiSessPoolForeachOp op;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.sess_pool_map_.foreach_refactored(op));
|
|
const ObTableApiSessPoolForeachOp::TelantIdArray &arr = op.get_telant_id_array();
|
|
ASSERT_EQ(1, arr.count());
|
|
}
|
|
|
|
TEST_F(TestTableSessPool, retire_session_then_get_session)
|
|
{
|
|
uint64_t tenant_id = 1;
|
|
uint64_t user_id = 0;
|
|
ObTableApiSessPoolMgr mgr;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.init());
|
|
ObTableApiSessPoolGuard pool_guard;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.extend_sess_pool(tenant_id, pool_guard));
|
|
ObTableApiCredential credential;
|
|
credential.tenant_id_ = tenant_id;
|
|
credential.user_id_ = user_id;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.update_sess(credential));
|
|
|
|
// 塞一个ObTableApiSessNodeVal
|
|
ObTableApiSessPool *pool = pool_guard.get_sess_pool();
|
|
ObTableApiSessNode *node = nullptr;
|
|
ASSERT_EQ(OB_SUCCESS, pool->get_sess_node(user_id, node));
|
|
void *buf = node->allocator_.alloc(sizeof(ObTableApiSessNodeVal));
|
|
ASSERT_NE(nullptr, buf);
|
|
ObTableApiSessNodeVal *val = new (buf) ObTableApiSessNodeVal(tenant_id, node);
|
|
val->is_inited_ = true;
|
|
ASSERT_EQ(true, node->sess_lists_.free_list_.add_last(val));
|
|
|
|
// 第一次获取了session
|
|
ObTableApiSessGuard sess_guard;
|
|
ASSERT_EQ(OB_SUCCESS, mgr.get_sess_info(credential, sess_guard));
|
|
sess_guard.~ObTableApiSessGuard(); // 模仿访问结束，析构
|
|
|
|
// 长时间没有访问ob，session被放到淘汰链表，后台定时回收
|
|
ASSERT_EQ(OB_SUCCESS, pool->get_sess_node(user_id, node));
|
|
ASSERT_EQ(OB_SUCCESS, pool->move_sess_to_retired_list(user_id));
|
|
ASSERT_EQ(1, pool->retired_nodes_.size_);
|
|
ASSERT_EQ(OB_SUCCESS, mgr.elimination_task_.run_recycle_retired_sess_task());
|
|
ASSERT_EQ(0, pool->retired_nodes_.size_);
|
|
|
|
// 连接隔了很长时间，突然又访问db
|
|
ASSERT_EQ(OB_HASH_NOT_EXIST, pool->get_sess_node(user_id, node));
|
|
|
|
// 新的访问会创建新的池子
|
|
ASSERT_EQ(OB_SUCCESS, mgr.get_or_create_sess_pool(credential, pool_guard));
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
OB_LOGGER.set_log_level("INFO");
|
|
OB_LOGGER.set_file_name("TestTableSessPool.log", true);
|
|
::testing::InitGoogleTest(&argc,argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|