154 lines
4.5 KiB
C++
154 lines
4.5 KiB
C++
/**
|
|
* Copyright (c) 2022 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 protected public
|
|
#define private public
|
|
#include "share/allocator/ob_reserve_arena.h"
|
|
#include "share/rc/ob_tenant_base.h"
|
|
|
|
#define OK(ass) ASSERT_EQ(OB_SUCCESS, (ass))
|
|
|
|
namespace oceanbase
|
|
{
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::share;
|
|
using namespace std;
|
|
|
|
namespace unittest
|
|
{
|
|
class TestReserveArenaAllocator : public::testing::Test
|
|
{
|
|
public:
|
|
typedef ObReserveArenaAllocator<1024> ObStorageReserveAllocator;
|
|
TestReserveArenaAllocator();
|
|
virtual ~TestReserveArenaAllocator();
|
|
|
|
ObReserveArenaAllocator<1024> test_allocator_;
|
|
};
|
|
TestReserveArenaAllocator::TestReserveArenaAllocator():
|
|
test_allocator_(ObModIds::OB_STORE_ROW_EXISTER, OB_MALLOC_NORMAL_BLOCK_SIZE, 500)
|
|
{
|
|
}
|
|
|
|
TestReserveArenaAllocator::~TestReserveArenaAllocator()
|
|
{
|
|
}
|
|
|
|
TEST_F(TestReserveArenaAllocator, test_reset)
|
|
{
|
|
// test reset
|
|
// alloc buf
|
|
int64_t sz = 256;
|
|
void* p;
|
|
for (int64_t i = 0; i < 3; ++i) {
|
|
p = test_allocator_.alloc(sz);
|
|
}
|
|
ASSERT_EQ(test_allocator_.used(), 768);
|
|
ASSERT_EQ(test_allocator_.total(), 768);
|
|
ASSERT_EQ(test_allocator_.pos_, 768);
|
|
|
|
test_allocator_.reset();
|
|
ASSERT_EQ(test_allocator_.used(), 0);
|
|
ASSERT_EQ(test_allocator_.total(), 0);
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
// alloc new page 1k < size < 8k
|
|
sz = 2048;
|
|
p = test_allocator_.alloc(sz);
|
|
STORAGE_LOG(INFO, "alloc 2K", K(test_allocator_.allocator_));
|
|
ASSERT_NE(p, nullptr);
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_GE(test_allocator_.used(), sz);
|
|
ASSERT_GE(test_allocator_.total(), sz);
|
|
|
|
test_allocator_.reset();
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_EQ(test_allocator_.used(), 0);
|
|
ASSERT_EQ(test_allocator_.total(), 0);
|
|
|
|
// alloc new page size > 8k
|
|
sz = 10240;
|
|
p = test_allocator_.alloc(sz);
|
|
STORAGE_LOG(INFO, "alloc 10K", K(test_allocator_.allocator_));
|
|
ASSERT_NE(p, nullptr);
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_GE(test_allocator_.used(), sz);
|
|
ASSERT_GE(test_allocator_.total(), sz);
|
|
|
|
test_allocator_.reset();
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_EQ(test_allocator_.used(), 0);
|
|
ASSERT_EQ(test_allocator_.total(), 0);
|
|
|
|
}
|
|
|
|
TEST_F(TestReserveArenaAllocator, test_reuse)
|
|
{
|
|
// test reuse
|
|
// alloc buf
|
|
int64_t sz = 256;
|
|
void* p;
|
|
for (int64_t i = 0; i < 3; ++i) {
|
|
p = test_allocator_.alloc(sz);
|
|
}
|
|
ASSERT_EQ(test_allocator_.used(), 768);
|
|
ASSERT_EQ(test_allocator_.total(), 768);
|
|
ASSERT_EQ(test_allocator_.pos_, 768);
|
|
|
|
test_allocator_.reuse();
|
|
STORAGE_LOG(INFO, "after reuse", K(test_allocator_.allocator_));
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
// alloc new page 1k < size < 8k
|
|
sz = 2048;
|
|
p = test_allocator_.alloc(sz);
|
|
STORAGE_LOG(INFO, "alloc 2K", K(test_allocator_.allocator_));
|
|
ASSERT_NE(p, nullptr);
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_GE(test_allocator_.used(), sz);
|
|
ASSERT_GE(test_allocator_.total(), sz);
|
|
int allocate_total = test_allocator_.total();
|
|
|
|
test_allocator_.reuse();
|
|
STORAGE_LOG(INFO, "after reuse 2K", K(test_allocator_.allocator_));
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_EQ(test_allocator_.used(), 0);
|
|
ASSERT_LE(test_allocator_.total(), allocate_total);
|
|
|
|
// alloc new page size > 8k
|
|
sz = 10240;
|
|
p = test_allocator_.alloc(sz);
|
|
STORAGE_LOG(INFO, "alloc 10K", K(test_allocator_.allocator_));
|
|
ASSERT_NE(p, nullptr);
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_GE(test_allocator_.used(), sz);
|
|
ASSERT_GE(test_allocator_.total(), sz);
|
|
allocate_total = test_allocator_.total();
|
|
|
|
test_allocator_.reuse();
|
|
STORAGE_LOG(INFO, "after reuse 10K", K(test_allocator_.allocator_));
|
|
ASSERT_EQ(test_allocator_.pos_, 0);
|
|
ASSERT_EQ(test_allocator_.used(), 0);
|
|
ASSERT_LT(test_allocator_.total(), allocate_total);
|
|
|
|
}
|
|
|
|
}//end namespace unittest
|
|
}//end namespace oceanbase
|
|
int main(int argc, char **argv)
|
|
{
|
|
system("rm -f test_reserve_arena_allocator.log*");
|
|
oceanbase::common::ObLogger::get_logger().set_log_level("DEBUG");
|
|
OB_LOGGER.set_file_name("test_reserve_arena_allocator.log", true);
|
|
srand(time(NULL));
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
} |