init push

2021-05-31 22:56:52 +08:00
commit cea7de1475
7020 changed files with 5689869 additions and 0 deletions
--- a/src/rootserver/ob_all_server_task.cpp
+++ b/src/rootserver/ob_all_server_task.cpp
@ -0,0 +1,84 @@
+/**
+ * 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 RS
+
+#include "ob_all_server_task.h"
+
+#include "lib/profile/ob_trace_id.h"
+#include "share/config/ob_server_config.h"
+#include "share/ob_server_status.h"
+#include "rootserver/ob_server_manager.h"
+#include "rootserver/ob_rebalance_task_mgr.h"
+#include "rootserver/ob_root_utils.h"
+#include "observer/ob_server_struct.h"
+
+namespace oceanbase {
+namespace rootserver {
+using namespace common;
+using namespace share;
+
+ObAllServerTask::ObAllServerTask(
+    ObServerManager& server_manager, ObRebalanceTaskMgr& rebalance_task_mgr, const ObAddr& server, bool with_rootserver)
+    : server_manager_(server_manager),
+      rebalance_task_mgr_(rebalance_task_mgr),
+      server_(server),
+      with_rootserver_(with_rootserver)
+{}
+
+ObAllServerTask::~ObAllServerTask()
+{}
+
+int ObAllServerTask::process()
+{
+  int ret = OB_SUCCESS;
+  if (OB_ISNULL(ObCurTraceId::get_trace_id())) {
+    // Prevent the current trace_id from being overwritten
+    ObCurTraceId::init(GCONF.self_addr_);
+  }
+
+  THIS_WORKER.set_timeout_ts(INT64_MAX);
+  if (!ObRootServiceRoleChecker::is_rootserver(reinterpret_cast<ObIPartPropertyGetter*>(GCTX.ob_service_))) {
+    ret = OB_NOT_MASTER;
+    LOG_WARN("not master", K(ret));
+  } else if (OB_FAIL(server_manager_.adjust_server_status(server_, rebalance_task_mgr_, with_rootserver_))) {
+    LOG_WARN("fail to adjust server status", K(ret), K(server_));
+  }
+  return ret;
+}
+
+int64_t ObAllServerTask::get_deep_copy_size() const
+{
+  return sizeof(*this);
+}
+
+ObAsyncTask* ObAllServerTask::deep_copy(char* buf, const int64_t buf_size) const
+{
+  int ret = OB_SUCCESS;
+  ObAllServerTask* task = NULL;
+  if (NULL == buf || buf_size < static_cast<int64_t>(sizeof(ObAllServerTask))) {
+    ret = OB_INVALID_ARGUMENT;
+    LOG_WARN("buf is null or buf size not large enough",
+        "buf",
+        OB_P(buf),
+        K(buf_size),
+        "need size",
+        sizeof(ObAllServerTask),
+        K(ret));
+  } else {
+    task = new (buf) ObAllServerTask(server_manager_, rebalance_task_mgr_, server_, with_rootserver_);
+  }
+  return task;
+}
+
+}  // end namespace rootserver
+}  // end namespace oceanbase