OmniSciDB  d2f719934e
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
PlanState.h
Go to the documentation of this file.
1 /*
2  * Copyright 2019 OmniSci, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <unordered_set>
20 
21 #include "Analyzer/Analyzer.h"
24 
25 class Executor;
26 
27 struct JoinInfo {
28  JoinInfo(const std::vector<std::shared_ptr<Analyzer::BinOper>>& equi_join_tautologies,
29  const std::vector<std::shared_ptr<HashJoin>>& join_hash_tables)
30  : equi_join_tautologies_(equi_join_tautologies)
31  , join_hash_tables_(join_hash_tables) {}
32 
33  std::vector<std::shared_ptr<Analyzer::BinOper>>
34  equi_join_tautologies_; // expressions we equi-join on are true by
35  // definition when using a hash join; we'll
36  // fold them to true during code generation
37  std::vector<std::shared_ptr<HashJoin>> join_hash_tables_;
38  std::unordered_set<size_t> sharded_range_table_indices_;
39 };
40 
41 struct PlanState {
42  using TableId = int;
43  using ColumnId = int;
44  using DeletedColumnsMap = std::unordered_map<TableId, const ColumnDescriptor*>;
45  using HoistedFiltersSet = std::unordered_set<std::shared_ptr<Analyzer::Expr>>;
46 
47  PlanState(const bool allow_lazy_fetch,
48  const std::vector<InputTableInfo>& query_infos,
49  const DeletedColumnsMap& deleted_columns,
50  const Executor* executor)
51  : allow_lazy_fetch_(allow_lazy_fetch)
52  , join_info_({std::vector<std::shared_ptr<Analyzer::BinOper>>{}, {}})
53  , deleted_columns_(deleted_columns)
54  , query_infos_(query_infos)
55  , executor_(executor) {}
56 
57  std::vector<int64_t> init_agg_vals_;
58  std::vector<Analyzer::Expr*> target_exprs_;
60  std::unordered_map<InputColDescriptor, size_t> global_to_local_col_ids_;
61  std::set<std::pair<TableId, ColumnId>> columns_to_fetch_;
62  std::set<std::pair<TableId, ColumnId>> columns_to_not_fetch_;
63  std::unordered_map<size_t, std::vector<std::shared_ptr<Analyzer::Expr>>>
66  JoinInfo join_info_;
68  const std::vector<InputTableInfo>& query_infos_;
69  std::list<std::shared_ptr<Analyzer::Expr>> simple_quals_;
70  const Executor* executor_;
71 
73  const std::list<std::shared_ptr<const InputColDescriptor>>& global_col_ids);
74 
75  int getLocalColumnId(const Analyzer::ColumnVar* col_var, const bool fetch_column);
76 
77  bool isLazyFetchColumn(const Analyzer::Expr* target_expr) const;
78 
79  bool isLazyFetchColumn(const InputColDescriptor& col_desc) {
81  col_desc.getScanDesc().getTableId(),
82  col_desc.getColId(),
83  col_desc.getScanDesc().getNestLevel());
84  return isLazyFetchColumn(&column);
85  }
86 
88  auto deleted_cols_it = deleted_columns_.find(table_id);
89  if (deleted_cols_it != deleted_columns_.end()) {
90  return deleted_cols_it->second;
91  }
92  return nullptr;
93  }
94 
95  void addSimpleQual(std::shared_ptr<Analyzer::Expr> simple_qual) {
96  simple_quals_.push_back(simple_qual);
97  }
98 
99  std::list<std::shared_ptr<Analyzer::Expr>> getSimpleQuals() const {
100  return simple_quals_;
101  }
102 
103  void addNonHashtableQualForLeftJoin(size_t idx, std::shared_ptr<Analyzer::Expr> expr);
104 };
Defines data structures for the semantic analysis phase of query processing.
JoinInfo join_info_
Definition: PlanState.h:66
std::list< std::shared_ptr< Analyzer::Expr > > simple_quals_
Definition: PlanState.h:69
std::unordered_map< size_t, std::vector< std::shared_ptr< Analyzer::Expr > > > left_join_non_hashtable_quals_
Definition: PlanState.h:64
std::unordered_map< InputColDescriptor, size_t > global_to_local_col_ids_
Definition: PlanState.h:60
bool allow_lazy_fetch_
Definition: PlanState.h:65
std::set< std::pair< TableId, ColumnId > > columns_to_fetch_
Definition: PlanState.h:61
std::unordered_set< std::shared_ptr< Analyzer::Expr >> HoistedFiltersSet
Definition: PlanState.h:45
JoinInfo(const std::vector< std::shared_ptr< Analyzer::BinOper >> &equi_join_tautologies, const std::vector< std::shared_ptr< HashJoin >> &join_hash_tables)
Definition: PlanState.h:28
void addNonHashtableQualForLeftJoin(size_t idx, std::shared_ptr< Analyzer::Expr > expr)
Definition: PlanState.cpp:77
std::unordered_set< size_t > sharded_range_table_indices_
Definition: PlanState.h:38
const Executor * executor_
Definition: PlanState.h:70
std::vector< int64_t > init_agg_vals_
Definition: PlanState.h:57
std::set< std::pair< TableId, ColumnId > > columns_to_not_fetch_
Definition: PlanState.h:62
executor_(executor)
Definition: PlanState.h:55
std::vector< std::shared_ptr< Analyzer::BinOper > > equi_join_tautologies_
Definition: PlanState.h:34
int getLocalColumnId(const Analyzer::ColumnVar *col_var, const bool fetch_column)
Definition: PlanState.cpp:62
bool isLazyFetchColumn(const Analyzer::Expr *target_expr) const
Definition: PlanState.cpp:20
std::unordered_map< TableId, const ColumnDescriptor * > DeletedColumnsMap
Definition: PlanState.h:44
int getColId() const
int getTableId() const
bool isLazyFetchColumn(const InputColDescriptor &col_desc)
Definition: PlanState.h:79
PlanState(const bool allow_lazy_fetch, const std::vector< InputTableInfo > &query_infos, const DeletedColumnsMap &deleted_columns, const Executor *executor)
Definition: PlanState.h:47
const std::vector< InputTableInfo > & query_infos_
Definition: PlanState.h:68
specifies the content in-memory of a row in the column metadata table
void allocateLocalColumnIds(const std::list< std::shared_ptr< const InputColDescriptor >> &global_col_ids)
Definition: PlanState.cpp:50
Executor(const ExecutorId id, Data_Namespace::DataMgr *data_mgr, const size_t block_size_x, const size_t grid_size_x, const size_t max_gpu_slab_size, const std::string &debug_dir, const std::string &debug_file)
Definition: Execute.cpp:156
int TableId
Definition: PlanState.h:42
const ColumnDescriptor * getDeletedColForTable(const TableId table_id)
Definition: PlanState.h:87
void addSimpleQual(std::shared_ptr< Analyzer::Expr > simple_qual)
Definition: PlanState.h:95
const DeletedColumnsMap deleted_columns_
Definition: PlanState.h:67
HoistedFiltersSet hoisted_filters_
Definition: PlanState.h:59
std::list< std::shared_ptr< Analyzer::Expr > > getSimpleQuals() const
Definition: PlanState.h:99
std::vector< Analyzer::Expr * > target_exprs_
Definition: PlanState.h:58
std::vector< std::shared_ptr< HashJoin > > join_hash_tables_
Definition: PlanState.h:37
int ColumnId
Definition: PlanState.h:43
int getNestLevel() const
const InputDescriptor & getScanDesc() const