Exec_query_scan.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 <NdbScanFilter.hpp>
  15. #include <common/StmtArea.hpp>
  16. #include <common/ResultArea.hpp>
  17. #include <codegen/Code_query_scan.hpp>
  18. #include <codegen/Code_table.hpp>
  19. #define startBuddyTransaction(x) hupp(x)
  20. void
  21. Exec_query_scan::execImpl(Ctx& ctx, Ctl& ctl)
  22. {
  23.     const Code& code = getCode();
  24.     Data& data = getData();
  25.     Ndb* const ndb = ndbObject();
  26.     NdbConnection* const tcon = ndbConnection();
  27.     if (data.m_con != 0) {
  28. data.m_con->stopScan();
  29. ndb->closeTransaction(data.m_con);
  30. data.m_con = 0;
  31. data.m_op = 0;
  32. ctx_log2(("scan closed at re-execute"));
  33.     }
  34.     data.m_con = ndb->startBuddyTransaction(tcon);
  35.     if (data.m_con == 0) {
  36. ctx.pushStatus(ndb, tcon, 0, "startBuddyTransaction");
  37. return;
  38.     }
  39.     data.m_op = data.m_con->getNdbOperation(code.m_tableName);
  40.     if (data.m_op == 0) {
  41. ctx.pushStatus(ndb, data.m_con, 0, "getNdbOperation");
  42. return;
  43.     }
  44.     if (! code.m_exclusive) {
  45. if (data.m_op->openScanReadCommitted(data.m_parallel) == -1) {
  46.     ctx.pushStatus(ndb, data.m_con, data.m_op, "openScanReadCommitted");
  47.     return;
  48. }
  49.     } else {
  50. if (data.m_op->openScanExclusive(data.m_parallel) == -1) {
  51.     ctx.pushStatus(ndb, data.m_con, data.m_op, "openScanExclusive");
  52.     return;
  53. }
  54.     }
  55.     if (m_interp == 0) {
  56. // XXX unnecessary
  57. if (data.m_op->interpret_exit_ok() == -1) {
  58.     ctx.pushStatus(ndb, data.m_con, data.m_op, "interpret_exit_ok");
  59.     return;
  60. }
  61.     } else {
  62. NdbScanFilter scanFilter(data.m_op);
  63. scanFilter.begin(NdbScanFilter::AND);
  64. ctl.m_scanFilter = &scanFilter;
  65. m_interp->execInterp(ctx, ctl);
  66. if (! ctx.ok())
  67.     return;
  68. scanFilter.end();
  69.     }
  70.     const SqlRow& sqlRow = data.sqlRow();
  71.     ctx_assert(sqlRow.count() == code.m_attrCount);
  72.     for (unsigned i = 1; i <= code.m_attrCount; i++) {
  73. const NdbAttrId attrId = code.m_attrId[i];
  74. SqlField& sqlField = sqlRow.getEntry(i);
  75. char* addr = static_cast<char*>(sqlField.addr());
  76. NdbRecAttr* recAttr = data.m_op->getValue(attrId, addr);
  77. if (recAttr == 0) {
  78.     ctx.pushStatus(ndb, data.m_con, data.m_op, "getValue attrId=%u", (unsigned)attrId);
  79.     return;
  80. }
  81. data.m_recAttr[i] = recAttr;
  82.     }
  83.     if (code.m_attrCount == 0) { // NDB requires one
  84. (void)data.m_op->getValue((NdbAttrId)0);
  85.     }
  86.     if (data.m_con->executeScan() == -1) {
  87. ctx.pushStatus(ndb, data.m_con, data.m_op, "executeScan");
  88. return;
  89.     }
  90.     ctx_log2(("scan %s [%08x] started", ! code.m_exclusive ? "read" : "exclusive", (unsigned)this));
  91.     ctl.m_scanOp = data.m_op;
  92. }
  93. bool
  94. Exec_query_scan::fetchImpl(Ctx &ctx, Ctl& ctl)
  95. {
  96.     const Code& code = getCode();
  97.     Data& data = getData();
  98.     Ndb* const ndb = ndbObject();
  99.     int ret = data.m_con->nextScanResult();
  100.     if (ret != 0) {
  101. if (ret == -1) {
  102.     ctx.pushStatus(ndb, data.m_con, data.m_op, "nextScanResult");
  103. }
  104. data.m_con->stopScan();
  105. ndb->closeTransaction(data.m_con);
  106. data.m_con = 0;
  107. data.m_op = 0;
  108. ctx_log2(("scan [%08x] closed at last row", (unsigned)this));
  109. return false;
  110.     }
  111.     // set null bits
  112.     const SqlRow& sqlRow = data.sqlRow();
  113.     ctx_assert(sqlRow.count() == code.m_attrCount);
  114.     for (unsigned i = 1; i <= code.m_attrCount; i++) {
  115. NdbRecAttr* recAttr = data.m_recAttr[i];
  116. int isNULL = recAttr->isNULL();
  117. SqlField& sqlField = sqlRow.getEntry(i);
  118. ctx_assert(isNULL == 0 || isNULL == 1);
  119. sqlField.sqlNull(isNULL == 1);
  120.     }
  121.     stmtArea().incTuplesFetched();
  122.     return true;
  123. }