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

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 <NDBT.hpp>
  14. #include <NDBT_Test.hpp>
  15. #include <HugoTransactions.hpp>
  16. #include <UtilTransactions.hpp>
  17. #include <NdbRestarter.hpp>
  18. #include <Vector.hpp>
  19. #include <random.h>
  20. int runLoadTable(NDBT_Context* ctx, NDBT_Step* step){
  21.   int records = ctx->getNumRecords();
  22.   HugoTransactions hugoTrans(*ctx->getTab());
  23.   if (hugoTrans.loadTable(GETNDB(step), records) != 0){
  24.     return NDBT_FAILED;
  25.   }
  26.   return NDBT_OK;
  27. }
  28. int runClearTable(NDBT_Context* ctx, NDBT_Step* step){
  29.   int records = ctx->getNumRecords();
  30.   
  31.   UtilTransactions utilTrans(*ctx->getTab());
  32.   if (utilTrans.clearTable2(GETNDB(step),  records) != 0){
  33.     return NDBT_FAILED;
  34.   }
  35.   return NDBT_OK;
  36. }
  37. int create_index_on_pk(Ndb* pNdb, const char* tabName){
  38.   int result  = NDBT_OK;
  39.   const NdbDictionary::Table * tab = NDBT_Table::discoverTableFromDb(pNdb,
  40.      tabName);
  41.   // Create index      
  42.   const char* idxName = "IDX_ON_PK";
  43.   ndbout << "Create: " <<idxName << "( ";
  44.   NdbDictionary::Index pIdx(idxName);
  45.   pIdx.setTable(tabName);
  46.   pIdx.setType(NdbDictionary::Index::UniqueHashIndex);
  47.   for (int c = 0; c< tab->getNoOfPrimaryKeys(); c++){    
  48.     pIdx.addIndexColumn(tab->getPrimaryKey(c));
  49.     ndbout << tab->getPrimaryKey(c)<<" ";
  50.   }
  51.   
  52.   ndbout << ") ";
  53.   if (pNdb->getDictionary()->createIndex(pIdx) != 0){
  54.     ndbout << "FAILED!" << endl;
  55.     const NdbError err = pNdb->getDictionary()->getNdbError();
  56.     ERR(err);
  57.     result = NDBT_FAILED;
  58.   } else {
  59.     ndbout << "OK!" << endl;
  60.   }
  61.   return result;
  62. }
  63. int drop_index_on_pk(Ndb* pNdb, const char* tabName){
  64.   int result = NDBT_OK;
  65.   const char* idxName = "IDX_ON_PK";
  66.   ndbout << "Drop: " << idxName;
  67.   if (pNdb->getDictionary()->dropIndex(idxName, tabName) != 0){
  68.     ndbout << "FAILED!" << endl;
  69.     const NdbError err = pNdb->getDictionary()->getNdbError();
  70.     ERR(err);
  71.     result = NDBT_FAILED;
  72.   } else {
  73.     ndbout << "OK!" << endl;
  74.   }
  75.   return result;
  76. }
  77. #define CHECK(b) if (!(b)) { 
  78.   g_err << "ERR: "<< step->getName() 
  79.          << " failed on line " << __LINE__ << endl; 
  80.   result = NDBT_FAILED; 
  81.   continue; } 
  82. int runTestSingleUserMode(NDBT_Context* ctx, NDBT_Step* step){
  83.   int result = NDBT_OK;
  84.   int loops = ctx->getNumLoops();
  85.   int records = ctx->getNumRecords();
  86.   Ndb* pNdb = GETNDB(step);
  87.   NdbRestarter restarter;
  88.   char tabName[255];
  89.   strncpy(tabName, ctx->getTab()->getName(), 255);
  90.   ndbout << "tabName="<<tabName<<endl;
  91.   
  92.   int i = 0;
  93.   int count;
  94.   HugoTransactions hugoTrans(*ctx->getTab());
  95.   UtilTransactions utilTrans(*ctx->getTab());
  96.   while (i<loops && result == NDBT_OK) {
  97.     g_info << i << ": ";
  98.     int timeout = 120;
  99.     // Test that the single user mode api can do everything     
  100.     CHECK(restarter.enterSingleUserMode(pNdb->getNodeId()) == 0);
  101.     CHECK(restarter.waitClusterSingleUser(timeout) == 0); 
  102.     CHECK(hugoTrans.loadTable(pNdb, records, 128) == 0);
  103.     CHECK(hugoTrans.pkReadRecords(pNdb, records) == 0);
  104.     CHECK(hugoTrans.pkUpdateRecords(pNdb, records) == 0);
  105.     CHECK(utilTrans.selectCount(pNdb, 64, &count) == 0);
  106.     CHECK(count == records);
  107.     CHECK(hugoTrans.pkDelRecords(pNdb, records/2) == 0);
  108.     CHECK(hugoTrans.scanReadRecords(pNdb, records/2, 0, 64) == 0);
  109.     CHECK(utilTrans.selectCount(pNdb, 64, &count) == 0);
  110.     CHECK(count == (records/2));
  111.     CHECK(utilTrans.clearTable(pNdb, records/2) == 0);
  112.     CHECK(restarter.exitSingleUserMode() == 0);
  113.     CHECK(restarter.waitClusterStarted(timeout) == 0); 
  114.     // Test create index in single user mode
  115.     CHECK(restarter.enterSingleUserMode(pNdb->getNodeId()) == 0);
  116.     CHECK(restarter.waitClusterSingleUser(timeout) == 0); 
  117.     CHECK(create_index_on_pk(pNdb, tabName) == 0);
  118.     CHECK(hugoTrans.loadTable(pNdb, records, 128) == 0);
  119.     CHECK(hugoTrans.pkReadRecords(pNdb, records) == 0);
  120.     CHECK(hugoTrans.pkUpdateRecords(pNdb, records) == 0);
  121.     CHECK(utilTrans.selectCount(pNdb, 64, &count) == 0);
  122.     CHECK(count == records);
  123.     CHECK(hugoTrans.pkDelRecords(pNdb, records/2) == 0);
  124.     CHECK(drop_index_on_pk(pNdb, tabName) == 0);   
  125.     CHECK(restarter.exitSingleUserMode() == 0);
  126.     CHECK(restarter.waitClusterStarted(timeout) == 0); 
  127.     // Test recreate index in single user mode
  128.     CHECK(create_index_on_pk(pNdb, tabName) == 0);
  129.     CHECK(hugoTrans.loadTable(pNdb, records, 128) == 0);
  130.     CHECK(utilTrans.selectCount(pNdb, 64, &count) == 0);
  131.     CHECK(restarter.enterSingleUserMode(pNdb->getNodeId()) == 0);
  132.     CHECK(restarter.waitClusterSingleUser(timeout) == 0); 
  133.     CHECK(drop_index_on_pk(pNdb, tabName) == 0);
  134.     CHECK(utilTrans.selectCount(pNdb, 64, &count) == 0);
  135.     CHECK(create_index_on_pk(pNdb, tabName) == 0);
  136.     CHECK(restarter.exitSingleUserMode() == 0);
  137.     CHECK(restarter.waitClusterStarted(timeout) == 0); 
  138.     CHECK(drop_index_on_pk(pNdb, tabName) == 0);
  139.     CHECK(utilTrans.clearTable(GETNDB(step),  records) == 0);
  140.     ndbout << "Restarting cluster" << endl;
  141.     CHECK(restarter.restartAll() == 0);
  142.     CHECK(restarter.waitClusterStarted(timeout) == 0);
  143.     CHECK(pNdb->waitUntilReady(timeout) == 0);
  144.     i++;
  145.   }
  146.   return result;
  147. }
  148. NDBT_TESTSUITE(testMgm);
  149. TESTCASE("SingleUserMode", 
  150.  "Test single user mode"){
  151.   INITIALIZER(runTestSingleUserMode);
  152.   FINALIZER(runClearTable);
  153. }
  154. NDBT_TESTSUITE_END(testMgm);
  155. int main(int argc, const char** argv){
  156.   ndb_init();
  157.   myRandom48Init(NdbTick_CurrentMillisecond());
  158.   return testMgm.execute(argc, argv);
  159. }