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