Code_create_table.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #include <common/StmtArea.hpp>
  14. #include "Code_create_table.hpp"
  15. #include "Code_root.hpp"
  16. // Plan_create_table
  17. Plan_create_table::~Plan_create_table()
  18. {
  19. }
  20. Plan_base*
  21. Plan_create_table::analyze(Ctx& ctx, Ctl& ctl)
  22. {
  23.     stmtArea().stmtInfo().setName(Stmt_name_create_table);
  24.     // analyze the create row
  25.     ctx_assert(m_createRow != 0);
  26.     m_createRow->analyze(ctx, ctl);
  27.     if (! ctx.ok())
  28. return 0;
  29.     return this;
  30. }
  31. void
  32. Plan_create_table::describe(Ctx& ctx)
  33. {
  34.     stmtArea().setFunction(ctx, "CREATE TABLE", SQL_DIAG_CREATE_TABLE);
  35. }
  36. Exec_base*
  37. Plan_create_table::codegen(Ctx& ctx, Ctl& ctl)
  38. {
  39.     ctx_assert(m_createRow != 0);
  40.     Exec_create_table* exec = new Exec_create_table(ctl.m_execRoot);
  41.     ctl.m_execRoot->saveNode(exec);
  42.     const unsigned count = m_createRow->countColumn();
  43.     Exec_create_table::Code::Attr* attrList = new Exec_create_table::Code::Attr[1 + count];
  44.     unsigned tupleId = 0;
  45.     unsigned autoIncrement = 0;
  46.     for (unsigned i = 1; i <= count; i++) {
  47. Plan_ddl_column* column = m_createRow->getColumn(i);
  48. Exec_create_table::Code::Attr& attr = attrList[i];
  49. attr.m_attrName.assign(column->getName());
  50. attr.m_sqlType = column->sqlType();
  51. attr.m_tupleKey = column->getPrimaryKey();
  52. attr.m_tupleId = column->getTupleId();
  53. attr.m_autoIncrement = column->getAutoIncrement();
  54. if (attr.m_tupleId)
  55.     tupleId = i;
  56. if (attr.m_autoIncrement)
  57.     autoIncrement = i;
  58. attr.m_defaultValue = 0;
  59. Plan_expr* expr;
  60. if ((expr = column->getDefaultValue()) != 0) {
  61.     Exec_expr* execExpr = static_cast<Exec_expr*>(expr->codegen(ctx, ctl));
  62.     if (! ctx.ok())
  63. return 0;
  64.     ctx_assert(execExpr != 0);
  65.     attr.m_defaultValue = execExpr;
  66. }
  67.     }
  68.     Exec_create_table::Code& code = *new Exec_create_table::Code(m_name, count, attrList, tupleId, autoIncrement);
  69.     exec->setCode(code);
  70.     code.m_fragmentType = m_fragmentType;
  71.     code.m_logging = m_logging;
  72.     return exec;
  73. }
  74. void
  75. Plan_create_table::print(Ctx& ctx)
  76. {
  77.     ctx.print(" [create_table '%s'", m_name.c_str());
  78.     Plan_base* a[] = { m_createRow };
  79.     printList(ctx, a, 1);
  80.     ctx.print("]");
  81. }
  82. // Exec_create_table
  83. Exec_create_table::Code::~Code()
  84. {
  85.     delete[] m_attrList;
  86. }
  87. Exec_create_table::Data::~Data()
  88. {
  89. }
  90. Exec_create_table::~Exec_create_table()
  91. {
  92. }
  93. void
  94. Exec_create_table::alloc(Ctx& ctx, Ctl& ctl)
  95. {
  96.     const Code& code = getCode();
  97.     for (unsigned i = 1; i <= code.m_attrCount; i++) {
  98. const Code::Attr& attr = code.m_attrList[i];
  99. if (attr.m_defaultValue != 0)
  100.     attr.m_defaultValue->alloc(ctx, ctl);
  101.     }
  102.     Data& data = *new Data;
  103.     setData(data);
  104. }
  105. void
  106. Exec_create_table::close(Ctx& ctx)
  107. {
  108.     const Code& code = getCode();
  109.     for (unsigned i = 1; i <= code.m_attrCount; i++) {
  110. const Code::Attr& attr = code.m_attrList[i];
  111. if (attr.m_defaultValue != 0)
  112.     attr.m_defaultValue->close(ctx);
  113.     }
  114. }
  115. void
  116. Exec_create_table::print(Ctx& ctx)
  117. {
  118.     const Code& code = getCode();
  119.     ctx.print(" [create_table %s]", code.m_tableName.c_str());
  120. }