185 lines
7.8 KiB
C++
185 lines
7.8 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.
|
|
*/
|
|
|
|
#ifndef OCEANBASE_SQL_RESOLVER_EXPR_OB_RAW_EXPR_PRINTER_H_
|
|
#define OCEANBASE_SQL_RESOLVER_EXPR_OB_RAW_EXPR_PRINTER_H_
|
|
|
|
#include "sql/resolver/expr/ob_raw_expr.h"
|
|
#include "lib/string/ob_sql_string.h"
|
|
#include "common/ob_smart_call.h"
|
|
#include "sql/ob_sql_utils.h"
|
|
#include "sql/dblink/ob_dblink_utils.h"
|
|
namespace oceanbase
|
|
{
|
|
namespace common
|
|
{
|
|
class ObTimeZoneInfo;
|
|
}
|
|
namespace sql
|
|
{
|
|
class ObRawExprPrinter
|
|
{
|
|
#define LEN_AND_PTR(str) (str.length()), (str.ptr())
|
|
#define SQL_ESCAPE_STR(str) (to_cstring(ObHexEscapeSqlStr(str)))
|
|
#define _DATA_PRINTF(...) databuff_printf(buf_, buf_len_, *pos_, __VA_ARGS__)
|
|
#define DATA_PRINTF(...) \
|
|
do { \
|
|
if (OB_SUCC(ret)) { \
|
|
if (OB_ISNULL(buf_) || OB_ISNULL(pos_)) { \
|
|
ret = OB_ERR_UNEXPECTED; \
|
|
LOG_WARN("buf_ or pos_ is null", K(ret)); \
|
|
} else if (OB_FAIL(_DATA_PRINTF(__VA_ARGS__))) { \
|
|
LOG_WARN("fail to print", K(ret)); \
|
|
} \
|
|
} \
|
|
} while(0) \
|
|
|
|
#define PRINT_IDENT(ident_str) \
|
|
do { \
|
|
if (OB_SUCC(ret) && OB_FAIL(ObSQLUtils::print_identifier(buf_, buf_len_, (*pos_), \
|
|
print_params_.cs_type_, \
|
|
ident_str, \
|
|
lib::is_oracle_mode()))) { \
|
|
LOG_WARN("fail to print ident str", K(ret), K(ident_str)); \
|
|
} \
|
|
} while(0)
|
|
|
|
#define CONVERT_CHARSET_FOR_RPINT(alloc, input_str) \
|
|
do { \
|
|
if (OB_SUCC(ret) && OB_FAIL(ObCharset::charset_convert(allocator, \
|
|
input_str, \
|
|
CS_TYPE_UTF8MB4_BIN, \
|
|
print_params_.cs_type_, \
|
|
input_str))) { \
|
|
LOG_WARN("fail to gen ident str", K(ret), K(input_str)); \
|
|
} \
|
|
} while(0)
|
|
|
|
#define PRINT_EXPR(expr) \
|
|
do { \
|
|
if (OB_SUCCESS == ret && OB_FAIL(SMART_CALL(print(expr)))) { \
|
|
LOG_WARN("fail to print expr", K(ret)); \
|
|
} \
|
|
} while(0) \
|
|
|
|
#define SET_SYMBOL_IF_EMPTY(str) \
|
|
do { \
|
|
if (0 == symbol.length()) { \
|
|
symbol = str; \
|
|
} \
|
|
} while (0) \
|
|
|
|
#define PRINT_QUOT \
|
|
do { \
|
|
if (lib::is_oracle_mode()) { \
|
|
DATA_PRINTF("\""); \
|
|
} else { \
|
|
DATA_PRINTF("`"); \
|
|
} \
|
|
} while (0);
|
|
|
|
#define PRINT_QUOT_WITH_SPACE \
|
|
DATA_PRINTF(" "); \
|
|
PRINT_QUOT;
|
|
|
|
#define PRINT_IDENT_WITH_QUOT(ident_str) \
|
|
do { \
|
|
PRINT_QUOT; \
|
|
PRINT_IDENT(ident_str); \
|
|
PRINT_QUOT; \
|
|
} while (0)
|
|
|
|
// cast函数在parse阶段用到这两个宏, 但定义在sql_parse_tab.c中
|
|
// cast函数功能不完善，beta之前不会修改, 先定义在这里
|
|
// TODO@nijia.nj
|
|
#define BINARY_COLLATION 63
|
|
#define INVALID_COLLATION 0
|
|
|
|
public:
|
|
ObRawExprPrinter();
|
|
ObRawExprPrinter(char *buf, int64_t buf_len, int64_t *pos, ObSchemaGetterGuard *schema_guard,
|
|
common::ObObjPrintParams print_params = common::ObObjPrintParams(),
|
|
const ParamStore *param_store = NULL);
|
|
virtual ~ObRawExprPrinter();
|
|
|
|
void init(char *buf, int64_t buf_len, int64_t *pos, ObSchemaGetterGuard *schema_guard,
|
|
ObObjPrintParams print_params, const ParamStore *param_store = NULL);
|
|
// stmt中会出现若干expr, 为了避免反复实例化，这里将expr作为do_print的参数
|
|
int do_print(ObRawExpr *expr, ObStmtScope scope, bool only_column_namespace = false, bool print_cte = false);
|
|
int pre_check_treat_opt(ObRawExpr *expr, bool &is_treat);
|
|
private:
|
|
int print(ObRawExpr *expr);
|
|
|
|
int print(ObConstRawExpr *expr);
|
|
int print(ObQueryRefRawExpr *expr);
|
|
int print(ObColumnRefRawExpr *expr);
|
|
int print(ObOpRawExpr *expr);
|
|
int print(ObCaseOpRawExpr *expr);
|
|
int print(ObSetOpRawExpr *expr);
|
|
int print(ObAggFunRawExpr *expr);
|
|
int print(ObSysFunRawExpr *expr);
|
|
int print_translate(ObSysFunRawExpr *expr);
|
|
int print(ObUDFRawExpr *expr);
|
|
int print(ObWinFunRawExpr *expr);
|
|
int print(ObPseudoColumnRawExpr *expr);
|
|
|
|
int print_date_unit(ObRawExpr *expr);
|
|
int print_get_format_unit(ObRawExpr *expr);
|
|
int print_cast_type(ObRawExpr *expr);
|
|
int print_json_expr(ObSysFunRawExpr *expr);
|
|
int print_json_value(ObSysFunRawExpr *expr);
|
|
int print_json_query(ObSysFunRawExpr *expr);
|
|
int print_json_exists(ObSysFunRawExpr *expr);
|
|
int print_json_equal(ObSysFunRawExpr *expr);
|
|
int print_json_array(ObSysFunRawExpr *expr);
|
|
int print_json_mergepatch(ObSysFunRawExpr *expr);
|
|
int print_json_return_type(ObRawExpr *expr);
|
|
int print_is_json(ObSysFunRawExpr *expr);
|
|
int print_json_object(ObSysFunRawExpr *expr);
|
|
int print_ora_json_arrayagg(ObAggFunRawExpr *expr);
|
|
int print_ora_json_objectagg(ObAggFunRawExpr *expr);
|
|
int print_partition_exprs(ObWinFunRawExpr *expr);
|
|
int print_order_items(ObWinFunRawExpr *expr);
|
|
int print_window_clause(ObWinFunRawExpr *expr);
|
|
int print_xml_parse_expr(ObSysFunRawExpr *expr);
|
|
int print_xml_element_expr(ObSysFunRawExpr *expr);
|
|
int print_xml_attributes_expr(ObSysFunRawExpr *expr);
|
|
int print_xml_agg_expr(ObAggFunRawExpr *expr);
|
|
int print_xml_serialize_expr(ObSysFunRawExpr *expr);
|
|
|
|
int print_type(const ObExprResType &dst_type);
|
|
|
|
int inner_print_fun_params(ObSysFunRawExpr &expr);
|
|
|
|
// disallow copy
|
|
DISALLOW_COPY_AND_ASSIGN(ObRawExprPrinter);
|
|
private:
|
|
// data members
|
|
char *buf_;
|
|
int64_t buf_len_;
|
|
// avoid to update pos_ between different printers(mainly ObRawExprPrinter
|
|
// and ObSelectStmtPrinter), we definite pointer of pos_ rather than object
|
|
int64_t *pos_;
|
|
ObStmtScope scope_;
|
|
bool only_column_namespace_;
|
|
const common::ObTimeZoneInfo *tz_info_;
|
|
ObObjPrintParams print_params_;
|
|
const ParamStore *param_store_;
|
|
ObSchemaGetterGuard *schema_guard_;
|
|
bool print_cte_;
|
|
};
|
|
|
|
} // end namespace sql
|
|
} // end namespace oceanbase
|
|
|
|
#endif // OCEANBASE_SQL_RESOLVER_EXPR_OB_RAW_EXPR_PRINTER_H_
|