diff options
Diffstat (limited to 'rasodmg/test/test_benchmark.cc')
| -rw-r--r-- | rasodmg/test/test_benchmark.cc | 273 |
1 files changed, 273 insertions, 0 deletions
diff --git a/rasodmg/test/test_benchmark.cc b/rasodmg/test/test_benchmark.cc new file mode 100644 index 0000000..2997d68 --- /dev/null +++ b/rasodmg/test/test_benchmark.cc @@ -0,0 +1,273 @@ +/* +* This file is part of rasdaman community. +* +* Rasdaman community 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, either version 3 of the License, or +* (at your option) any later version. +* +* Rasdaman community 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 rasdaman community. If not, see <http://www.gnu.org/licenses/>. +* +* Copyright 2003, 2004, 2005, 2006, 2007, 2008, 2009 Peter Baumann / +rasdaman GmbH. +* +* For more information please see <http://www.rasdaman.org> +* or contact Peter Baumann via <baumann@rasdaman.com>. +/ +/** + * SOURCE: test_benchmark.cc + * + * MODULE: rasodmg + * + * COMMENTS: + * None +*/ + +#ifdef EARLY_TEMPLATE +#define __EXECUTABLE__ +#include "raslib/template_inst.hh" +#endif + +#ifdef __VISUALC__ +#include <strstrea.h> +#else +#include <strstream.h> +#endif + +#include <iostream> +#include <fstream.h> +#include <string.h> + +#include "rasodmg/transaction.hh" +#include "rasodmg/database.hh" +#include "rasodmg/ref.hh" +#include "rasodmg/set.hh" +#include "rasodmg/marray.hh" +#include "rasodmg/iterator.hh" +#include "rasodmg/oqlquery.hh" + +#include "raslib/type.hh" +#include "raslib/shhopt.h" +#include "raslib/rmdebug.hh" + +#include <sys/time.h> + +static int numExec = 1; // number of repetitions for each query +static int readEach = 0; // number of repeated queries in query file +static int qNum = 0; // counter for current query + +RMTimer execTimer("test_benchmark", "r_oql_execute()"); +RMTimer wholeTimer("test_benchmark", "execQuery()"); + +int checkArguments( int argc, char** argv, const char* searchText, + int& optionValueIndex ) +{ + int found = 0; + int i=1; + + while( !found && i<argc ) + found = !strcmp( searchText, argv[i++] ); + + if( found && i<argc && !strchr(argv[i],'-') ) + optionValueIndex = i; + else + optionValueIndex = 0; + + return found; +} + +void execQuery( char* serverName, char* baseName, char* queryBuffer ) +{ +} + +static void +printUsage(void) +{ + cout << "Usage: test_benchmark [options] server_name base_name file_name" + << endl; + cout << " -h, --help Print this message and exit." << endl; + cout << " -n, --nrepeat Followed by number of repeated" << endl + << " executions for each query in file_name." + << endl; + cout << " -e, --readeach Followed by number of queries read from" + << endl + << " the query file treated as variations of" + << endl + << " the same query." << endl; + exit(0); +} + +void execQuery(char* serverName, char* baseName, char* comment, char* query) +{ + int i; + r_Database db; + r_Transaction ta; + + RMInit::bmOut << comment; + RMInit::bmOut << "Query " << (qNum/numExec)+1 << "." << (qNum%numExec)+1 + << ": " << query; + qNum++; + + wholeTimer.start(); + + r_Set< r_Ref_Any > image_set; + db.set_servername( serverName ); + + try + { + r_OQL_Query q1( query ); + + db.open( baseName ); + + if( q1.is_update_query() ) + ta.begin(); + else + ta.begin( r_Transaction::read_only ); + + // here the query is executed and times are taken + + try + { + execTimer.start(); + if( q1.is_update_query() ) + r_oql_execute( q1 ); + else + r_oql_execute( q1, image_set ); + execTimer.stop(); + } + + catch( r_Error& errorObj ) + { + execTimer.stop(); + cerr << endl << "QUERY FAILED" << endl << errorObj.what() << endl; + ta.commit(); + db.close(); + } + + ta.commit(); + db.close(); + + wholeTimer.stop(); + } + catch( r_Error& errorObj ) + { + wholeTimer.stop(); + cerr << endl << "FAILED" << endl << errorObj.what() << endl; + } +} + +void execQueries(char* serverName, char* baseName, char* comment, char* query) +{ + if(!readEach) { + for(int i = 0; i < numExec; i++) + execQuery(serverName, baseName, comment, query); + } + else { + execQuery(serverName, baseName, comment, query); + } +} + +void parseFile(ifstream& fileStream, char* serverName, char* baseName) +{ + enum legalStates { FIRSTCOMMENT, COMMENT, QUERY }; + legalStates parseState = FIRSTCOMMENT; + ostrstream commentStream; + ostrstream queryStream; + char buf[256]; + char dummy; + + while( fileStream.get( buf, 255, '\n' ) ) { + // read end of line + fileStream.get(dummy); + // checking for empty line + if(buf[0] == 0) { + if(parseState == FIRSTCOMMENT) { + // end of first comment, print it and switch to normal comment + commentStream << ends; + RMInit::bmOut << commentStream.str() << endl; + commentStream.rdbuf()->freeze(0); + commentStream.seekp(0, ios::beg); + parseState = COMMENT; + } + } + // checking for comment + else if(buf[0] == '/' && buf[1] == '/') { + if(parseState == QUERY) { + // execute the query, reset buffers + queryStream << endl << ends; + commentStream << ends; + execQueries(serverName, baseName, commentStream.str(), + queryStream.str()); + queryStream.rdbuf()->freeze(0); + queryStream.seekp(0, ios::beg); + commentStream.rdbuf()->freeze(0); + commentStream.seekp(0, ios::beg); + parseState = COMMENT; + } + commentStream << buf << endl; + } + // must be a query + else { + parseState = QUERY; + queryStream << buf; + queryStream << " "; + } + } + // last query + if(parseState == QUERY) { + // execute the query, reset buffers + queryStream << endl << ends; + commentStream << ends; + execQueries(serverName, baseName, commentStream.str(), + queryStream.str()); + queryStream.rdbuf()->freeze(0); + queryStream.seekp(0, ios::beg); + commentStream.rdbuf()->freeze(0); + commentStream.seekp(0, ios::beg); + } +} + +int main( int argc, char** argv ) +{ + int optionValueIndex; + char serverName[255]; + char baseName[255]; + char fileName[255]; + + optStruct testBenchmarkOpt[] = { + /* short long type var/func special */ + { 'h', "help", OPT_FLAG, printUsage, OPT_CALLFUNC }, + { 'n', "nrepeat", OPT_INT, &numExec, 0 }, + { 'e', "readeach", OPT_INT, &readEach, 0 }, + { 0, 0, OPT_END, 0, 0 } /* no more options */ + }; + + /* parse all options */ + optParseOptions(&argc, argv, testBenchmarkOpt, 0); + + if( argc < 4 ) + printUsage(); + + if(readEach) + numExec = readEach; + + strcpy( serverName, argv[argc-3] ); + strcpy( baseName, argv[argc-2] ); + strcpy( fileName, argv[argc-1] ); + + ifstream fileStream( fileName ); + if( !fileStream ) + { + cout << "Error: File not found." << endl; + return -1; + } + + parseFile(fileStream, serverName, baseName); + return 0; +} |