hugoFill.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 <ndb_global.h>
  14. #include <NdbOut.hpp>
  15. #include <NdbApi.hpp>
  16. #include <NdbSleep.h>
  17. #include <NDBT.hpp>
  18. #include <HugoTransactions.hpp>
  19. #include <getarg.h>
  20. int main(int argc, const char** argv){
  21.   ndb_init();
  22.   int _records = 0;
  23.   const char* _tabname = NULL;
  24.   int _help = 0;
  25.   int _batch = 512;
  26.   
  27.   struct getargs args[] = {
  28.     { "batch", 'b', arg_integer, &_batch, "Number of operations in each transaction", "batch" },
  29.     { "usage", '?', arg_flag, &_help, "Print help", "" }
  30.   };
  31.   int num_args = sizeof(args) / sizeof(args[0]);
  32.   int optind = 0;
  33.   char desc[] = 
  34.     "tabnamen"
  35.     "This program will load one table in Ndb with calculated data n"
  36.     "until the database is full. n";
  37.   
  38.   if(getarg(args, num_args, argc, argv, &optind) ||
  39.      argv[optind] == NULL  || _help) {
  40.     arg_printusage(args, num_args, argv[0], desc);
  41.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  42.   }
  43.   _tabname = argv[optind];
  44.   
  45.   // Connect to Ndb
  46.   Ndb MyNdb( "TEST_DB" );
  47.   if(MyNdb.init() != 0){
  48.     ERR(MyNdb.getNdbError());
  49.     return NDBT_ProgramExit(NDBT_FAILED);
  50.   }
  51.   // Connect to Ndb and wait for it to become ready
  52.   while(MyNdb.waitUntilReady() != 0)
  53.     ndbout << "Waiting for ndb to become ready..." << endl;
  54.    
  55.   // Check if table exists in db
  56.   const NdbDictionary::Table* pTab = NDBT_Table::discoverTableFromDb(&MyNdb, _tabname);
  57.   if(pTab == NULL){
  58.     ndbout << " Table " << _tabname << " does not exist!" << endl;
  59.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  60.   }
  61.   HugoTransactions hugoTrans(*pTab);
  62.   if (hugoTrans.fillTable(&MyNdb, 
  63.   _batch) != 0){
  64.     return NDBT_ProgramExit(NDBT_FAILED);
  65.   }
  66.   return NDBT_ProgramExit(NDBT_OK);
  67. }