| 1 | #include "duckdb/parser/statement/create_statement.hpp" |
| 2 | #include "duckdb/planner/operator/logical_create.hpp" |
| 3 | #include "duckdb/planner/operator/logical_create_table.hpp" |
| 4 | #include "duckdb/planner/operator/logical_create_index.hpp" |
| 5 | #include "duckdb/planner/operator/logical_get.hpp" |
| 6 | #include "duckdb/planner/parsed_data/bound_create_table_info.hpp" |
| 7 | #include "duckdb/catalog/catalog.hpp" |
| 8 | #include "duckdb/catalog/catalog_entry/schema_catalog_entry.hpp" |
| 9 | #include "duckdb/planner/binder.hpp" |
| 10 | #include "duckdb/planner/expression_binder/index_binder.hpp" |
| 11 | #include "duckdb/parser/parsed_data/create_view_info.hpp" |
| 12 | #include "duckdb/parser/parsed_data/create_index_info.hpp" |
| 13 | #include "duckdb/planner/bound_query_node.hpp" |
| 14 | #include "duckdb/planner/tableref/bound_basetableref.hpp" |
| 15 | |
| 16 | using namespace duckdb; |
| 17 | using namespace std; |
| 18 | |
| 19 | SchemaCatalogEntry *Binder::BindSchema(CreateInfo &info) { |
| 20 | if (info.schema == INVALID_SCHEMA) { |
| 21 | info.schema = info.temporary ? TEMP_SCHEMA : DEFAULT_SCHEMA; |
| 22 | } |
| 23 | |
| 24 | if (!info.temporary) { |
| 25 | // non-temporary create: not read only |
| 26 | if (info.schema == TEMP_SCHEMA) { |
| 27 | throw ParserException("Only TEMPORARY table names can use the \"temp\" schema" ); |
| 28 | } |
| 29 | this->read_only = false; |
| 30 | } else { |
| 31 | if (info.schema != TEMP_SCHEMA) { |
| 32 | throw ParserException("TEMPORARY table names can *only* use the \"%s\" schema" , TEMP_SCHEMA); |
| 33 | } |
| 34 | } |
| 35 | // fetch the schema in which we want to create the object |
| 36 | auto schema_obj = Catalog::GetCatalog(context).GetSchema(context, info.schema); |
| 37 | assert(schema_obj->type == CatalogType::SCHEMA); |
| 38 | info.schema = schema_obj->name; |
| 39 | return schema_obj; |
| 40 | } |
| 41 | |
| 42 | BoundStatement Binder::Bind(CreateStatement &stmt) { |
| 43 | BoundStatement result; |
| 44 | result.names = {"Count" }; |
| 45 | result.types = {SQLType::BIGINT}; |
| 46 | |
| 47 | auto catalog_type = stmt.info->type; |
| 48 | switch (catalog_type) { |
| 49 | case CatalogType::SCHEMA: |
| 50 | result.plan = make_unique<LogicalCreate>(LogicalOperatorType::CREATE_SCHEMA, move(stmt.info)); |
| 51 | break; |
| 52 | case CatalogType::VIEW: { |
| 53 | auto &base = (CreateViewInfo &)*stmt.info; |
| 54 | // bind the schema |
| 55 | auto schema = BindSchema(*stmt.info); |
| 56 | |
| 57 | // bind the view as if it were a query so we can catch errors |
| 58 | // note that we bind a copy and don't actually use the bind result |
| 59 | auto copy = base.query->Copy(); |
| 60 | auto query_node = Bind(*copy); |
| 61 | if (base.aliases.size() > query_node.names.size()) { |
| 62 | throw BinderException("More VIEW aliases than columns in query result" ); |
| 63 | } |
| 64 | // fill up the aliases with the remaining names of the bound query |
| 65 | for (idx_t i = base.aliases.size(); i < query_node.names.size(); i++) { |
| 66 | base.aliases.push_back(query_node.names[i]); |
| 67 | } |
| 68 | base.types = query_node.types; |
| 69 | result.plan = make_unique<LogicalCreate>(LogicalOperatorType::CREATE_VIEW, move(stmt.info), schema); |
| 70 | break; |
| 71 | } |
| 72 | case CatalogType::SEQUENCE: { |
| 73 | auto schema = BindSchema(*stmt.info); |
| 74 | result.plan = make_unique<LogicalCreate>(LogicalOperatorType::CREATE_SEQUENCE, move(stmt.info), schema); |
| 75 | break; |
| 76 | } |
| 77 | case CatalogType::INDEX: { |
| 78 | auto &base = (CreateIndexInfo &)*stmt.info; |
| 79 | |
| 80 | // visit the table reference |
| 81 | auto bound_table = Bind(*base.table); |
| 82 | if (bound_table->type != TableReferenceType::BASE_TABLE) { |
| 83 | throw BinderException("Can only delete from base table!" ); |
| 84 | } |
| 85 | // bind the index expressions |
| 86 | vector<unique_ptr<Expression>> expressions; |
| 87 | IndexBinder binder(*this, context); |
| 88 | for (auto &expr : base.expressions) { |
| 89 | expressions.push_back(binder.Bind(expr)); |
| 90 | } |
| 91 | |
| 92 | auto plan = CreatePlan(*bound_table); |
| 93 | if (plan->type != LogicalOperatorType::GET) { |
| 94 | throw BinderException("Cannot create index on a view!" ); |
| 95 | } |
| 96 | auto &get = (LogicalGet &)*plan; |
| 97 | for (auto &column_id : get.column_ids) { |
| 98 | if (column_id == COLUMN_IDENTIFIER_ROW_ID) { |
| 99 | throw BinderException("Cannot create an index on the rowid!" ); |
| 100 | } |
| 101 | } |
| 102 | // this gives us a logical table scan |
| 103 | // we take the required columns from here |
| 104 | // create the logical operator |
| 105 | result.plan = make_unique<LogicalCreateIndex>(*get.table, get.column_ids, move(expressions), |
| 106 | unique_ptr_cast<CreateInfo, CreateIndexInfo>(move(stmt.info))); |
| 107 | break; |
| 108 | } |
| 109 | case CatalogType::TABLE: { |
| 110 | auto bound_info = BindCreateTableInfo(move(stmt.info)); |
| 111 | auto root = move(bound_info->query); |
| 112 | |
| 113 | // create the logical operator |
| 114 | auto create_table = make_unique<LogicalCreateTable>(bound_info->schema, move(bound_info)); |
| 115 | if (root) { |
| 116 | create_table->children.push_back(move(root)); |
| 117 | } |
| 118 | result.plan = move(create_table); |
| 119 | return result; |
| 120 | } |
| 121 | default: |
| 122 | throw Exception("Unrecognized type!" ); |
| 123 | } |
| 124 | return result; |
| 125 | } |
| 126 | |