patch 4.0

2022-10-24 10:34:53 +08:00
parent 4ad6e00ec3
commit 93a1074b0c
10533 changed files with 2588271 additions and 2299373 deletions
--- a/src/share/ob_simple_mem_limit_getter.cpp
+++ b/src/share/ob_simple_mem_limit_getter.cpp
@ -0,0 +1,113 @@
+/**
+ * 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 COMMON
+
+#include "lib/ob_errno.h"
+#include "lib/oblog/ob_log_module.h"
+#include "share/ob_simple_mem_limit_getter.h"
+
+namespace oceanbase
+{
+namespace common
+{
+int ObSimpleMemLimitGetter::add_tenant(
+    const uint64_t tenant_id,
+    const int64_t lower_limit,
+    const int64_t upper_limit)
+{
+  int ret = OB_SUCCESS;
+  if (OB_UNLIKELY(OB_INVALID_TENANT_ID == tenant_id) ||
+      OB_UNLIKELY(lower_limit < 0) ||
+      OB_UNLIKELY(upper_limit < 0)) {
+    ret = OB_INVALID_ARGUMENT;
+    LOG_WARN("invalid argument", K(ret), K(tenant_id),
+             K(lower_limit), K(upper_limit));
+  } else {
+    ObTenantInfo tenant_info(tenant_id,
+                             lower_limit,
+                             upper_limit);
+    SpinWLockGuard guard(lock_);
+    if (has_tenant_(tenant_id)) {
+      // tenant is exist do nothing
+    } else if (OB_FAIL(tenant_infos_.push_back(tenant_info))) {
+      LOG_ERROR("push back into tenant_infos failed", K(ret), K(tenant_id));
+    }
+  }
+  return ret;
+}
+
+bool ObSimpleMemLimitGetter::has_tenant(const uint64_t tenant_id) const
+{
+  bool found = false;
+  SpinRLockGuard guard(lock_);
+  found = has_tenant_(tenant_id);
+  return found;
+}
+
+bool ObSimpleMemLimitGetter::has_tenant_(const uint64_t tenant_id) const
+{
+  bool found = false;
+  for (int i = 0; i < tenant_infos_.count(); i++) {
+    const ObTenantInfo &tenant_info = tenant_infos_[i];
+    if (tenant_info.tenant_id_ == tenant_id) {
+      found = true;
+      break;
+    }
+  }
+  return found;
+}
+
+int ObSimpleMemLimitGetter::get_all_tenant_id(ObIArray<uint64_t> &tenant_ids) const
+{
+  int ret = OB_SUCCESS;
+  SpinRLockGuard guard(lock_);
+  for (int i = 0; OB_SUCC(ret) && i < tenant_infos_.count(); i++) {
+    const ObTenantInfo &tenant_info = tenant_infos_[i];
+    if (OB_FAIL(tenant_ids.push_back(tenant_info.tenant_id_))) {
+      LOG_ERROR("push back tenant id failed", K(ret));
+    }
+  }
+  return ret;
+}
+
+int ObSimpleMemLimitGetter::get_tenant_mem_limit(
+    const uint64_t tenant_id,
+    int64_t &lower_limit,
+    int64_t &upper_limit) const
+{
+  int ret = OB_SUCCESS;
+  bool found = false;
+  SpinRLockGuard guard(lock_);
+  for (int i = 0; i < tenant_infos_.count(); i++) {
+    const ObTenantInfo &tenant_info = tenant_infos_[i];
+    if (tenant_info.tenant_id_ == tenant_id) {
+      found = true;
+      lower_limit = tenant_info.mem_lower_limit_;
+      upper_limit = tenant_info.mem_upper_limit_;
+      break;
+    }
+  }
+  if (!found) {
+    ret = OB_ENTRY_NOT_EXIST;
+    LOG_ERROR("tenant is not exist", K(ret), K(tenant_id));
+  }
+  return ret;
+}
+
+void ObSimpleMemLimitGetter::reset()
+{
+  tenant_infos_.reset();
+}
+
+}
+}