fix tenant_allocator reuse error
This commit is contained in:
@ -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;
|
||||
|
@ -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);
|
||||
|
@ -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);
|
||||
}
|
||||
|
Reference in New Issue
Block a user