f18cddc63c
The DatabaseCampaign interacts with the MySQL tables that are created
by the import-trace and prune-trace tools. It does offer all
unfinished experiment pilots from the database to the
fail-clients. Those clients send back a (by the experiment) defined
protobuf message as a result. The custom protobuf message does have to
need the form:
import "DatabaseCampaignMessage.proto";
message ExperimentMsg {
required DatabaseCampaignMessage fsppilot = 1;
repeated group Result = 2 {
// custom fields
required int32 bitoffset = 1;
optional int32 result = 2;
}
}
The DatabaseCampaignMessage is the pilot identifier from the
database. For each of the repeated result entries a row in a table is
allocated. The structure of this table is constructed (by protobuf
reflection) from the description of the message. Each field in the
Result group becomes a column in the result table. For the given
example it would be:
CREATE TABLE result_ExperimentMessage(
pilot_id INT,
bitoffset INT NOT NULL,
result INT,
PRIMARY_KEY(pilot_id)
)
Change-Id: I28fb5488e739d4098b823b42426c5760331027f8
98 lines
2.6 KiB
C++
98 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;
|
|
}
|