110 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			110 lines
		
	
	
		
			3.4 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_join_order.h"
 | |
| #include "observer/ob_req_time_service.h"
 | |
| #include "test_optimizer_utils.h"
 | |
| 
 | |
| namespace oceanbase
 | |
| {
 | |
| namespace sql
 | |
| {
 | |
| class Path;
 | |
| }
 | |
| }
 | |
| namespace test
 | |
| {
 | |
| class TestJoinOrder: public TestOptimizerUtils
 | |
| {
 | |
| public:
 | |
|   TestJoinOrder(){}
 | |
|   ~TestJoinOrder(){}
 | |
| };
 | |
| 
 | |
| TEST_F(TestJoinOrder, ob_join_order_select)
 | |
| {
 | |
|   const char* test_file = "./test_join_order_case.sql";
 | |
|   const char* result_file = "./test_join_order_case.result";
 | |
|   const char* tmp_file = "./test_join_order_case.tmp";
 | |
|   run_test(test_file, result_file, tmp_file);
 | |
| }
 | |
| 
 | |
| class TestPath : public oceanbase::sql::Path
 | |
| {
 | |
|   virtual void get_name_internal(char *buf, const int64_t buf_len, int64_t &pos) const
 | |
|   { UNUSED(buf), UNUSED(buf_len), UNUSED(pos);}
 | |
|   virtual int estimate_cost() override { return OB_SUCCESS; }
 | |
|   virtual int re_estimate_cost(double need_row_count, double &cost) { UNUSED(need_row_count); cost = 0; return OB_SUCCESS; }
 | |
|   virtual int contain_fake_cte(bool &contains) const { contains = false; return OB_SUCCESS; }
 | |
| 
 | |
| };
 | |
| 
 | |
| TEST_F(TestJoinOrder, ob_join_order_src)
 | |
| {
 | |
|   ObArenaAllocator allocator(ObModIds::OB_BUFFER);
 | |
|   JoinInfo out_join(LEFT_OUTER_JOIN);
 | |
|   char buf[256] = {0};
 | |
|   AccessPath a_path(123, 123, 456, NULL, oceanbase::sql::NULLS_FIRST_ASC);
 | |
|   a_path.op_cost_ = 100;
 | |
|   a_path.cost_ = 100;
 | |
|   int64_t pos = 0;
 | |
|   int ret = a_path.get_name(buf, 256, pos);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ASSERT_EQ(a_path.get_cost(), 100);
 | |
|   memset(buf, '\0', 256);
 | |
|   a_path.get_name_internal(buf, 256, pos);
 | |
|   JoinPath join_path;
 | |
|   memset(buf, '\0', 256);
 | |
|   join_path.get_name_internal(buf, 256, pos);
 | |
|   SubQueryPath sp_path;
 | |
|   memset(buf, '\0', 256);
 | |
|   sp_path.get_name_internal(buf, 256, pos);
 | |
|   ObGlobalHint global_hint;
 | |
|   ObAddr addr;
 | |
|   ObRawExprFactory expr_factory(allocator);
 | |
|   ObOptimizerContext ctx(NULL, NULL, NULL, NULL, NULL,
 | |
|                          allocator, NULL, NULL, addr, NULL,
 | |
|                          global_hint, expr_factory, NULL, false, NULL);
 | |
|   ObSelectLogPlan plan(ctx, NULL);
 | |
|   ObJoinOrder join_order(NULL, &plan, sql::INVALID);
 | |
| 
 | |
|   TestPath test_path;
 | |
|   LOG_INFO("test to string func", K(test_path), K(join_order));
 | |
|   pos = join_order.get_name(buf, 256);
 | |
|   //case: test reset()
 | |
|   ret = join_order.interesting_paths_.push_back(&sp_path);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   join_order.join_info_ = &out_join;
 | |
|   ObSysFunRawExpr raw_expr;
 | |
|   ret = join_order.restrict_info_set_.push_back(&raw_expr);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
| }
 | |
| 
 | |
| }//end of namespace test
 | |
| int main(int argc, char **argv)
 | |
| {
 | |
|   OB_LOGGER.set_file_name("test_join_order.log", true);
 | |
|   observer::ObReqTimeGuard req_timeinfo_guard;
 | |
|   init_sql_factories();
 | |
|   ::testing::InitGoogleTest(&argc,argv);
 | |
|   if(argc >= 2)
 | |
|   {
 | |
|     if (strcmp("DEBUG", argv[1]) == 0
 | |
|         || strcmp("INFO", argv[1]) == 0
 | |
|         || strcmp("WARN", argv[1]) == 0)
 | |
|     OB_LOGGER.set_log_level(argv[1]);
 | |
|   }
 | |
|   test::parse_cmd_line_param(argc, argv, test::clp);
 | |
|   return RUN_ALL_TESTS();
 | |
| }
 | 
