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

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 "mgmapi.h"
  14. #include <string.h>
  15. #include <NdbMain.h>
  16. #include <OutputStream.hpp>
  17. #include <NdbOut.hpp>
  18. #include <NdbSleep.h>
  19. #include <getarg.h>
  20. #include <NdbApi.hpp>
  21. #include <NDBT.hpp>
  22. #include <NDBT_Test.hpp>
  23. #include <HugoTransactions.hpp>
  24. #include <UtilTransactions.hpp>
  25. #include <ConfigRetriever.hpp>
  26. #include <ndb_version.h>
  27. #define CHECK(b) if (!(b)) { 
  28.   g_err << "ERR: "<< "getStep" 
  29.          << " failed on line " << __LINE__ << endl; 
  30.   result = NDBT_FAILED; 
  31.   continue; } 
  32. int main(int argc, const char** argv){
  33.   ndb_init();
  34.   const char * connectString = NULL;
  35.   const char * table = NULL;
  36.   int records = 0;
  37.   int _help = 0;
  38.   
  39.   struct getargs args[] = {
  40.     { "connectString", 'c', arg_string, &connectString, 
  41.       "ConnectString", "nodeid=<api id>;host=<hostname:port>" },
  42.     { "tableName", 't', arg_string, &table, 
  43.       "table", "Table" },
  44.     { "records", 'r', arg_integer, &records, "Number of records", "recs"},
  45.     { "usage", '?', arg_flag, &_help, "Print help", "" }    
  46.   };
  47.   int num_args = sizeof(args) / sizeof(args[0]);
  48.   int optind = 0;
  49.   char desc[] = 
  50.     "hostname:portn"
  51.     "This program will connect to the mgmsrv of a NDB cluster.n"
  52.     "It will then wait for all nodes to be started, then restart node(s)n"
  53.     "and wait for all to restart inbetween. It will do this n"
  54.     "loop number of timesn";
  55.   
  56.   if(getarg(args, num_args, argc, argv, &optind) || _help) {
  57.     arg_printusage(args, num_args, argv[0], desc);
  58.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  59.   }
  60.   ndbout_c("table %s connectStirng %s", table, connectString);  
  61.   if(connectString == 0)
  62.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  63.   if(table == 0) 
  64.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  65.   Ndb * m_ndb = new Ndb("");
  66.   m_ndb->useFullyQualifiedNames(false);
  67.   m_ndb->setConnectString(connectString);
  68.   /**
  69.    * @todo  Set proper max no of transactions?? needed?? Default 12??
  70.    */
  71.   m_ndb->init(2048);
  72.   if (m_ndb->waitUntilReady() != 0){
  73.     ndbout_c("NDB Cluster not ready for connections");
  74.   }
  75.   int count = 0;
  76.   int result = NDBT_OK;
  77.   const NdbDictionary::Table * tab =  NDBT_Table::discoverTableFromDb( m_ndb, table);
  78. //  ndbout << *tab << endl;
  79.   UtilTransactions utilTrans(*tab);
  80.   HugoTransactions hugoTrans(*tab);
  81.   do{
  82.     // Check that there are as many records as we expected
  83.     CHECK(utilTrans.selectCount(m_ndb, 64, &count) == 0);
  84.     
  85.     g_err << "count = " << count;
  86.     g_err << " records = " << records;
  87.     g_err << endl;
  88.     CHECK(count == records);
  89.     
  90.     // Read and verify every record
  91.     CHECK(hugoTrans.pkReadRecords(m_ndb, records) == 0);
  92.   } while (false);
  93.   
  94.   return NDBT_ProgramExit(result);    
  95. }