/** * 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_dot_notation(ObSysFunRawExpr *expr); int print_json_object_star(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_