oceanbase/src/sql/engine/expr/ob_expr_lower.h

/**
 * 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_ENGINE_EXPR_LOWER_
#define OCEANBASE_SQL_ENGINE_EXPR_LOWER_

#include "sql/engine/expr/ob_expr_operator.h"

namespace oceanbase
{
namespace sql
{
class ObExprLowerUpper : public ObStringExprOperator
{
public:
  static const char SEPARATOR_IN_NLS_SORT_PARAM = '=';
  ObExprLowerUpper(common::ObIAllocator &alloc,
                   ObExprOperatorType type, const char *name, int32_t param_num);
  virtual ~ObExprLowerUpper() {}
  // For lower/upper of mysql and oracle
  virtual int calc_result_type1(ObExprResType &type, ObExprResType &text,
                                common::ObExprTypeCtx &type_ctx) const;
  // For oracle only nls_lower/nls_upper
  virtual int calc_result_typeN(ObExprResType &type,
                              ObExprResType *texts,
                              int64_t param_num,
                              common::ObExprTypeCtx &type_ctx) const;
  static int calc_common(const ObExpr &expr, ObEvalCtx &ctx,
                         ObDatum &expr_datum, bool lower, common::ObCollationType cs_type);
  template <char CA, char CZ>
  static void calc_common_inner_optimized(char *buf, const int32_t &buf_len,
                                          const ObString &m_text);
  template <bool lower>
  static int calc_common_vector(const ObExpr &expr, ObEvalCtx &ctx,
      const ObBitVector &skip, const EvalBound &bound, common::ObCollationType cs_type);

  static int calc_nls_common(const ObExpr &expr, ObEvalCtx &ctx,
                             ObDatum &expr_datum, bool lower);
  int cg_expr_common(ObExprCGCtx &op_cg_ctx, const ObRawExpr &raw_expr, ObExpr &rt_expr) const;
  int cg_expr_nls_common(ObExprCGCtx &op_cg_ctx,
                         const ObRawExpr &raw_expr,
                         ObExpr &rt_expr) const;
  DECLARE_SET_LOCAL_SESSION_VARS;

protected:
  virtual int calc(const common::ObCollationType cs_type, char *src, int32_t src_len,
                   char *dest, int32_t det_len, int32_t &out_len) const = 0;
  virtual int32_t get_case_mutiply(const common::ObCollationType cs_type) const = 0;

private:
  template <typename ArgVec, typename ResVec, bool lower>
  static int vector_lower_upper(VECTOR_EVAL_FUNC_ARG_DECL, common::ObCollationType cs_type);
  int calc(common::ObObj &result,
           const common::ObString &text,
           common::ObCollationType cs_type,
           common::ObIAllocator &calc_buf) const;
  DISALLOW_COPY_AND_ASSIGN(ObExprLowerUpper);
};

class ObExprLower : public ObExprLowerUpper
{
public:
  explicit  ObExprLower(common::ObIAllocator &alloc);
  virtual ~ObExprLower() {}
  virtual int calc(const common::ObCollationType cs_type, char *src, int32_t src_len,
                   char *dest, int32_t det_len, int32_t &out_len) const;
  virtual int32_t get_case_mutiply(const common::ObCollationType cs_type) const;
  virtual int cg_expr(ObExprCGCtx &op_cg_ctx,
                      const ObRawExpr &raw_expr,
                      ObExpr &rt_expr) const override;
  static int calc_lower(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum);
  static int eval_lower_vector(VECTOR_EVAL_FUNC_ARG_DECL);
private:
  DISALLOW_COPY_AND_ASSIGN(ObExprLower);
};

class ObExprUpper : public ObExprLowerUpper
{
public:
  explicit  ObExprUpper(common::ObIAllocator &alloc);
  virtual ~ObExprUpper() {}
  virtual int calc(const common::ObCollationType cs_type, char *src, int32_t src_len,
                   char *dest, int32_t det_len, int32_t &out_len) const;
  virtual int32_t get_case_mutiply(const common::ObCollationType cs_type) const;
  virtual int cg_expr(ObExprCGCtx &op_cg_ctx,
                      const ObRawExpr &raw_expr,
                      ObExpr &rt_expr) const override;
  static int calc_upper(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum);
  static int eval_upper_vector(VECTOR_EVAL_FUNC_ARG_DECL);
private:
  DISALLOW_COPY_AND_ASSIGN(ObExprUpper);
};

class ObExprNlsLower : public ObExprLowerUpper
{
public:
  explicit  ObExprNlsLower(common::ObIAllocator &alloc);
  virtual ~ObExprNlsLower() {}
  virtual int calc(const common::ObCollationType cs_type, char *src, int32_t src_len,
                   char *dest, int32_t det_len, int32_t &out_len) const;
  virtual int32_t get_case_mutiply(const common::ObCollationType cs_type) const;
  virtual int cg_expr(ObExprCGCtx &op_cg_ctx,
                      const ObRawExpr &raw_expr,
                      ObExpr &rt_expr) const override;
  static int calc_lower(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum);
private:
  DISALLOW_COPY_AND_ASSIGN(ObExprNlsLower);
};

class ObExprNlsUpper : public ObExprLowerUpper
{
public:
  explicit  ObExprNlsUpper(common::ObIAllocator &alloc);
  virtual ~ObExprNlsUpper() {}
  virtual int calc(const common::ObCollationType cs_type, char *src, int32_t src_len,
                   char *dest, int32_t det_len, int32_t &out_len) const;
  virtual int32_t get_case_mutiply(const common::ObCollationType cs_type) const;
  virtual int cg_expr(ObExprCGCtx &op_cg_ctx,
                      const ObRawExpr &raw_expr,
                      ObExpr &rt_expr) const override;
  static int calc_upper(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &expr_datum);
private:
  DISALLOW_COPY_AND_ASSIGN(ObExprNlsUpper);
};

}
}
#endif /* OCEANBASE_SQL_ENGINE_EXPR_LOWER_ */