[fix][coordinator]Change balance region defult region size

Author: LiuRuoyu01

src/common/constant.h

@@ -239,6 +239,8 @@ class Constant {
 
   // balance region
   inline static const float kBalanceRegionDefaultRegionCountRatio = 0.8f;
+  inline static const int32_t kBalanceRegionDefaultIndexRegionSize = 536870912;  // 512M
+  inline static const int32_t kBalanceRegionDefaultStoreRegionSize = 268435456;  // 256M
 
   // force_read_only
   inline static const std::string kForceReadOnlyKey = "FORCE_READ_ONLY";

src/config/config_helper.cc

@@ -280,6 +280,24 @@ float ConfigHelper::GetBalanceRegionCountRatio() {
   return (count < 0 || count > 1) ? Constant::kBalanceRegionDefaultRegionCountRatio : count;
 }
 
+int32_t ConfigHelper::GetBalanceRegionDefaultIndexRegionSize() {
+  auto config = ConfigManager::GetInstance().GetRoleConfig();
+  if (config == nullptr) {
+    return Constant::kBalanceRegionDefaultIndexRegionSize;
+  }
+  int32_t size = config->GetInt("coordinator.balance_region_default_index_region_size");
+  return (size <= 0) ? Constant::kBalanceRegionDefaultIndexRegionSize : size;
+}
+
+int32_t ConfigHelper::GetBalanceRegionDefaultStoreRegionSize() {
+  auto config = ConfigManager::GetInstance().GetRoleConfig();
+  if (config == nullptr) {
+    return Constant::kBalanceRegionDefaultStoreRegionSize;
+  }
+  int32_t size = config->GetInt("coordinator.balance_region_default_store_region_size");
+  return (size <= 0) ? Constant::kBalanceRegionDefaultStoreRegionSize : size;
+}
+
 int32_t ConfigHelper::GetWorkerThreadNum() {
   auto config = ConfigManager::GetInstance().GetRoleConfig();
   if (config == nullptr) {

src/config/config_helper.h

@@ -47,6 +47,8 @@ class ConfigHelper {
 
   static std::string GetBalanceRegionInspectionTimePeriod();
   static float GetBalanceRegionCountRatio();
+  static int32_t GetBalanceRegionDefaultIndexRegionSize();
+  static int32_t GetBalanceRegionDefaultStoreRegionSize();
 
   static int32_t GetWorkerThreadNum();
   static double GetWorkerThreadRatio();

src/coordinator/balance_region.cc

@@ -287,15 +287,23 @@ float StoreEntry::Score() {
   capacity_score_ = (capacity_score > 100) ? 100 : capacity_score;
 
   float region_count_score = 0;
-  int64_t region_approximate_size = ConfigHelper::GetSplitCheckApproximateSize();
   auto region_count = store_metric_.region_metrics_map_size();
-  float max_region_count = static_cast<float>(capacity) / region_approximate_size;
+  int32_t default_region_size = 0;
+  if (store_.store_type() == pb::common::StoreType::NODE_TYPE_INDEX) {
+    default_region_size = ConfigHelper::GetBalanceRegionDefaultIndexRegionSize();
+  } else if (store_.store_type() == pb::common::StoreType::NODE_TYPE_STORE) {
+    default_region_size = ConfigHelper::GetBalanceRegionDefaultStoreRegionSize();
+  } else {
+    DINGO_LOG(FATAL) << fmt::format("[balance.region] not support store_type: {}",
+                                    pb::common::StoreType_Name(store_.store_type()));
+  }
+  float max_region_count = static_cast<float>(capacity) / default_region_size;
   region_count_score = static_cast<float>(region_count) / max_region_count * FLAGS_balance_region_default_score;
   if (region_count_score > 100 || region_count_score < 0) {
     DINGO_LOG(WARNING) << fmt::format(
-        "[balance.region] region count score : {} {} , region_approximate_size : {} , region_count : {} , "
+        "[balance.region] region count score : {} {} , default region size : {} , region_count : {} , "
         "max region count : {}",
-        region_count_score, region_count_score > 100 ? "higher than 100" : "lower than 0", region_approximate_size,
+        region_count_score, region_count_score > 100 ? "higher than 100" : "lower than 0", default_region_size,
         region_count, max_region_count);
   }
   region_count_score_ = (region_count_score < 0) ? 0 : region_count_score;