init push

2021-05-31 22:56:52 +08:00
commit cea7de1475
7020 changed files with 5689869 additions and 0 deletions
--- a/unittest/share/test_locality_priority.cpp
+++ b/unittest/share/test_locality_priority.cpp
@ -0,0 +1,504 @@
+/**
+ * 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.
+ */
+
+#define USING_LOG_PREFIX SHARE
+#include "share/ob_locality_priority.h"
+#include <gtest/gtest.h>
+
+namespace oceanbase {
+namespace share {
+using namespace common;
+
+class TestLocalityPriority : public ::testing::Test {
+  public:
+  virtual void SetUp();
+  virtual void TearDown();
+  void get_primary_region_prioriry(const char* primary_zone,
+      const common::ObIArray<ObLocalityRegion>& locality_region_array,
+      common::ObIArray<ObLocalityRegion>& tenant_locality_region);
+
+  private:
+  typedef common::ObFixedLengthString<common::MAX_ZONE_LENGTH> Zone;
+
+  protected:
+  common::ObSEArray<ObLocalityRegion, 5> locality_region_array_;
+};
+
+void TestLocalityPriority::SetUp()
+{
+  locality_region_array_.reset();
+  ObLocalityRegion region;
+  region.region_ = "hz";
+  Zone zone1 = "hz1";
+  Zone zone2 = "hz2";
+  region.zone_array_.push_back(zone1);
+  region.zone_array_.push_back(zone2);
+  locality_region_array_.push_back(region);
+
+  region.reset();
+  region.region_ = "sh";
+  Zone zone3 = "sh1";
+  Zone zone4 = "sh2";
+  region.zone_array_.push_back(zone3);
+  region.zone_array_.push_back(zone4);
+  locality_region_array_.push_back(region);
+
+  region.reset();
+  region.region_ = "sz";
+  Zone zone5 = "sz1";
+  region.zone_array_.push_back(zone5);
+  locality_region_array_.push_back(region);
+
+  SHARE_LOG(INFO, "locality_region", K_(locality_region_array));
+}
+
+void TestLocalityPriority::TearDown()
+{
+  locality_region_array_.reset();
+}
+
+void TestLocalityPriority::get_primary_region_prioriry(const char* primary_zone,
+    const common::ObIArray<ObLocalityRegion>& locality_region_array,
+    common::ObIArray<ObLocalityRegion>& tenant_locality_region)
+{
+  int ret = OB_SUCCESS;
+
+  if (OB_FAIL(ObLocalityPriority::get_primary_region_prioriry(
+          primary_zone, locality_region_array, tenant_locality_region))) {
+    SHARE_LOG(WARN, "ObLocalityPriority::get_primary_region_prioriry error", K(ret));
+  } else {
+    SHARE_LOG(INFO, "ObLocalityPriority::get_primary_region_prioriry success", K(tenant_locality_region));
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_three_region)
+{
+  const char* tenant_primary_zone = "hz1,hz2;sh1,sh2;sz1";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz3";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz4";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh3";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh100";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "hz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 128);
+  }
+
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 128);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz3";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 128);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "bj";
+    locality_info.local_zone_ = "bj1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_one_region)
+{
+  const char* tenant_primary_zone = "hz1";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "bj";
+    locality_info.local_zone_ = "bj1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_two_region)
+{
+  const char* tenant_primary_zone = "hz1;sh2";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_same_region)
+{
+  const char* tenant_primary_zone = "hz1;hz2";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_diff_region)
+{
+  const char* tenant_primary_zone = "  hz1  ,  sh1  ; hz2   ";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sh";
+    locality_info.local_zone_ = "sh2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 0);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz1";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_one_empty)
+{
+  const char* tenant_primary_zone = "  ;  hz1";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, 64);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_two_empty)
+{
+  const char* tenant_primary_zone = "  ;  ";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_all_empty)
+{
+  const char* tenant_primary_zone = "      ";
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  get_primary_region_prioriry(tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "hz";
+    locality_info.local_zone_ = "hz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+  {
+    ObLocalityInfo locality_info;
+    locality_info.local_region_ = "sz";
+    locality_info.local_zone_ = "sz2";
+    uint64_t zone_priority = UINT64_MAX;
+    ObLocalityPriority::get_region_priority(locality_info, tenant_locality_region, zone_priority);
+    ASSERT_EQ(zone_priority, UINT64_MAX);
+  }
+}
+
+TEST_F(TestLocalityPriority, get_region_priority_primary_NULL)
+{
+  const char* tenant_primary_zone = NULL;
+  ObSEArray<ObLocalityRegion, 5> tenant_locality_region;
+  int ret = ObLocalityPriority::get_primary_region_prioriry(
+      tenant_primary_zone, locality_region_array_, tenant_locality_region);
+  EXPECT_EQ(common::OB_INVALID_ARGUMENT, ret);
+}
+
+}  // end namespace share
+}  // end namespace oceanbase
+
+int main(int argc, char** argv)
+{
+  oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
+  OB_LOGGER.set_log_level("INFO");
+  testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}