init push

src/sql/optimizer/ob_table_partition_info.h

@@ -0,0 +1,143 @@
+/**
+ * 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_OB_TABLE_PARTITION_INFO_
+#define OCEANBASE_SQL_OB_TABLE_PARTITION_INFO_
+
+#include "sql/optimizer/ob_phy_table_location_info.h"
+#include "sql/optimizer/ob_table_location.h"
+#include "sql/ob_phy_table_location.h"
+
+namespace oceanbase {
+namespace sql {
+
+class ObTablePartitionInfo {
+  public:
+  ObTablePartitionInfo()
+      : inner_allocator_(common::ObModIds::OB_SQL_TABLE_LOCATION),
+        allocator_(inner_allocator_),
+        table_location_array_(NULL),
+        table_location_array_size_(0),
+        best_location_idx_(OB_INVALID_INDEX)
+  {}
+  ObTablePartitionInfo(common::ObIAllocator& allocator)
+      : allocator_(allocator),
+        table_location_(allocator),
+        table_location_array_(NULL),
+        table_location_array_size_(0),
+        best_location_idx_(OB_INVALID_INDEX)
+  {}
+  virtual ~ObTablePartitionInfo()
+  {}
+
+  int assign(const ObTablePartitionInfo& other);
+
+  int init_table_location(ObSqlSchemaGuard& schema_guard, ObDMLStmt& stmt, ObSQLSessionInfo* session_info,
+      const common::ObIArray<ObRawExpr*>& filter_exprs, const uint64_t table_id, const uint64_t ref_table_id,
+      // const uint64_t index_table_id,
+      const ObPartHint* part_hint, const common::ObDataTypeCastParams& dtc_params, const bool is_dml_table,
+      common::ObIArray<ObRawExpr*>* sort_exprs = NULL);
+
+  int init_table_location_with_rowid(ObSqlSchemaGuard& schema_guard, const common::ObIArray<int64_t>& param_idx,
+      const uint64_t table_id, const uint64_t ref_table_id, ObSQLSessionInfo& session_info, const bool is_dml_table);
+
+  int get_not_insert_dml_part_sort_expr(ObDMLStmt& stmt, common::ObIArray<ObRawExpr*>* sort_exprs) const;
+
+  int calculate_phy_table_location_info(ObExecContext& exec_ctx, common::ObPartMgr* part_mgr, const ParamStore& params,
+      share::ObIPartitionLocationCache& location_cache, const common::ObDataTypeCastParams& dtc_params);
+
+  int calculate_phy_table_location_info(ObExecContext& exec_ctx, ObPartMgr* part_mgr, const ParamStore& params,
+      share::ObIPartitionLocationCache& location_cache, const common::ObDataTypeCastParams& dtc_params,
+      const common::ObIArray<int64_t>& partition_ids);
+
+  int calculate_phy_table_location_info(ObExecContext& exec_ctx, share::schema::ObSchemaGetterGuard* schema_guard,
+      const ParamStore& params, share::ObIPartitionLocationCache& location_cache);
+
+  int calc_phy_table_loc_and_select_leader(ObExecContext& exec_ctx, common::ObPartMgr* part_mgr,
+      const ParamStore& params, share::ObIPartitionLocationCache& location_cache,
+      const common::ObDataTypeCastParams& dtc_params, const uint64_t index_table_id, const ObOrderDirection& direction);
+
+  int calc_phy_table_loc_and_select_fixed_location(ObExecContext& exec_ctx, const common::ObAddr& fixed_server,
+      common::ObPartMgr* part_mgr, const ParamStore& params, share::ObIPartitionLocationCache& location_cache,
+      const common::ObDataTypeCastParams& dtc_params, const uint64_t index_table_id, const ObOrderDirection& direction);
+
+  int set_log_op_infos(const uint64_t index_table_id, const ObOrderDirection& direction);
+
+  int set_table_location_direction(const ObOrderDirection& direction);
+  int fill_phy_tbl_loc_info_direction();
+  int get_location_type(const common::ObAddr& server, ObTableLocationType& type) const;
+  int get_all_local_partition_key(ObIArray<ObPartitionKey>& pkeys, const common::ObAddr& server) const;
+  int get_all_servers(common::ObIArray<common::ObAddr>& servers) const;
+
+  ObTableLocation& get_table_location()
+  {
+    return table_location_;
+  }
+
+  const ObTableLocation& get_table_location() const
+  {
+    return table_location_;
+  }
+
+  const ObPhyTableLocationInfo& get_phy_tbl_location_info() const
+  {
+    return phy_tbl_location_info_;
+  }
+  ObPhyTableLocationInfo& get_phy_tbl_location_info_for_update()
+  {
+    return phy_tbl_location_info_;
+  }
+  /*ObPhyTableLocation &get_phy_table_location()
+  {
+    return phy_tbl_location_;
+  }*/
+
+  uint64_t get_table_id() const
+  {
+    return table_location_.get_table_id();
+  }
+  uint64_t get_ref_table_id() const
+  {
+    return table_location_.get_ref_table_id();
+  }
+  share::schema::ObPartitionLevel get_part_level() const
+  {
+    return table_location_.get_part_level();
+  }
+  TO_STRING_KV(K_(table_location), K_(phy_tbl_location_info));
+
+  private:
+  DISALLOW_COPY_AND_ASSIGN(ObTablePartitionInfo);
+
+  private:
+  /**
+   * ObTableLocation is the structure we stored to calculate the physical table location,
+   * which is represented by ObPhyTableLocation, for a given table(logical) in a given statement.
+   *
+   * The info we keep in ObTableLocation is constant for a parameterized query among multiple
+   * times of execution, which is stored to short-cut to plan matching and location calculation
+   * since we can skip a bunch of steps in query processing, such as parsing, resolving,
+   * optimization and code generation.
+   */
+  common::ObArenaAllocator inner_allocator_;
+  ObIAllocator& allocator_;
+  ObTableLocation table_location_;
+  ObTableLocation* table_location_array_;
+  int64_t table_location_array_size_;
+  int64_t best_location_idx_;
+  ObPhyTableLocationInfo phy_tbl_location_info_;
+  // ObPhyTableLocation phy_tbl_location_;
+};
+
+}  // namespace sql
+}  // namespace oceanbase
+#endif  // OCEANBASE_SQL_OB_TABLE_PARTITION_INFO_