98 lines
4.0 KiB
C++
98 lines
4.0 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.
|
|
*/
|
|
|
|
#include "pl/parser/ob_pl_parser.h"
|
|
#include <gtest/gtest.h>
|
|
#include "lib/utility/ob_test_util.h"
|
|
#include "../test_sql_utils.h"
|
|
#include "lib/allocator/page_arena.h"
|
|
#include "lib/json/ob_json_print_utils.h" // for SJ
|
|
//#include "sql/plan_cache/ob_sql_parameterization.cpp"
|
|
#include <fstream>
|
|
#include <iterator>
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::sql;
|
|
using namespace oceanbase::pl;
|
|
namespace test
|
|
{
|
|
class TestPLParser: public ::testing::Test
|
|
{
|
|
public:
|
|
TestPLParser();
|
|
virtual ~TestPLParser();
|
|
virtual void SetUp();
|
|
virtual void TearDown();
|
|
private:
|
|
// disallow copy
|
|
DISALLOW_COPY_AND_ASSIGN(TestPLParser);
|
|
protected:
|
|
// data members
|
|
ObArenaAllocator allocator_;
|
|
};
|
|
|
|
TestPLParser::TestPLParser() : allocator_(ObModIds::TEST)
|
|
{
|
|
}
|
|
|
|
TestPLParser::~TestPLParser()
|
|
{
|
|
}
|
|
|
|
void TestPLParser::SetUp()
|
|
{
|
|
}
|
|
|
|
void TestPLParser::TearDown()
|
|
{
|
|
}
|
|
|
|
TEST_F(TestPLParser, basic_test)
|
|
{
|
|
ObPLParser parser(allocator_, CS_TYPE_UTF8MB4_BIN);
|
|
ParseResult parse_result;
|
|
// const char *query_str = "CREATE PROCEDURE sp (a varchar(10), b int(10)) BEGIN DECLARE c int default 1; DECLARE d, e varchar(11); IF 1=1 THEN select * from t1; ELSE IF 2=1 THEN select * from t1; ELSE select 1; END IF; END";
|
|
// const char *query_str = "create procedure sp() begin declare i bigint; if(i=1) then select 1 from dual; end if; end";
|
|
// const char *query_str = "do begin declare i bigint default 0; set i=i+1; if(i=1) then select 1 from dual; end if; end";
|
|
// const char *query_str = "CREATE PROCEDURE handlerdemo () BEGIN DECLARE CONTINUE HANDLER FOR SQLSTATE '23000' SET @x2 = 1; SET @x = 1; INSERT INTO test.t VALUES (1); SET @x = 2; INSERT INTO test.t VALUES (1); SET @x = 3; END;";
|
|
// const char *query_str = "CREATE PROCEDURE sp() insert into t1 values(1, 1);";
|
|
// const char *query_str = "CREATE PROCEDURE sp (a varchar(10), b int(10)) BEGIN DECLARE TYPE RecType IS RECORD(rno int, rname int, rsal int); DECLARE TYPE Tabletype IS TABLE OF RecType; DECLARE curval CURSOR FOR select c1 from t1; OPEN curval; FETCH curval INTO a, b; FETCH curval BULK INTO tab1; FETCH curval BULK INTO tab2 LIMIT 3; CLOSE curval; IF 1=1 THEN select * from t1; ELSE IF 2=1 THEN select * from t1; ELSE select 1; END IF; END";
|
|
// const char *query_str = "do begin declare i bigint default 0; set i=i+1; if(i=1) then select 1 from dual; end if; end";
|
|
// const char *query_str = "create procedure p(x int) begin declare i int; set i=i+1; if(1=1) then begin declare j int; select 1 from a where a1=i into x; end; end if; while i>1 do set i=i-1; end while; call f(x); end";
|
|
// const char* query_str = "do begin DECLARE total_sale INT DEFAULT 0; set total_sale = ?; update t1 set c1 = c1 + total_sale; end";
|
|
const char* query_str = "create procedure pro() begin DECLARE total_sale INT DEFAULT 0; set total_sale = 10; if total_sale > ? then update t1 set c1 = c1 + total_sale; end if; end";
|
|
ObString query = ObString::make_string(query_str);
|
|
int ret = OB_SUCCESS;
|
|
_OB_LOG(INFO, "QUERY: %s", query_str);
|
|
ret = parser.parse(query, query, parse_result);
|
|
ASSERT_EQ(OB_SUCCESS, -ret);
|
|
if (NULL != parse_result.result_tree_) {
|
|
_OB_LOG(INFO, "%s", CSJ(ObParserResultPrintWrapper(*parse_result.result_tree_)));
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
::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]);
|
|
} else {
|
|
OB_LOGGER.set_log_level("INFO");
|
|
}
|
|
|
|
OB_LOGGER.set_file_name("test_pl_parser.log", true);
|
|
return RUN_ALL_TESTS();
|
|
}
|