124 lines
4.6 KiB
C++
124 lines
4.6 KiB
C++
/**
|
|
* 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;
|
|
}
|