hugoScanRead.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 <ndb_global.h>
  14. #include <NdbOut.hpp>
  15. #include <NdbApi.hpp>
  16. #include <NdbMain.h>
  17. #include <NDBT.hpp> 
  18. #include <NdbSleep.h>
  19. #include <getarg.h>
  20. #include <HugoTransactions.hpp>
  21. int main(int argc, const char** argv){
  22.   ndb_init();
  23.   int _records = 0;
  24.   int _loops = 1;
  25.   int _abort = 0;
  26.   int _parallelism = 1;
  27.   const char* _tabname = NULL, *db = 0;
  28.   int _help = 0;
  29.   int lock = NdbOperation::LM_Read;
  30.   int sorted = 0;
  31.   struct getargs args[] = {
  32.     { "aborts", 'a', arg_integer, &_abort, "percent of transactions that are aborted", "abort%" },
  33.     { "loops", 'l', arg_integer, &_loops, "number of times to run this program(0=infinite loop)", "loops" },
  34.     { "parallelism", 'p', arg_integer, &_parallelism, "parallelism(1-240)", "para" },
  35.     { "records", 'r', arg_integer, &_records, "Number of records", "recs" },
  36.     { "usage", '?', arg_flag, &_help, "Print help", "" },
  37.     { "lock", 'm', arg_integer, &lock, "lock mode", "" },
  38.     { "sorted", 's', arg_flag, &sorted, "sorted", "" },
  39.     { "database", 'd', arg_string, &db, "Database", "" }
  40.   };
  41.   int num_args = sizeof(args) / sizeof(args[0]);
  42.   int optind = 0;
  43.   char desc[] = 
  44.     " tabnamen"
  45.     "This program will scan read all records in one table in Ndb.n"
  46.     "It will verify every column read by calculating the expected value.n";
  47.   
  48.   if(getarg(args, num_args, argc, argv, &optind) || argv[optind] == NULL || _help) {
  49.     arg_printusage(args, num_args, argv[0], desc);
  50.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  51.   }
  52.   _tabname = argv[optind];
  53.   // Connect to Ndb
  54.   Ndb MyNdb( db ? db : "TEST_DB" );
  55.   if(MyNdb.init() != 0){
  56.     ERR(MyNdb.getNdbError());
  57.     return NDBT_ProgramExit(NDBT_FAILED);
  58.   }
  59.   while(MyNdb.waitUntilReady() != 0)
  60.     ndbout << "Waiting for ndb to become ready..." << endl;
  61.    
  62.   // Check if table exists in db
  63.   const NdbDictionary::Table * pTab = NDBT_Table::discoverTableFromDb(&MyNdb, _tabname);
  64.   if(pTab == NULL){
  65.     ndbout << " Table " << _tabname << " does not exist!" << endl;
  66.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  67.   }
  68.   const NdbDictionary::Index * pIdx = 0;
  69.   if(optind+1 < argc)
  70.   {
  71.     pIdx = MyNdb.getDictionary()->getIndex(argv[optind+1], _tabname);
  72.     if(!pIdx)
  73.       ndbout << " Index " << argv[optind+1] << " not found" << endl;
  74.     else
  75.       if(pIdx->getType() != NdbDictionary::Index::UniqueOrderedIndex &&
  76.  pIdx->getType() != NdbDictionary::Index::OrderedIndex)
  77.       {
  78. ndbout << " Index " << argv[optind+1] << " is not scannable" << endl;
  79. pIdx = 0;
  80.       }
  81.   }
  82.   
  83.   HugoTransactions hugoTrans(*pTab);
  84.   int i = 0;
  85.   while (i<_loops || _loops==0) {
  86.     ndbout << i << ": ";
  87.     if(!pIdx)
  88.     {
  89.       if(hugoTrans.scanReadRecords(&MyNdb, 
  90.    0,
  91.    _abort,
  92.    _parallelism,
  93.    (NdbOperation::LockMode)lock) != 0)
  94.       {
  95. return NDBT_ProgramExit(NDBT_FAILED);
  96.       }
  97.     }
  98.     else
  99.     {
  100.       if(hugoTrans.scanReadRecords(&MyNdb, pIdx, 
  101.    0,
  102.    _abort,
  103.    _parallelism,
  104.    (NdbOperation::LockMode)lock,
  105.    sorted) != 0)
  106.       {
  107. return NDBT_ProgramExit(NDBT_FAILED);
  108.       }
  109.     }
  110.     i++;
  111.   }
  112.   return NDBT_ProgramExit(NDBT_OK);
  113. }