init push

src/sql/engine/expr/ob_expr_obj_access.h

@@ -0,0 +1,98 @@
+/**
+ * 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 DEV_SRC_SQL_ENGINE_EXPR_OB_EXPR_OBJ_ACCESS_H_
+#define DEV_SRC_SQL_ENGINE_EXPR_OB_EXPR_OBJ_ACCESS_H_
+#include "sql/engine/expr/ob_expr_operator.h"
+#include "lib/container/ob_fast_array.h"
+#include "lib/container/ob_2d_array.h"
+
+namespace oceanbase {
+using namespace common;
+namespace sql {
+typedef Ob2DArray<ObObjParam, OB_MALLOC_BIG_BLOCK_SIZE, ObWrapperAllocator, false> ParamStore;
+class ObExprObjAccess : public ObExprOperator {
+  OB_UNIS_VERSION(1);
+
+  public:
+  explicit ObExprObjAccess(common::ObIAllocator& alloc);
+  virtual ~ObExprObjAccess();
+
+  virtual void reset();
+  int assign(const ObExprOperator& other);
+  inline int init_param_idxs(int64_t param_idx_cnt)
+  {
+    return param_idxs_.init(param_idx_cnt);
+  }
+  virtual int calc_resultN(
+      common::ObObj& result, const common::ObObj* objs_stack, int64_t param_num, common::ObExprCtx& expr_ctx) const;
+  inline void set_get_attr_func(uint64_t attr_func)
+  {
+    get_attr_func_ = attr_func;
+  }
+  inline int add_param_idx(int64_t param_idx)
+  {
+    return param_idxs_.push_back(param_idx);
+  }
+  inline const common::ObFixedArray<int64_t, common::ObIAllocator>& get_param_idxs() const
+  {
+    return param_idxs_;
+  }
+  inline void set_write(bool for_write)
+  {
+    for_write_ = for_write;
+  }
+  inline bool get_write() const
+  {
+    return for_write_;
+  }
+  inline void set_access_idx_cnt(int64_t cnt)
+  {
+    access_idx_cnt_ = cnt;
+  }
+  inline int64_t get_access_idx_cnt() const
+  {
+    return access_idx_cnt_;
+  }
+  inline void set_coll_idx(int64_t idx)
+  {
+    coll_idx_ = idx;
+  }
+  inline int64_t get_coll_idx()
+  {
+    return coll_idx_;
+  }
+  int calc_result(
+      common::ObObj& result, const common::ObObj* objs_stack, int64_t param_num, const ParamStore& param_store) const;
+
+  VIRTUAL_TO_STRING_KV(N_EXPR_TYPE, get_type_name(type_), N_EXPR_NAME, name_, N_PARAM_NUM, param_num_, N_DIM,
+      row_dimension_, N_REAL_PARAM_NUM, real_param_num_, K_(get_attr_func), K_(param_idxs), K_(access_idx_cnt),
+      K_(for_write), K_(coll_idx));
+
+  private:
+  typedef common::ObFastArray<int64_t, 4> ParamArray;
+  int init_param_array(
+      const ParamStore& param_store, const common::ObObj* objs_stack, int64_t param_num, ParamArray& param_array) const;
+
+  DISALLOW_COPY_AND_ASSIGN(ObExprObjAccess);
+
+  private:
+  uint64_t get_attr_func_;
+  common::ObFixedArray<int64_t, common::ObIAllocator> param_idxs_;
+  int64_t access_idx_cnt_;
+  bool for_write_;
+  int64_t coll_idx_;
+};
+
+}  // namespace sql
+}  // namespace oceanbase
+#endif /* DEV_SRC_SQL_ENGINE_EXPR_OB_EXPR_OBJ_ACCESS_H_ */