init push

src/sql/engine/expr/ob_expr_subquery_ns_equal.cpp

@@ -0,0 +1,62 @@
+/**
+ * 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_ENG
+#include "sql/engine/expr/ob_expr_subquery_ns_equal.h"
+#include "common/row/ob_row.h"
+#include "common/object/ob_obj_compare.h"
+#include "sql/session/ob_sql_session_info.h"
+#include "sql/engine/expr/ob_expr_equal.h"
+#include "sql/engine/expr/ob_expr_result_type_util.h"
+
+namespace oceanbase {
+using namespace common;
+namespace sql {
+
+ObExprSubQueryNSEqual::ObExprSubQueryNSEqual(ObIAllocator& alloc)
+    : ObSubQueryRelationalExpr(alloc, T_OP_SQ_NSEQ, N_SQ_NS_EQUAL, 2, NOT_ROW_DIMENSION)
+{}
+
+ObExprSubQueryNSEqual::~ObExprSubQueryNSEqual()
+{}
+
+int ObExprSubQueryNSEqual::compare_single_row(
+    const ObNewRow& left_row, const ObNewRow& right_row, ObExprCtx& expr_ctx, ObObj& result) const
+{
+  int ret = OB_SUCCESS;
+  const int64_t left_size = left_row.get_count();
+  const int64_t right_size = right_row.get_count();
+  if (left_size != right_size) {
+    ret = OB_ERR_UNEXPECTED;
+    LOG_ERROR("projector size is not equal", K(left_size), K(right_size), K(ret));
+  } else {
+    const ObIArray<ObExprCalcType>& cmp_types = result_type_.get_row_calc_cmp_types();
+    ObCompareCtx cmp_ctx(ObMaxType, CS_TYPE_INVALID, true, expr_ctx.tz_offset_, default_null_pos());
+    EXPR_DEFINE_CAST_CTX(expr_ctx, CM_WARN_ON_FAIL);
+    for (int64_t i = 0; OB_SUCC(ret) && i < left_size; ++i) {
+      const ObObj& left_param = left_row.get_cell(i);
+      const ObObj& right_param = right_row.get_cell(i);
+      cmp_ctx.cmp_type_ = cmp_types.at(i).get_type();
+      cmp_ctx.cmp_cs_type_ = cmp_types.at(i).get_collation_type();
+      if (OB_FAIL(ObExprEqual::calc(result, left_param, right_param, cmp_ctx, cast_ctx))) {
+        LOG_WARN("Compare expression failed", K(i), K(ret));
+      } else if (result.is_false()) {
+        // when is not equal, should break immediately
+        break;
+      } else {
+      }
+    }
+  }
+  return ret;
+}
+}  // namespace sql
+}  // namespace oceanbase