|
15 | 15 | #include "parser/alter_statement.h"
|
16 | 16 | namespace peloton {
|
17 | 17 |
|
| 18 | +namespace parser { |
| 19 | +class AlterTableStatement; |
| 20 | +} |
| 21 | +namespace catalog { |
| 22 | +class Schema; |
| 23 | +} |
| 24 | +namespace storage { |
| 25 | +class DataTable; |
| 26 | +} |
| 27 | + |
18 | 28 | namespace planner {
|
19 | 29 | /** @brief The plan used for altering
|
20 | 30 | * TODO: adding support for add/drop column
|
21 | 31 | */
|
22 | 32 | class AlterPlan : public AbstractPlan {
|
23 | 33 | public:
|
24 | 34 | AlterPlan() = delete;
|
25 |
| - explicit AlterPlan(UNUSED_ATTRIBUTE parser::AlterTableStatement *parse_tree) { |
26 |
| - LOG_TRACE("%s", parse_tree->GetInfo().c_str()); |
27 |
| - } |
| 35 | + |
| 36 | + explicit AlterPlan(const std::string &database_name, |
| 37 | + const std::string &table_name, |
| 38 | + //std::unique_ptr<catalog::Schema> added_columns, |
| 39 | + const std::vector<std::string> &dropped_columns, |
| 40 | + AlterType a_type); |
| 41 | + explicit AlterPlan(const std::string &database_name, |
| 42 | + const std::string &table_name, |
| 43 | + const std::vector<std::string> &old_names, |
| 44 | + const std::vector<std::string> &new_names, AlterType a_type); |
| 45 | + explicit AlterPlan(parser::AlterTableStatement *parse_tree); |
| 46 | + |
28 | 47 | virtual ~AlterPlan() {}
|
29 | 48 |
|
30 |
| - virtual PlanNodeType GetPlanNodeType() const override { |
31 |
| - return PlanNodeType::ALTER; |
| 49 | + virtual PlanNodeType GetPlanNodeType() const { return PlanNodeType::ALTER; } |
| 50 | + |
| 51 | + virtual std::unique_ptr<AbstractPlan> Copy() { return nullptr; } |
| 52 | + |
| 53 | + const std::string GetInfo() const { |
| 54 | + return StringUtil::Format("AlterPlan table:%s, database:%s", |
| 55 | + this->table_name.c_str(), this->database_name.c_str()); |
| 56 | + } |
| 57 | + |
| 58 | + std::unique_ptr<AbstractPlan> Copy() const { |
| 59 | + switch(this->type) { |
| 60 | + case AlterType::DROP: |
| 61 | + case AlterType::ADD: |
| 62 | + return std::unique_ptr<AbstractPlan>( |
| 63 | + new AlterPlan(database_name, table_name, |
| 64 | + //std::unique_ptr<catalog::Schema>( |
| 65 | + // catalog::Schema::CopySchema(added_columns)), |
| 66 | + dropped_columns, type)); |
| 67 | + case AlterType::RENAME: |
| 68 | + return std::unique_ptr<AbstractPlan>( |
| 69 | + new AlterPlan(database_name, table_name, old_names_, new_names_, type)); |
| 70 | + default: |
| 71 | + LOG_ERROR("Not supported Copy of Alter type yet"); |
| 72 | + return nullptr; |
| 73 | + } |
32 | 74 | }
|
33 | 75 |
|
34 |
| - virtual std::unique_ptr<AbstractPlan> Copy() const override { return nullptr; } |
| 76 | + std::string GetTableName() const { return table_name; } |
| 77 | + |
| 78 | + std::string GetDatabaseName() const { return database_name; } |
| 79 | + |
| 80 | + //catalog::Schema *GetAddedColumns() const { return added_columns; } |
| 81 | + |
| 82 | + const std::vector<std::string> &GetDroppedColumns() const { |
| 83 | + return dropped_columns; |
| 84 | + } |
| 85 | + |
| 86 | + AlterType GetAlterTableType() const { return type; } |
| 87 | + |
| 88 | + //function used for rename statement |
| 89 | + std::string GetOldName() const { return this->old_names_[0]; } |
| 90 | + |
| 91 | + //function used for rename statement |
| 92 | + std::string GetNewName() const { return this->new_names_[0]; } |
| 93 | + |
| 94 | + //return true if the alter plan is rename statement |
| 95 | + bool IsRename() const { return this->type==AlterType::RENAME;} |
| 96 | +private: |
| 97 | +// Target Table |
| 98 | + storage::DataTable *target_table_ = nullptr; |
| 99 | + |
| 100 | + // Table Name |
| 101 | + std::string table_name; |
| 102 | + |
| 103 | + // Database Name |
| 104 | + std::string database_name; |
| 105 | + |
| 106 | + // Schema delta, define the column txn want to add |
| 107 | + // catalog::Schema *added_columns; |
| 108 | + // dropped_column, define the column you want to drop |
| 109 | + std::vector<std::string> dropped_columns; |
| 110 | + |
| 111 | + //used for store rename function data |
| 112 | + std::vector<std::string> old_names_; |
| 113 | + std::vector<std::string> new_names_; |
| 114 | + |
| 115 | + // Check to either AlterTable Table, INDEX or Rename |
| 116 | + AlterType type; |
| 117 | + |
35 | 118 | };
|
36 | 119 | }
|
37 | 120 | }
|
0 commit comments