3a5dc285ab
This tool creates the fault-space pruning pilot and group entries. Those are used by the generic campaign to do fault experiments. Currently prune-trace only implements conventional def/use pruning (--prune-method "basic"). Change-Id: I1dfb431e3b1d3cd2ee891a49a3b6ac01210be11f
99 lines
2.6 KiB
C++
99 lines
2.6 KiB
C++
#include <mysql/mysql.h>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#include "util/CommandLine.hpp"
|
|
#include "util/Logger.hpp"
|
|
static fail::Logger log("prune-trace", true);
|
|
|
|
using namespace fail;
|
|
using std::endl;
|
|
|
|
#include "Pruner.hpp"
|
|
#include "BasicPruner.hpp"
|
|
|
|
int main(int argc, char *argv[]) {
|
|
std::string username, hostname, database, benchmark, variant;
|
|
|
|
// Manually fill the command line option parser
|
|
CommandLine &cmd = CommandLine::Inst();
|
|
for (int i = 1; i < argc; ++i)
|
|
cmd.add_args(argv[i]);
|
|
|
|
CommandLine::option_handle IGNORE = cmd.addOption("", "", Arg::None, "USAGE: import-trace [options]");
|
|
CommandLine::option_handle HELP = cmd.addOption("h", "help", Arg::None, "-h,--help\t Print usage and exit");
|
|
|
|
Database::cmdline_setup();
|
|
|
|
CommandLine::option_handle VARIANT = cmd.addOption("v", "variant", Arg::Required,
|
|
"-v/--variant\t Variant label (default: \"none\")");
|
|
CommandLine::option_handle BENCHMARK = cmd.addOption("b", "benchmark", Arg::Required,
|
|
"-b/--benchmark\t Benchmark label (default: \"none\")\n");
|
|
|
|
CommandLine::option_handle PRUNER = cmd.addOption("p", "prune-method", Arg::Required,
|
|
"-p/--prune-method\t Which import method to use (default: basic)");
|
|
|
|
if(!cmd.parse()) {
|
|
std::cerr << "Error parsing arguments." << std::endl;
|
|
exit(-1);
|
|
}
|
|
|
|
Pruner *pruner;
|
|
if (cmd[PRUNER].count() > 0) {
|
|
std::string imp(cmd[PRUNER].first()->arg);
|
|
if (imp == "basic") {
|
|
log << "Using BasicPruner" << endl;
|
|
pruner = new BasicPruner();
|
|
} else {
|
|
log << "Unkown import method: " << imp << endl;
|
|
exit(-1);
|
|
}
|
|
|
|
} else {
|
|
log << "Using BasicPruner" << endl;
|
|
pruner = new BasicPruner();
|
|
}
|
|
|
|
if (cmd[HELP]) {
|
|
cmd.printUsage();
|
|
exit(0);
|
|
}
|
|
|
|
Database *db = Database::cmdline_connect();
|
|
|
|
if (cmd[VARIANT].count() > 0)
|
|
variant = std::string(cmd[VARIANT].first()->arg);
|
|
else
|
|
variant = "none";
|
|
|
|
if (cmd[BENCHMARK].count() > 0)
|
|
benchmark = std::string(cmd[BENCHMARK].first()->arg);
|
|
else
|
|
benchmark = "none";
|
|
|
|
if (!pruner->init(variant, benchmark, db)) {
|
|
log << "pruner->init() failed" << endl;
|
|
exit(-1);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// Do the actual import
|
|
////////////////////////////////////////////////////////////////
|
|
if (!pruner->create_database()) {
|
|
log << "create_database() failed" << endl;
|
|
exit(-1);
|
|
}
|
|
|
|
if (!pruner->clear_database()) {
|
|
log << "clear_database() failed" << endl;
|
|
exit(-1);
|
|
}
|
|
|
|
if (!pruner->prune_all()) {
|
|
log << "prune_all() failed" << endl;
|
|
exit(-1);
|
|
}
|
|
|
|
return 0;
|
|
}
|