patch 4.0

deps/oblib/unittest/lib/hash/test_ob_ref_mgr.cpp

@@ -0,0 +1,126 @@
+/**
+ * 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>
+#include <vector>
+#include <thread>
+
+#include "lib/hash/ob_multi_mod_ref_mgr.h"
+
+using namespace oceanbase;
+using namespace common;
+
+class TestLS
+{
+public:
+  enum LSGetMod
+  {
+    TRANS_GET_STORE_CTX = 0,
+    TRANS_PART_CTX_INIT = 1,
+    STORAGE_MOD = 2,
+    SQL_MOD = 3,
+    TOTAL_MAX_MOD = 4,
+  };
+
+};
+
+TEST(TestObRefMgr, basic)
+{
+  ObMultiModRefMgr<TestLS::LSGetMod> ref_mgr;
+  ASSERT_EQ(0, ref_mgr.get_total_ref_cnt());
+  ref_mgr.inc(TestLS::LSGetMod::TRANS_GET_STORE_CTX);
+  ASSERT_EQ(1, ref_mgr.get_total_ref_cnt());
+  ref_mgr.inc(TestLS::LSGetMod::TRANS_GET_STORE_CTX);
+  ref_mgr.inc(TestLS::LSGetMod::TRANS_PART_CTX_INIT);
+  ref_mgr.inc(TestLS::LSGetMod::STORAGE_MOD);
+  ref_mgr.inc(TestLS::LSGetMod::SQL_MOD);
+  ASSERT_EQ(5, ref_mgr.get_total_ref_cnt());
+
+  ref_mgr.print();
+  ref_mgr.dec(TestLS::LSGetMod::TRANS_GET_STORE_CTX);
+  ASSERT_EQ(4, ref_mgr.get_total_ref_cnt());
+  ref_mgr.dec(TestLS::LSGetMod::TRANS_GET_STORE_CTX);
+  ref_mgr.dec(TestLS::LSGetMod::TRANS_PART_CTX_INIT);
+  ref_mgr.dec(TestLS::LSGetMod::STORAGE_MOD);
+  ref_mgr.dec(TestLS::LSGetMod::SQL_MOD);
+  ASSERT_EQ(0, ref_mgr.get_total_ref_cnt());
+
+  ref_mgr.print();
+}
+
+TEST(TestObRefMgr, bench)
+{
+  ObMultiModRefMgr<TestLS::LSGetMod> ref_mgr;
+  std::vector<std::thread> ths;
+  bool g_stop = false;
+  for (int i = 0; i < 200; i++) {
+    std::thread th([&, i] () {
+      while (!g_stop) {
+        ref_mgr.inc(TestLS::LSGetMod(i%TestLS::LSGetMod::TOTAL_MAX_MOD));
+        ref_mgr.dec(TestLS::LSGetMod(i%TestLS::LSGetMod::TOTAL_MAX_MOD));
+      }
+      ::sleep(1);
+    });
+    ths.push_back(std::move(th));
+  }
+  for (int i = 0; i < 3; i++) {
+    ref_mgr.print();
+    ::sleep(1);
+  }
+  g_stop = true;
+  for (auto &it : ths) {
+    it.join();
+  }
+  ASSERT_EQ(0, ref_mgr.get_total_ref_cnt());
+  ref_mgr.print();
+}
+
+
+TEST(TestObRefMgr, bench2)
+{
+  ObMultiModRefMgr<TestLS::LSGetMod> ref_mgr;
+  std::vector<std::thread> ths;
+  bool g_start = false;
+  int release_cnt = 0;
+  for (int i = 0; i < 200; i++) {
+    ref_mgr.inc(TestLS::LSGetMod::STORAGE_MOD);
+    std::thread th([&] () {
+      while (true) {
+        if (ATOMIC_LOAD(&g_start)) {
+          bool can_release = ref_mgr.dec(TestLS::LSGetMod::STORAGE_MOD);
+          if (can_release) {
+            ATOMIC_INC(&release_cnt);
+          }
+          break;
+        }
+      }
+      ::sleep(1);
+    });
+    ths.push_back(std::move(th));
+  }
+  ref_mgr.set_delete();
+  ASSERT_EQ(200, ref_mgr.get_total_ref_cnt());
+  ATOMIC_STORE(&g_start, true);
+  for (auto &it : ths) {
+    it.join();
+  }
+  ASSERT_EQ(0, ref_mgr.get_total_ref_cnt());
+  ASSERT_EQ(1, release_cnt);
+  ref_mgr.print();
+}
+
+int main(int argc, char **argv)
+{
+  OB_LOGGER.set_log_level("INFO");
+  testing::InitGoogleTest(&argc,argv);
+  return RUN_ALL_TESTS();
+}