Merge branch 'column_store'

Co-authored-by: wangt1xiuyi <13547954130@163.com>
Co-authored-by: yangqise7en <877793735@qq.com>
Co-authored-by: Zach41 <zach_41@163.com>

unittest/storage/test_compaction_memory_context.cpp

@@ -0,0 +1,155 @@
+/*
+ * test_compaction_mem_context.cpp
+ * Author:
+ */
+
+#include <gtest/gtest.h>
+#include <gmock/gmock.h>
+#include <thread>
+
+#define USING_LOG_PREFIX STORAGE
+#define protected public
+#define private public
+
+#include "lib/utility/ob_print_utils.h"
+#include "storage/compaction/ob_compaction_memory_context.h"
+#include "mtlenv/mock_tenant_module_env.h"
+#include "storage/blocksstable/ob_data_buffer.h"
+
+namespace oceanbase
+{
+using namespace share;
+using namespace common;
+using namespace storage;
+using namespace compaction;
+
+
+class TestCompactionMemCtx : public ::testing::Test
+{
+public:
+  TestCompactionMemCtx();
+  virtual ~TestCompactionMemCtx() = default;
+  virtual void SetUp() override;
+  virtual void TearDown() override;
+  static void SetUpTestCase();
+  static void TearDownTestCase();
+
+
+private:
+  const uint64_t tenant_id_;
+  ObTenantBase tenant_base_;
+};
+
+
+TestCompactionMemCtx::TestCompactionMemCtx()
+  : tenant_id_(1),
+    tenant_base_(tenant_id_)
+{
+}
+
+void TestCompactionMemCtx::SetUpTestCase()
+{
+  EXPECT_EQ(OB_SUCCESS, MockTenantModuleEnv::get_instance().init());
+}
+
+void TestCompactionMemCtx::TearDownTestCase()
+{
+  MockTenantModuleEnv::get_instance().destroy();
+}
+
+void TestCompactionMemCtx::SetUp()
+{
+  int ret = OB_SUCCESS;
+
+  ObTenantEnv::set_tenant(&tenant_base_);
+  ASSERT_EQ(OB_SUCCESS, tenant_base_.init());
+  ASSERT_EQ(tenant_id_, MTL_ID());
+}
+
+void TestCompactionMemCtx::TearDown()
+{
+  ObTenantEnv::set_tenant(nullptr);
+}
+
+
+namespace unittest
+{
+
+TEST_F(TestCompactionMemCtx, basic_test)
+{
+  ObLocalArena arena("TestArena", OB_MALLOC_NORMAL_BLOCK_SIZE);
+
+  void *buf = arena.alloc(128);
+  ASSERT_TRUE(nullptr != buf);
+  EXPECT_EQ(144, arena.used());
+  EXPECT_EQ(true, arena.total() <= 8064);
+
+  arena.free(buf);
+  arena.clear();
+  EXPECT_EQ(0, arena.used());
+  EXPECT_EQ(0, arena.total());
+}
+
+
+TEST_F(TestCompactionMemCtx, monitor_arena_test)
+{
+  ObArenaAllocator inner_arena;
+  ObTabletMergeDagParam param;
+  ObCompactionMemoryContext mem_ctx(param, inner_arena);
+
+  ObLocalArena arena("TestArena");
+  arena.bind_mem_ctx(mem_ctx);
+  EXPECT_EQ(&mem_ctx, arena.ref_mem_ctx_);
+
+  void *buf = arena.alloc(128);
+  ASSERT_TRUE(nullptr != buf);
+  EXPECT_EQ(144, arena.used());
+  EXPECT_EQ(true, arena.total() <= 8064);
+
+  mem_ctx.mem_click();
+  EXPECT_EQ(true, mem_ctx.get_total_mem_peak() <= 8064);
+
+  arena.free(buf);
+  arena.clear();
+  EXPECT_EQ(0, arena.used());
+  EXPECT_EQ(0, arena.total());
+
+  // only stat the peak val
+  mem_ctx.mem_click();
+  EXPECT_EQ(true, mem_ctx.get_total_mem_peak() <= 8064);
+
+  arena.~ObLocalArena();
+  EXPECT_EQ(nullptr, arena.ref_mem_ctx_);
+}
+
+
+TEST_F(TestCompactionMemCtx, mem_ctx_test)
+{
+  ObArenaAllocator inner_arena;
+  ObTabletMergeDagParam param;
+  ObCompactionMemoryContext mem_ctx(param, inner_arena);
+
+  void *buf1 = mem_ctx.alloc(128);
+  ASSERT_TRUE(nullptr != buf1);
+
+  void *buf2 = mem_ctx.local_alloc(128);
+  ASSERT_TRUE(nullptr != buf2);
+
+  mem_ctx.mem_click();
+  ASSERT_TRUE(0 != mem_ctx.get_total_mem_peak());
+}
+
+
+} // namespace unittest
+} // namespace oceanbase
+
+
+int main(int argc, char **argv)
+{
+  ::testing::InitGoogleTest(&argc, argv);
+  system("rm -f test_compaction_memory_context.log*");
+  OB_LOGGER.set_file_name("test_compaction_memory_context.log", true);
+  OB_LOGGER.set_log_level("INFO");
+  OB_LOGGER.set_max_file_size(256*1024*1024);
+  return RUN_ALL_TESTS();
+}