122 lines
3.8 KiB
C++
122 lines
3.8 KiB
C++
/*
|
|
Copyright (c) 2003, 2016, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License, version 2.0, for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <ndb_global.h>
|
|
|
|
#include <NdbOut.hpp>
|
|
|
|
#include <NdbApi.hpp>
|
|
#include <NDBT.hpp>
|
|
#include <NdbSleep.h>
|
|
#include <getarg.h>
|
|
|
|
#include <HugoTransactions.hpp>
|
|
|
|
int main(int argc, const char** argv){
|
|
ndb_init();
|
|
|
|
int _records = 0;
|
|
int _loops = 1;
|
|
int _parallelism = 1;
|
|
int _ver2 = 0;
|
|
const char* _tabname = NULL, *db = 0;
|
|
int _help = 0;
|
|
int abort= 0;
|
|
|
|
struct getargs args[] = {
|
|
{ "loops", 'l', arg_integer, &_loops, "number of times to run this program(0=infinite loop)", "loops" },
|
|
{ "parallelism", 'p', arg_integer, &_parallelism, "parallelism(1-240)", "para" },
|
|
{ "records", 'r', arg_integer, &_records, "Number of records", "recs" },
|
|
{ "ver2", '2', arg_flag, &_ver2, "Use version 2 of scanUpdateRecords", "" },
|
|
{ "ver2", '1', arg_negative_flag, &_ver2, "Use version 1 of scanUpdateRecords (default)", "" },
|
|
{ "abort", 'a', arg_integer, &abort, "Abort probability", "" },
|
|
{ "usage", '?', arg_flag, &_help, "Print help", "" },
|
|
{ "database", 'd', arg_string, &db, "Database", "" }
|
|
};
|
|
int num_args = sizeof(args) / sizeof(args[0]);
|
|
int optind = 0;
|
|
char desc[] =
|
|
"tabname\n"\
|
|
"This program will scan update all records in one table in Ndb\n";
|
|
|
|
if(getarg(args, num_args, argc, argv, &optind) ||
|
|
argv[optind] == NULL || _help) {
|
|
arg_printusage(args, num_args, argv[0], desc);
|
|
return NDBT_ProgramExit(NDBT_WRONGARGS);
|
|
}
|
|
_tabname = argv[optind];
|
|
|
|
// Connect to Ndb
|
|
Ndb_cluster_connection con;
|
|
if(con.connect(12, 5, 1) != 0)
|
|
{
|
|
return NDBT_ProgramExit(NDBT_FAILED);
|
|
}
|
|
|
|
if (con.wait_until_ready(30,0) < 0)
|
|
{
|
|
ndbout << "Cluster nodes not ready in 30 seconds." << endl;
|
|
return NDBT_ProgramExit(NDBT_FAILED);
|
|
}
|
|
|
|
Ndb MyNdb( &con, db ? db : "TEST_DB" );
|
|
|
|
if(MyNdb.init() != 0){
|
|
NDB_ERR(MyNdb.getNdbError());
|
|
return NDBT_ProgramExit(NDBT_FAILED);
|
|
}
|
|
|
|
// Check if table exists in db
|
|
const NdbDictionary::Table * pTab = NDBT_Table::discoverTableFromDb(&MyNdb, _tabname);
|
|
if(pTab == NULL){
|
|
ndbout << " Table " << _tabname << " does not exist!" << endl;
|
|
return NDBT_ProgramExit(NDBT_WRONGARGS);
|
|
}
|
|
|
|
HugoTransactions hugoTrans(*pTab);
|
|
int i = 0;
|
|
int res = NDBT_FAILED;
|
|
while (i<_loops || _loops==0) {
|
|
ndbout << i << ": ";
|
|
if (_ver2 == 0){
|
|
res = hugoTrans.scanUpdateRecords(&MyNdb,
|
|
_records,
|
|
abort % 101,
|
|
_parallelism);
|
|
} else{
|
|
res = hugoTrans.scanUpdateRecords2(&MyNdb,
|
|
_records,
|
|
abort % 101,
|
|
_parallelism);
|
|
}
|
|
if (res != NDBT_OK ){
|
|
return NDBT_ProgramExit(NDBT_FAILED);
|
|
}
|
|
i++;
|
|
//NdbSleep_MilliSleep(300);
|
|
}
|
|
|
|
return NDBT_ProgramExit(NDBT_OK);
|
|
}
|