[CP] [MDS] deserialize buffer_ctx by multi_data_source id

unittest/storage/multi_data_source/test_mds_new_ctx_deserialized.cpp

@@ -0,0 +1,122 @@
+/**
+ * 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.
+ */
+#include "storage/multi_data_source/compile_utility/compile_mapper.h"
+#include "storage/multi_data_source/mds_ctx.h"
+#include "storage/multi_data_source/mds_writer.h"
+#include "storage/multi_data_source/ob_tablet_create_mds_ctx.h"
+#include "storage/tx/ob_trans_define.h"
+#define UNITTEST_DEBUG
+#include "lib/utility/utility.h"
+#include <atomic>
+#include <gtest/gtest.h>
+#define private public
+#define protected public
+#include "storage/multi_data_source/compile_utility/mds_dummy_key.h"
+#include "storage/multi_data_source/runtime_utility/common_define.h"
+
+#include <thread>
+#include <iostream>
+#include <vector>
+#include <chrono>
+#include "storage/multi_data_source/runtime_utility/mds_factory.h"
+#include "common/ob_clock_generator.h"
+#include "storage/multi_data_source/mds_node.h"
+#include "common/meta_programming/ob_type_traits.h"
+#include "storage/multi_data_source/mds_row.h"
+namespace oceanbase {
+namespace storage
+{
+namespace mds
+{
+void *MdsAllocator::alloc(const int64_t size)
+{
+  void *ptr = ob_malloc(size, "MDS");
+  ATOMIC_INC(&alloc_times_);
+  MDS_LOG(DEBUG, "alloc obj", KP(ptr), K(size), K(lbt()));
+  return ptr;
+}
+void MdsAllocator::free(void *ptr) {
+  ATOMIC_INC(&free_times_);
+  MDS_LOG(DEBUG, "free obj", KP(ptr), K(lbt()));
+  ob_free(ptr);
+}
+}
+}
+namespace unittest {
+
+using namespace common;
+using namespace std;
+using namespace storage;
+using namespace mds;
+
+class TestMdsNewCtxDeserialized: public ::testing::Test
+{
+public:
+  TestMdsNewCtxDeserialized() {};
+  virtual ~TestMdsNewCtxDeserialized() {};
+  virtual void SetUp() {
+  };
+  virtual void TearDown() {
+  };
+private:
+  // disallow copy
+  DISALLOW_COPY_AND_ASSIGN(TestMdsNewCtxDeserialized);
+};
+
+TEST_F(TestMdsNewCtxDeserialized, deserialized_from_mds_ctx) {
+  MdsCtx old_ctx;
+  old_ctx.set_writer(MdsWriter(transaction::ObTransID(1)));
+  old_ctx.set_binding_type_id(TupleTypeIdx<mds::BufferCtxTupleHelper, MdsCtx>::value);
+  char buffer[1024];
+  for (auto &ch : buffer)
+    ch = 0xff;
+  int64_t pos = 0;
+  ASSERT_EQ(OB_SUCCESS, old_ctx.serialize(buffer, 1024, pos));
+  int64_t buffer_len = pos;
+  pos = 0;
+  ObTabletCreateMdsCtx new_ctx1;
+  ASSERT_EQ(OB_SUCCESS, new_ctx1.deserialize(buffer, buffer_len, pos));
+  pos = 0;
+  ASSERT_EQ(OB_SUCCESS, new_ctx1.deserialize(buffer, buffer_len + 1, pos));
+  pos = 0;
+  for (int idx = buffer_len; idx < 1024; ++idx)
+    buffer[idx] = 0;
+  ASSERT_EQ(OB_SUCCESS, new_ctx1.deserialize(buffer, buffer_len + 10, pos));
+  pos = 0;
+  ObStartTransferInMdsCtx new_ctx2;
+  ASSERT_EQ(OB_SUCCESS, new_ctx2.deserialize(buffer, buffer_len, pos));
+  pos = 0;
+  ObFinishTransferInMdsCtx new_ctx3;
+  ASSERT_EQ(OB_SUCCESS, new_ctx3.deserialize(buffer, buffer_len, pos));
+}
+
+}
+}
+
+int main(int argc, char **argv)
+{
+  system("rm -rf test_mds_new_ctx_deserialized.log");
+  oceanbase::common::ObLogger &logger = oceanbase::common::ObLogger::get_logger();
+  logger.set_file_name("test_mds_new_ctx_deserialized.log", false);
+  logger.set_log_level(OB_LOG_LEVEL_DEBUG);
+  testing::InitGoogleTest(&argc, argv);
+  int ret = RUN_ALL_TESTS();
+  int64_t alloc_times = oceanbase::storage::mds::MdsAllocator::get_alloc_times();
+  int64_t free_times = oceanbase::storage::mds::MdsAllocator::get_free_times();
+  if (alloc_times != free_times) {
+    MDS_LOG(ERROR, "memory may leak", K(free_times), K(alloc_times));
+    ret = -1;
+  } else {
+    MDS_LOG(INFO, "all memory released", K(free_times), K(alloc_times));
+  }
+  return ret;
+}