Exec_query_index.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 <NdbApi.hpp>
  14. #include <common/StmtArea.hpp>
  15. #include <common/ResultArea.hpp>
  16. #include <codegen/Code_expr.hpp>
  17. #include <codegen/Code_query_index.hpp>
  18. #include <codegen/Code_table.hpp>
  19. void
  20. Exec_query_index::execImpl(Ctx& ctx, Ctl& ctl)
  21. {
  22.     const Code& code = getCode();
  23.     Data& data = getData();
  24.     data.m_done = false;
  25.     Ndb* const ndb = ndbObject();
  26.     NdbConnection* const tcon = ndbConnection();
  27.     if (data.m_con != 0) {
  28. ndb->closeTransaction(data.m_con);
  29. data.m_con = 0;
  30. data.m_op = 0;
  31. ctx_log2(("lookup closed at re-execute"));
  32.     }
  33.     const bool unco = connArea().uncommitted();
  34.     if (! unco) {
  35. // use new transaction to not run out of operations
  36. data.m_con = ndb->startTransaction();
  37. if (data.m_con == 0) {
  38.     ctx.pushStatus(ndb, "startTransaction");
  39.     return;
  40. }
  41.     } else {
  42. ctx_log3(("lookup using main transaction"));
  43.     }
  44.     data.m_op = (unco ? tcon : data.m_con)->getNdbIndexOperation(code.m_indexName, code.m_tableName);
  45.     if (data.m_op == 0) {
  46. ctx.pushStatus(ndb, (unco ? tcon : data.m_con), 0, "getNdbIndexOperation");
  47. return;
  48.     }
  49.     if (data.m_op->readTuple() == -1) {
  50. ctx.pushStatus(ndb, (unco ? tcon : data.m_con), data.m_op, "readTuple");
  51. return;
  52.     }
  53.     // key attributes
  54.     for (unsigned k = 1; k <= code.m_keyCount; k++) {
  55. Exec_expr* exprMatch = code.m_keyMatch[k];
  56. ctx_assert(exprMatch != 0);
  57. exprMatch->evaluate(ctx, ctl);
  58. if (! ctx.ok())
  59.     return;
  60. const SqlField& keyMatch = exprMatch->getData().sqlField();
  61. SqlField f(code.m_keySpecs.getEntry(k));
  62. if (! keyMatch.cast(ctx, f)) {
  63.     data.m_done = true; // match is not possible
  64.     return;
  65. }
  66. const NdbAttrId keyId = code.m_keyId[k];
  67. const void* addr = f.addr();
  68. const char* value = static_cast<const char*>(addr);
  69. if (data.m_op->equal(keyId, value) == -1) {
  70.     ctx.pushStatus(ndb, (unco ? tcon : data.m_con), data.m_op, "equal attrId=%u", (unsigned)keyId);
  71.     return;
  72. }
  73.     }
  74.     // queried attributes
  75.     const SqlRow& sqlRow = data.sqlRow();
  76.     ctx_assert(sqlRow.count() == code.m_attrCount);
  77.     for (unsigned i = 1; i <= code.m_attrCount; i++) {
  78. const NdbAttrId attrId = code.m_attrId[i];
  79. SqlField& f = sqlRow.getEntry(i);
  80. char* addr = static_cast<char*>(f.addr());
  81. NdbRecAttr* recAttr = data.m_op->getValue(attrId, addr);
  82. if (recAttr == 0) {
  83.     ctx.pushStatus(ndb, (unco ? tcon : data.m_con), data.m_op, "getValue attrId=%u", (unsigned)attrId);
  84.     return;
  85. }
  86. data.m_recAttr[i] = recAttr;
  87.     }
  88.     if (code.m_attrCount == 0) { // NDB requires one
  89. (void)data.m_op->getValue((NdbAttrId)0);
  90.     }
  91.     data.setCount(0);
  92.     if ((unco ? tcon : data.m_con)->execute(unco ? NoCommit : Commit) == -1) {
  93. // XXX when did 626 move to connection level
  94. if ((unco ? tcon : data.m_con)->getNdbError().code != 626 && data.m_op->getNdbError().code != 626) {
  95.     ctx.pushStatus(ndb, (unco ? tcon : data.m_con), data.m_op, "execute xxx");
  96.     return;
  97. }
  98. data.m_done = true;
  99.     } else {
  100. stmtArea().incTuplesFetched();
  101. data.m_done = false;
  102.     }
  103.     if (! unco) {
  104. ndb->closeTransaction(data.m_con);
  105. data.m_con = 0;
  106. data.m_op = 0;
  107. ctx_log3(("index lookup closed at execute"));
  108.     }
  109. }
  110. bool
  111. Exec_query_index::fetchImpl(Ctx &ctx, Ctl& ctl)
  112. {
  113.     const Code& code = getCode();
  114.     Data& data = getData();
  115.     // returns at most one row
  116.     if (data.m_done)
  117. return false;
  118.     // set null bits
  119.     const SqlRow& sqlRow = data.sqlRow();
  120.     ctx_assert(sqlRow.count() == code.m_attrCount);
  121.     for (unsigned i = 1; i <= code.m_attrCount; i++) {
  122. NdbRecAttr* recAttr = data.m_recAttr[i];
  123. int isNULL = recAttr->isNULL();
  124. SqlField& f = sqlRow.getEntry(i);
  125. ctx_assert(isNULL == 0 || isNULL == 1);
  126. f.sqlNull(isNULL == 1);
  127.     }
  128.     data.m_done = true;
  129.     return true;
  130. }