/**

* 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_TEMP_TABLE_

#define OCEANBASE_SQL_TEMP_TABLE_

#include "sql/resolver/dml/ob_dml_stmt.h"

#include "sql/resolver/expr/ob_raw_expr.h"

#include "sql/optimizer/ob_sharding_info.h"

#include "observer/ob_server_struct.h"

namespace oceanbase

{

namespace sql

{

struct ObTempTableResultInfo

{

OB_UNIS_VERSION(1);

public:

ObTempTableResultInfo() : addr_(),

interm_result_ids_() {}

virtual ~ObTempTableResultInfo() {}

TO_STRING_KV(K_(addr),

K_(interm_result_ids));

//数据所在server

ObAddr addr_;

//数据集的key

ObSEArray<uint64_t, 2> interm_result_ids_;

};

class ObSqlTempTableCtx

{

OB_UNIS_VERSION(1);

public:

ObSqlTempTableCtx() : interm_result_infos_(),

temp_table_id_(0),

is_local_interm_result_(true) {}

virtual ~ObSqlTempTableCtx() {}

TO_STRING_KV(K_(interm_result_infos),

K_(temp_table_id),

K_(is_local_interm_result));

//结果集的分布信息：所在机器及KEY

ObSEArray<ObTempTableResultInfo, 2> interm_result_infos_;

//结果集所属的temp table

uint64_t temp_table_id_;

//结果集是否在本地

bool is_local_interm_result_;

};

typedef ObIArray<ObRawExpr *> ObRawExprCondition;

struct TableInfo {

TableInfo()

:table_filters_(),

table_item_(NULL),

upper_stmt_(NULL)

{}

virtual ~TableInfo(){}

TO_STRING_KV(

K_(table_filters),

K_(table_item),

K_(upper_stmt)

);

ObSEArray<ObRawExpr *, 4, common::ModulePageAllocator, true> table_filters_;

ObSEArray<ObRawExprCondition *, 4, common::ModulePageAllocator, true> filter_conditions_;

TableItem *table_item_;

ObDMLStmt* upper_stmt_;

};

class ObSqlTempTableInfo

{

public:

ObSqlTempTableInfo() : temp_table_id_(OB_INVALID_ID),

table_name_(),

table_query_(NULL),

table_plan_(NULL) {}

virtual ~ObSqlTempTableInfo() {}

void reset() {

temp_table_id_ = OB_INVALID_ID;

table_name_.reset();

table_query_ = NULL;

table_plan_ = NULL;

table_infos_.reset();

}

TO_STRING_KV(K_(temp_table_id),

K_(table_name),

K_(table_infos));

static int collect_temp_tables(ObIAllocator &allocator,

ObDMLStmt &stmt,

ObIArray<ObSqlTempTableInfo*> &temp_table_infos,

ObQueryCtx *query_ctx,

bool do_collect_filter);

static int collect_specified_temp_table(ObIAllocator &allocator,

ObSelectStmt *specified_query,

const ObIArray<ObDMLStmt *> &upper_stmts,

const ObIArray<TableItem *> &table_items,

ObSqlTempTableInfo &temp_table_info,

bool &all_has_filter);

static int collect_temp_table_filters(ObDMLStmt *stmt,

TableItem *table,

ObIArray<ObRawExpr*> &table_filters,

ObIArray<ObRawExprCondition*> &filter_conditions);

static int collect_table_filters_in_joined_table(JoinedTable *table,

uint64_t table_id,

const ObSqlBitSet<> &table_ids,

ObIArray<ObRawExpr*> &table_filters,

ObIArray<ObRawExprCondition*> &filter_conditions);

static int get_candi_exprs(const ObSqlBitSet<> &table_ids,

ObIArray<ObRawExpr*> &exprs,

ObIArray<ObRawExpr*> &candi_exprs,

ObIArray<ObRawExprCondition*> &candi_conditions);

public:

uint64_t temp_table_id_;

common::ObString table_name_;

ObSelectStmt *table_query_;

ObLogicalOperator *table_plan_;

ObSEArray<TableInfo, 8, common::ModulePageAllocator, true> table_infos_;

};

} /* ns sql*/

} /* ns oceanbase */

#endif //OCEANBASE_SQL_TEMP_TABLE_