fix tenant_allocator reuse error

This commit is contained in:
obdev
2023-02-07 14:04:14 +08:00
committed by ob-robot
parent 02c1517a3e
commit d8d2e34880
3 changed files with 73 additions and 4 deletions

View File

@ -347,9 +347,7 @@ int ObMallocAllocator::add_tenant_allocator(ObTenantCtxAllocator *allocator)
while ((NULL != *cur) && (*cur)->get_tenant_id() < tenant_id) {
cur = &((*cur)->get_next());
}
if (OB_ISNULL(*cur)) {
*cur = allocator;
} else if ((*cur)->get_tenant_id() > tenant_id) {
if (OB_ISNULL(*cur) || (*cur)->get_tenant_id() > tenant_id) {
ObTenantCtxAllocator *next_allocator = *cur;
*cur = allocator;
((*cur)->get_next()) = next_allocator;

View File

@ -18,6 +18,21 @@
using namespace oceanbase::lib;
using namespace oceanbase::common;
class TestMallocAllocator : public ::testing::Test
{
public:
virtual void SetUp()
{
// cleanup
ObMallocAllocator *malloc_allocator = ObMallocAllocator::get_instance();
for (int i = OB_USER_TENANT_ID + 1; i < ObMallocAllocator::PRESERVED_TENANT_COUNT; i++) {
malloc_allocator->allocators_[i] = NULL;
}
malloc_allocator->unrecycled_allocators_ = NULL;
}
virtual void TearDown() {}
};
int64_t g_ctx_id = ObCtxIds::DEFAULT_CTX_ID;
TEST(TestMallocAllocator, reserved_tenant)
@ -171,6 +186,60 @@ TEST(TestMallocAllocator, allocator_guard)
}
}
TEST(TestMallocAllocator, bug_47543065)
{
std::pair<int, int64_t> flow[] = {
std::make_pair(1, 1002),
std::make_pair(1, 1001),
std::make_pair(1, 1006),
std::make_pair(1, 1005),
std::make_pair(1, 1004),
std::make_pair(1, 1003),
std::make_pair(0, 1001),
std::make_pair(1, 1008),
std::make_pair(1, 1007),
std::make_pair(0, 1005),
std::make_pair(0, 1007),
std::make_pair(0, 1008),
std::make_pair(1, 1008),
std::make_pair(1, 1007),
std::make_pair(0, 1007),
std::make_pair(0, 1008),
std::make_pair(1, 1008)
};
ObMallocAllocator *malloc_allocator = ObMallocAllocator::get_instance();
for (int i = 0; i < sizeof(flow)/sizeof(flow[0]);i++) {
int create = flow[i].first;
int64_t tenant_id = flow[i].second;
if (create) {
ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_and_add_tenant_allocator(tenant_id));
// make tenant memleak
auto ta = malloc_allocator->get_tenant_ctx_allocator(tenant_id, 0);
ASSERT_NE(nullptr, ta);
ASSERT_NE(nullptr, ta->alloc(8, ObMemAttr(tenant_id)));
} else {
malloc_allocator->recycle_tenant_allocator(tenant_id);
}
}
// check status
for (int i = 0; i < ObMallocAllocator::PRESERVED_TENANT_COUNT; i++) {
auto ta = malloc_allocator->allocators_[i];
uint64_t prev_tenant_id = 0;
while (ta != NULL) {
ASSERT_EQ(ta->get_tenant_id() % ObMallocAllocator::PRESERVED_TENANT_COUNT, i);
ASSERT_GT(ta->get_tenant_id(), prev_tenant_id);
auto ta_other = malloc_allocator->unrecycled_allocators_;
while (ta_other) {
ASSERT_NE(ta, ta_other);
ta_other = ta_other->get_next();
}
prev_tenant_id = ta->get_tenant_id();
ta = ta->get_next();
}
}
}
int main(int argc, char *argv[])
{
signal(49, SIG_IGN);

View File

@ -844,7 +844,9 @@ int ObMultiTenant::create_tenant(const ObTenantMeta &meta, bool write_slog, cons
tmp_ret = OB_SUCCESS;
if (create_step >= ObTenantCreateStep::STEP_CTX_MEM_CONFIG_SETTED) {
for (uint64_t ctx_id = 0; ctx_id < ObCtxIds::MAX_CTX_ID; ctx_id++) {
if (OB_SUCCESS != (tmp_ret = malloc_allocator->set_tenant_ctx_idle(tenant_id, ctx_id, 0))) {
if (NULL == malloc_allocator->get_tenant_ctx_allocator(tenant_id, ctx_id)) {
// do-nothing
} else if (OB_SUCCESS != (tmp_ret = malloc_allocator->set_tenant_ctx_idle(tenant_id, ctx_id, 0))) {
LOG_ERROR("fail to cleanup ctx mem config", K(tmp_ret), K(tenant_id), K(ctx_id));
SLEEP(1);
}