mirror of
https://github.com/MariaDB/server.git
synced 2025-01-20 22:12:30 +01:00
119 lines
3.6 KiB
C++
119 lines
3.6 KiB
C++
/* Copyright (C) 2003 MySQL AB
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; version 2 of the License.
|
|
|
|
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 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
|
|
|
|
|
|
#include <NdbOut.hpp>
|
|
#include <NdbApi.hpp>
|
|
#include <NdbSleep.h>
|
|
#include <NDBT.hpp>
|
|
#include <HugoTransactions.hpp>
|
|
#include <getarg.h>
|
|
|
|
|
|
int main(int argc, const char** argv){
|
|
ndb_init();
|
|
|
|
int _records = 0;
|
|
int _help = 0;
|
|
int _batch = 512;
|
|
int _loops = -1;
|
|
int _rand = 0;
|
|
int _onetrans = 0;
|
|
int _abort = 0;
|
|
const char* db = 0;
|
|
|
|
struct getargs args[] = {
|
|
{ "records", 'r', arg_integer, &_records, "Number of records", "recs" },
|
|
{ "batch", 'b', arg_integer, &_batch, "Number of operations in each transaction", "batch" },
|
|
{ "loops", 'l', arg_integer, &_loops, "Number of loops", "" },
|
|
{ "database", 'd', arg_string, &db, "Database", "" },
|
|
{ "usage", '?', arg_flag, &_help, "Print help", "" },
|
|
{ "rnd-rows", 0, arg_flag, &_rand, "Rand number of records", "recs" },
|
|
{ "one-trans", 0, arg_flag, &_onetrans, "Insert as 1 trans", "" },
|
|
{ "abort", 0, arg_integer, &_abort, "Abort probability", "" }
|
|
};
|
|
int num_args = sizeof(args) / sizeof(args[0]);
|
|
int optind = 0;
|
|
char desc[] =
|
|
"tabname\n"\
|
|
"This program will load one table in Ndb with calculated data. \n"\
|
|
"This means that it is possible to check the validity of the data \n"\
|
|
"at a later time. The last column in each table is used as an update \n"\
|
|
"counter, it's initialised to zero and should be incremented for each \n"\
|
|
"update of the record. \n";
|
|
|
|
if(getarg(args, num_args, argc, argv, &optind) ||
|
|
argv[optind] == NULL || _records == 0 || _help) {
|
|
arg_printusage(args, num_args, argv[0], desc);
|
|
return NDBT_ProgramExit(NDBT_WRONGARGS);
|
|
}
|
|
|
|
|
|
// 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){
|
|
ERR(MyNdb.getNdbError());
|
|
return NDBT_ProgramExit(NDBT_FAILED);
|
|
}
|
|
|
|
for(Uint32 i = optind; i<argc; i++)
|
|
{
|
|
const char* _tabname = argv[i];
|
|
// 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);
|
|
loop:
|
|
int rows = (_rand ? rand() % _records : _records);
|
|
int abort = (rand() % 100) < _abort ? 1 : 0;
|
|
if (abort)
|
|
ndbout << "load+abort" << endl;
|
|
if (hugoTrans.loadTable(&MyNdb,
|
|
rows,
|
|
_batch,
|
|
true, 0, _onetrans, _loops, abort) != 0){
|
|
return NDBT_ProgramExit(NDBT_FAILED);
|
|
}
|
|
|
|
if(_loops > 0)
|
|
{
|
|
ndbout << "clearing..." << endl;
|
|
hugoTrans.clearTable(&MyNdb);
|
|
//hugoTrans.pkDelRecords(&MyNdb, _records);
|
|
_loops--;
|
|
goto loop;
|
|
}
|
|
}
|
|
|
|
return NDBT_ProgramExit(NDBT_OK);
|
|
}
|