Add direct load client service

src/observer/table/ob_table_direct_load_processor.cpp

@@ -0,0 +1,69 @@
+/**
+ * Copyright (c) 2023 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 SERVER
+
+#include "ob_table_direct_load_processor.h"
+#include "observer/table_load/ob_table_load_client_service.h"
+
+namespace oceanbase
+{
+namespace observer
+{
+using namespace table;
+
+ObTableDirectLoadP::ObTableDirectLoadP(const ObGlobalContext &gctx)
+  : ObTableRpcProcessor(gctx),
+    allocator_(ObModIds::TABLE_PROC, OB_MALLOC_NORMAL_BLOCK_SIZE, MTL_ID()),
+    exec_ctx_(allocator_)
+{
+}
+
+int ObTableDirectLoadP::check_arg()
+{
+  int ret = OB_SUCCESS;
+  if (OB_UNLIKELY(arg_.operation_type_ == ObTableDirectLoadOperationType::MAX_TYPE ||
+                  arg_.arg_content_.empty())) {
+    ret = OB_INVALID_ARGUMENT;
+    LOG_WARN("invalid args", KR(ret), K(arg_));
+  }
+  return ret;
+}
+
+int ObTableDirectLoadP::try_process()
+{
+  int ret = OB_SUCCESS;
+  exec_ctx_.set_tenant_id(credential_.tenant_id_);
+  exec_ctx_.set_user_id(credential_.user_id_);
+  exec_ctx_.set_database_id(credential_.database_id_);
+  if (OB_FAIL(ObTableLoadClientService::direct_load_operate(exec_ctx_, arg_, result_))) {
+    LOG_WARN("fail to do direct load operate", KR(ret), K(arg_));
+  }
+  return ret;
+}
+
+ObTableAPITransCb *ObTableDirectLoadP::new_callback(rpc::ObRequest *req)
+{
+  UNUSED(req);
+  return nullptr;
+}
+
+uint64_t ObTableDirectLoadP::get_request_checksum()
+{
+  uint64_t checksum = 0;
+  checksum = ob_crc64(checksum, &result_.operation_type_, sizeof(result_.operation_type_));
+  checksum = ob_crc64(checksum, result_.res_content_.ptr(), result_.res_content_.length());
+  return checksum;
+}
+
+} // namespace observer
+} // namespace oceanbase