Code_delete_lookup.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 <dictionary/DictTable.hpp>
  15. #include "Code_expr.hpp"
  16. #include "Code_delete_lookup.hpp"
  17. #include "Code_table.hpp"
  18. #include "Code_root.hpp"
  19. Plan_delete_lookup::~Plan_delete_lookup()
  20. {
  21. }
  22. Plan_base*
  23. Plan_delete_lookup::analyze(Ctx& ctx, Ctl& ctl)
  24. {
  25.     ctx_assert(m_query != 0);
  26.     m_query->analyze(ctx, ctl);
  27.     if (! ctx.ok())
  28. return 0;
  29.     ctx_assert(m_table != 0);
  30.     m_table->analyze(ctx, ctl);
  31.     if (! ctx.ok())
  32. return 0;
  33.     return this;
  34. }
  35. void
  36. Plan_delete_lookup::describe(Ctx& ctx)
  37. {
  38.     stmtArea().setFunction(ctx, "DELETE WHERE", SQL_DIAG_DELETE_WHERE);
  39. }
  40. Exec_base*
  41. Plan_delete_lookup::codegen(Ctx& ctx, Ctl& ctl)
  42. {   
  43.     // create code for the query
  44.     ctx_assert(m_query != 0);
  45.     Exec_query* execQuery = static_cast<Exec_query*>(m_query->codegen(ctx, ctl));
  46.     if (! ctx.ok())
  47. return 0;
  48.     ctx_assert(execQuery != 0);
  49.     // set up
  50.     ctx_assert(m_table != 0);
  51.     const BaseString& tableName = m_table->getName();
  52.     const DictTable& dictTable = m_table->dictTable();
  53.     const unsigned keyCount = dictTable.keyCount();
  54.     // create the code
  55.     Exec_delete_lookup::Code& code = *new Exec_delete_lookup::Code(keyCount);
  56.     code.m_tableName = strcpy(new char[tableName.length() + 1], tableName.c_str());
  57.     // key attributes
  58.     code.m_keyId = new NdbAttrId[1 + keyCount];
  59.     code.m_keyId[0] = (NdbAttrId)-1;
  60.     for (unsigned k = 1; k <= keyCount; k++) {
  61. const DictColumn* keyColumn = dictTable.getKey(k);
  62. const SqlType& sqlType = keyColumn->sqlType();
  63. SqlSpec sqlSpec(sqlType, SqlSpec::Physical);
  64. code.m_keySpecs.setEntry(k, sqlSpec);
  65. code.m_keyId[k] = keyColumn->getAttrId();
  66.     }
  67.     // matching expressions
  68.     const Plan_table::Index& index = m_table->m_indexList[0];
  69.     ctx_assert(index.m_keyFound);
  70.     const ExprVector& keyEq = index.m_keyEq;
  71.     ctx_assert(keyEq.size() == 1 + keyCount);
  72.     code.m_keyMatch = new Exec_expr* [1 + keyCount];
  73.     code.m_keyMatch[0] = 0;
  74.     for (unsigned k = 1; k <= keyCount; k++) {
  75. Plan_expr* expr = keyEq[k];
  76. Exec_expr* execExpr = static_cast<Exec_expr*>(expr->codegen(ctx, ctl));
  77. if (! ctx.ok())
  78.     return 0;
  79. ctx_assert(execExpr != 0);
  80. code.m_keyMatch[k] = execExpr;
  81.     }
  82.     // create the exec
  83.     Exec_delete_lookup* exec = new Exec_delete_lookup(ctl.m_execRoot);
  84.     ctl.m_execRoot->saveNode(exec);
  85.     exec->setCode(code);
  86.     exec->setQuery(execQuery);
  87.     return exec;
  88. }    
  89. void
  90. Plan_delete_lookup::print(Ctx& ctx)
  91. {
  92.     ctx.print(" [delete_lookup");
  93.     Plan_base* a[] = { m_query, m_table };
  94.     printList(ctx, a, 2);
  95.     ctx.print("]");
  96. }
  97. // Exec_delete_lookup
  98. Exec_delete_lookup::Code::~Code()
  99. {
  100.     delete[] m_tableName;
  101.     delete[] m_keyId;
  102.     delete[] m_keyMatch;
  103. }
  104. Exec_delete_lookup::Data::~Data()
  105. {
  106. }
  107. Exec_delete_lookup::~Exec_delete_lookup()
  108. {
  109. }
  110. void
  111. Exec_delete_lookup::alloc(Ctx& ctx, Ctl& ctl)
  112. {
  113.     const Code& code = getCode();
  114.     // allocate the subquery
  115.     ctx_assert(m_query != 0);
  116.     m_query->alloc(ctx, ctl);
  117.     if (! ctx.ok())
  118.         return;
  119.     // allocate matching expressions
  120.     for (unsigned k = 1; k <= code.m_keyCount; k++) {
  121. Exec_expr* expr = code.m_keyMatch[k];
  122. ctx_assert(expr != 0);
  123. expr->alloc(ctx, ctl);
  124. if (! ctx.ok())
  125.     return;
  126.     }
  127.     // create data
  128.     Data& data = *new Data;
  129.     setData(data);
  130. }
  131. void
  132. Exec_delete_lookup::close(Ctx& ctx)
  133. {
  134.     ctx_assert(m_query != 0);
  135.     m_query->close(ctx);
  136. }
  137. void
  138. Exec_delete_lookup::print(Ctx& ctx)
  139. {
  140.     const Code& code = getCode();
  141.     ctx.print(" [delete_lookup");
  142.     Exec_base* a[] = { m_query };
  143.     printList(ctx, a, 1);
  144.     printList(ctx, (Exec_base**)&code.m_keyMatch[1], code.m_keyCount);
  145.     ctx.print("]");
  146. }