init push

unittest/sql/optimizer/test_location_part_id.cpp

@@ -0,0 +1,123 @@
+/**
+ * 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 SQL_OPT
+#include "sql/optimizer/ob_opt_est_cost.h"
+#include "sql/optimizer/ob_opt_est_utils.h"
+#include "observer/ob_server.h"
+#include "observer/ob_req_time_service.h"
+#include <test_optimizer_utils.h>
+#define BUF_LEN 102400  // 100K
+using std::cout;
+using namespace oceanbase::json;
+using oceanbase::sql::ObTableLocation;
+using namespace oceanbase::sql;
+using namespace oceanbase::observer;
+namespace test {
+class TestLocationPartitionId : public TestOptimizerUtils {
+  public:
+  TestLocationPartitionId()
+  {}
+  ~TestLocationPartitionId()
+  {}
+};
+
+TEST_F(TestLocationPartitionId, calc_part_id_by_rowkeys)
+{
+  ObSQLSessionInfo session_info;
+  ObTZInfoMap tz_info;
+  ASSERT_EQ(OB_SUCCESS, tz_info.init(ObModIds::OB_HASH_BUCKET_TIME_ZONE_INFO_MAP));
+  ASSERT_EQ(OB_SUCCESS, session_info.init(0, 0, 0, &allocator_, &tz_info));
+  ASSERT_EQ(OB_SUCCESS, session_info.load_default_sys_variable(false, false));
+  const char* result_file = "./test_location_part_id.result";
+  const char* tmp_file = "./test_location_part_id.tmp";
+  const char* test_file = "./test_location_part_id.test";
+  std::ofstream of_result(tmp_file);
+  ASSERT_TRUE(of_result.is_open());
+  ObString database_name = "calc_partition";
+  std::ifstream if_tests(test_file);
+  ASSERT_TRUE(if_tests.is_open());
+  std::string line;
+  std::string total_line;
+  char buf[BUF_LEN];
+  int64_t case_num = 0;
+  while (std::getline(if_tests, line)) {
+    // allow human readable formatting
+    if (line.size() <= 0)
+      continue;
+    std::size_t begin = line.find_first_not_of('\t');
+    if (line.at(begin) == '#')
+      continue;
+    std::size_t end = line.find_last_not_of('\t');
+    std::string exact_line = line.substr(begin, end - begin + 1);
+    total_line += exact_line;
+    total_line += " ";
+    if (exact_line.at(exact_line.length() - 1) != ';')
+      continue;
+    else {
+      ++case_num;
+      ObString json_str(total_line.length() - 1, total_line.c_str());
+      ObString table_name;
+      ObSEArray<ObSEArray<ObObj, 3>, 1> row_array;
+      ASSERT_EQ(OB_SUCCESS, parse_row_from_json(json_str, table_name, row_array));
+      ObSEArray<ObRowkey, 3> rowkeys;
+      ObTableLocation table_location;
+      ObSEArray<int64_t, 3> part_ids;
+      ObSEArray<RowkeyArray, 3> rowkey_lists;
+      const ObTableSchema* table_schema = NULL;
+      ASSERT_EQ(
+          OB_SUCCESS, schema_guard_.get_table_schema(sys_tenant_id_, database_name, table_name, false, table_schema));
+      ASSERT_EQ(OB_SUCCESS,
+          convert_row_array_to_rowkeys(
+              row_array, rowkeys, table_schema->get_rowkey_info(), session_info_.get_timezone_info()));
+      ASSERT_EQ(OB_SUCCESS,
+          table_location.calculate_partition_ids_by_rowkey(
+              session_info_, schema_guard_, table_schema->get_table_id(), rowkeys, part_ids, rowkey_lists));
+      memset(buf, 0, BUF_LEN);
+      of_result << "----------------Case" << case_num << " start----------------" << std::endl;
+      of_result << "rowkey json"
+                << ": " << total_line << std::endl;
+      of_result << "part_ids: " << to_cstring(part_ids) << std::endl;
+      rowkey_lists.to_string(buf, BUF_LEN);
+      of_result << "rowkey_lists: " << buf << std::endl;
+      of_result << "----------------Case" << case_num << " end----------------" << std::endl;
+      total_line = "";
+    }
+  }
+  if_tests.close();
+  of_result.close();
+  formalize_tmp_file(tmp_file);
+  TestSqlUtils::is_equal_content(tmp_file, result_file);
+}
+}  // namespace test
+int main(int argc, char** argv)
+{
+  // char c;
+  // std::cout<<"Press any key to continue..."<<std::endl;
+  // std::cin>>c;
+  int ret = OB_SUCCESS;
+  observer::ObReqTimeGuard req_timeinfo_guard;
+  system("rm -rf test_location_part_id.log");
+  OB_LOGGER.set_log_level("INFO");
+  OB_LOGGER.set_file_name("test_location_part_id.log", true);
+  ObIntermResultItemPool::build_instance();
+  ObIntermResultPool::build_instance();
+  oceanbase::sql::init_sql_factories();
+  ::testing::InitGoogleTest(&argc, argv);
+  if (argc >= 2) {
+    if (strcmp("DEBUG", argv[1]) == 0 || strcmp("WARN", argv[1]) == 0)
+      OB_LOGGER.set_log_level(argv[1]);
+  }
+  test::parse_cmd_line_param(argc, argv, test::clp);
+  ret = RUN_ALL_TESTS();
+  return ret;
+}