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

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_update_scan.hpp>
  17. #include <codegen/Code_query.hpp>
  18. void
  19. Exec_update_scan::execImpl(Ctx& ctx, Ctl& ctl)
  20. {
  21.     const Code& code = getCode();
  22.     Data& data = getData();
  23.     Ndb* const ndb = ndbObject();
  24.     NdbConnection* const tcon = ndbConnection();
  25.     // execute subquery
  26.     ctx_assert(m_query != 0);
  27.     m_query->execute(ctx, ctl);
  28.     if (! ctx.ok())
  29. return;
  30.     ctx_assert(ctl.m_scanOp != 0);
  31.     // update each row from query
  32.     data.setCount(0);
  33.     while (m_query->fetch(ctx, ctl)) {
  34. NdbOperation* toOp = ctl.m_scanOp->takeOverForUpdate(tcon);
  35. if (toOp == 0) {
  36.     ctx.pushStatus(ndb, tcon, ctl.m_scanOp, "takeOverScanOp");
  37.     return;
  38. }
  39. const SqlRow& sqlRow = m_query->getData().sqlRow();
  40. for (unsigned i = 1; i <= sqlRow.count(); i++) {
  41.     const SqlField& f = sqlRow.getEntry(i);
  42.     const void* addr = f.sqlNull() ? 0 : f.addr();
  43.     const char* value = static_cast<const char*>(addr);
  44.     if (toOp->setValue(code.m_attrId[i], value) == -1) {
  45. ctx.pushStatus(ndb, tcon, toOp, "setValue");
  46. return;
  47.     }
  48. }
  49. if (tcon->execute(NoCommit) == -1) {
  50.     ctx.pushStatus(ndb, tcon, toOp, "execute without commit");
  51.     return;
  52. }
  53. data.addCount();
  54.     }
  55.     stmtArea().setRowCount(ctx, data.getCount());
  56. }