/**
 * 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_)
  {}
  ObTablePartitionInfo(common::ObIAllocator &allocator)
  : allocator_(allocator),
  table_location_(allocator)
  {}
  virtual ~ObTablePartitionInfo() {}

  int assign(const ObTablePartitionInfo &other);

  int init_table_location(ObSqlSchemaGuard &schema_guard,
                          const ObDMLStmt &stmt,
                          ObExecContext *exec_ctx,
                          const common::ObIArray<ObRawExpr*> &filter_exprs,
                          const uint64_t table_id,
                          const uint64_t ref_table_id,
                          //const uint64_t index_table_id,
                          const common::ObIArray<ObObjectID> *part_ids,
                          const common::ObDataTypeCastParams &dtc_params,
                          const bool is_dml_table,
                          common::ObIArray<ObRawExpr *> *sort_exprs = NULL);

  int get_not_insert_dml_part_sort_expr(const ObDMLStmt &stmt,
                                        common::ObIArray<ObRawExpr *> *sort_exprs) const;

  int calculate_phy_table_location_info(ObExecContext &exec_ctx,
                                        const ParamStore &params,
                                        const common::ObDataTypeCastParams &dtc_params);

  int calc_phy_table_loc_and_select_leader(ObExecContext &exec_ctx,
                                           const ParamStore &params,
                                           const common::ObDataTypeCastParams &dtc_params);
  int replace_final_location_key(ObExecContext &exec_ctx, uint64_t ref_table_id, bool is_local_index);

  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_servers(common::ObIArray<common::ObAddr> &servers) const;

  ObTableLocation &get_table_location()
  {
    return table_location_;
  }

  const ObTableLocation &get_table_location() const
  {
    return table_location_;
  }

  const ObCandiTableLoc &get_phy_tbl_location_info() const
  {
    return candi_table_loc_;
  }
  ObCandiTableLoc &get_phy_tbl_location_info_for_update()
  {
    return candi_table_loc_;
  }

  void set_table_location(ObTableLocation &tbl)
  {
    table_location_ = tbl;
  }

  /*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_(candi_table_loc));

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_;
  ObCandiTableLoc candi_table_loc_;
  //ObPhyTableLocation phy_tbl_location_;
};

}
}
#endif // OCEANBASE_SQL_OB_TABLE_PARTITION_INFO_