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

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/ConnArea.hpp>
  15. #include "DictCatalog.hpp"
  16. #include "DictSchema.hpp"
  17. #include "DictTable.hpp"
  18. #include "DictTable.hpp"
  19. #include "DictColumn.hpp"
  20. #include "DictIndex.hpp"
  21. #include "DictSys.hpp"
  22. DictSchema::~DictSchema()
  23. {
  24.     for (Tables::iterator i = m_tables.begin(); i != m_tables.end(); i++) {
  25. delete *i;
  26. *i = 0;
  27.     }
  28. }
  29. DictTable*
  30. DictSchema::findTable(const BaseString& name)
  31. {
  32.     for (Tables::iterator i = m_tables.begin(); i != m_tables.end(); i++) {
  33. DictTable* table = *i;
  34. ctx_assert(table != 0);
  35. if (strcmp(table->getName().c_str(), name.c_str()) == 0)
  36.     return table;
  37.     }
  38.     return 0;
  39. }
  40. void
  41. DictSchema::deleteTable(Ctx& ctx, const BaseString& name)
  42. {
  43.     Tables::iterator i = m_tables.begin();
  44.     while (i != m_tables.end()) {
  45. DictTable* table = *i;
  46. ctx_assert(table != 0);
  47. if (strcmp(table->getName().c_str(), name.c_str()) == 0) {
  48.     ctx_log2(("purge table %s from dictionary", name.c_str()));
  49.     delete table;
  50.     Tables::iterator j = i;
  51.     i++;
  52.     m_tables.erase(j);
  53.     break;
  54. }
  55. i++;
  56.     }
  57. }
  58. void
  59. DictSchema::deleteTableByIndex(Ctx& ctx, const BaseString& indexName)
  60. {
  61.     DictTable* foundTable = 0;
  62.     for (Tables::iterator i = m_tables.begin(); i != m_tables.end(); i++) {
  63. DictTable* table = *i;
  64. ctx_assert(table != 0);
  65. for (unsigned k = 1; k <= table->indexCount(); k++) {
  66.     const DictIndex* index = table->getIndex(k);
  67.     if (strcmp(index->getName().c_str(), indexName.c_str()) == 0) {
  68. foundTable = table;
  69. break;
  70.     }
  71. }
  72. if (foundTable != 0)
  73.     break;
  74.     }
  75.     if (foundTable != 0)
  76. deleteTable(ctx, foundTable->getName());
  77. }
  78. DictTable*
  79. DictSchema::loadTable(Ctx& ctx, const BaseString& name)
  80. {
  81.     ctx_log4(("%s: load from NDB", name.c_str()));
  82.     Ndb* ndb = m_connArea.ndbObject();
  83.     NdbDictionary::Dictionary* ndbDictionary = ndb->getDictionary();
  84.     if (ndbDictionary == 0) {
  85. ctx.pushStatus(ndb, "getDictionary");
  86.         return 0;
  87.     }
  88.     const NdbDictionary::Table* ndbTable = ndbDictionary->getTable(name.c_str());
  89.     if (ndbTable == 0) {
  90. const NdbError& ndbError = ndbDictionary->getNdbError();
  91. if (ndbError.code == 709) {
  92.     // try built-in system table
  93.     DictTable* table = DictSys::loadTable(ctx, this, name);
  94.     if (table != 0) {
  95. return table;
  96.     }
  97.     ctx_log3(("%s: not found in NDB", name.c_str()));
  98.     return 0;
  99. }
  100. ctx.pushStatus(ndbDictionary->getNdbError(), "getTable");
  101. return 0;
  102.     }
  103.     int nattr = ndbTable->getNoOfColumns();
  104.     DictTable* table = new DictTable(m_connArea, name, nattr);
  105.     for (unsigned position = 1; position <= (unsigned)nattr; position++) {
  106. DictColumn* column = table->loadColumn(ctx, position);
  107. if (column == 0)
  108.     return 0;
  109. ctx_log4(("add column %u %s", column->getPosition(), column->getName().c_str()));
  110.     }
  111.     // load indexes
  112.     NdbDictionary::Dictionary::List list;
  113.     if (ndbDictionary->listIndexes(list, name.c_str()) == -1) {
  114. ctx.pushStatus(ndbDictionary->getNdbError(), "listIndexes");
  115. return 0;
  116.     }
  117.     for (unsigned i = 0; i < list.count; i++) {
  118. const NdbDictionary::Dictionary::List::Element& elt = list.elements[i];
  119. if (elt.state != NdbDictionary::Object::StateOnline) {
  120.     ctx_log1(("%s: skip broken index %s", name.c_str(), elt.name));
  121.     continue;
  122. }
  123. if (elt.type != NdbDictionary::Object::UniqueHashIndex && elt.type != NdbDictionary::Object::OrderedIndex) {
  124.     ctx_log1(("%s: skip unknown index type %s", name.c_str(), elt.name));
  125.     continue;
  126. }
  127. const NdbDictionary::Index* ndbIndex = ndbDictionary->getIndex(elt.name, name.c_str());
  128. if (ndbIndex == 0) {
  129.     ctx.pushStatus(ndbDictionary->getNdbError(), "table %s getIndex %s", name.c_str(), elt.name);
  130.     return 0;
  131. }
  132. DictIndex* index = new DictIndex(m_connArea, elt.name, elt.type, ndbIndex->getNoOfIndexColumns());
  133. for (unsigned j = 0; j < index->getSize(); j++) {
  134.     const char* cname = ndbIndex->getIndexColumn(j);
  135.     ctx_assert(cname != 0);
  136.     DictColumn* icolumn = table->findColumn(cname);
  137.     ctx_assert(icolumn != 0);
  138.     index->setColumn(1 + j, icolumn);
  139. }
  140. table->addIndex(index);
  141. ctx_log3(("%s: index %s: load from NDB done", name.c_str(), elt.name));
  142.     }
  143.     addTable(table);
  144.     ctx_log3(("%s: load from NDB done", name.c_str()));
  145.     return table;
  146. }