tools/prune-trace: various improvements
- Variants/benchmarks can now be selected with wildcards
(--variant/--benchmark), and can be excluded from pruning
(--variant-exclude/--benchmark-exclude).
- The database clearing step can be skipped with --no-delete to
avoid deadlocks with concurrent DB accesses.
- Internals:
* injection_instr / injection_instr_absolute moves from
fspgroup to fsppilot. fsppilot now contains all information we
need for running FI experiments.
TODO: generic campaign needs to be modified, too.
* Force MySQL to use an efficient join order (STRAIGHT_JOIN).
Change-Id: I6241ea2de9da1a1e709fae6374df4fc06ef262a0
This commit is contained in:
@ -8,17 +8,42 @@ static Logger LOG ("Pruner");
|
||||
#include "Pruner.hpp"
|
||||
|
||||
|
||||
bool Pruner::init(const std::string &variant, const std::string &benchmark, Database *db) {
|
||||
bool Pruner::init(fail::Database *db,
|
||||
const std::vector<std::string>& variants,
|
||||
const std::vector<std::string>& variants_exclude,
|
||||
const std::vector<std::string>& benchmarks,
|
||||
const std::vector<std::string>& benchmarks_exclude)
|
||||
{
|
||||
this->db = db;
|
||||
if (!(m_variant_id = db->get_variant_id(variant, benchmark))) {
|
||||
return false;
|
||||
std::stringstream ss;
|
||||
|
||||
// FIXME string escaping
|
||||
ss << "SELECT id FROM variant WHERE ";
|
||||
for (std::vector<std::string>::const_iterator it = variants.begin();
|
||||
it != variants.end(); ++it) {
|
||||
ss << "variant LIKE '" << *it << "' AND ";
|
||||
}
|
||||
for (std::vector<std::string>::const_iterator it = variants_exclude.begin();
|
||||
it != variants_exclude.end(); ++it) {
|
||||
ss << "variant NOT LIKE '" << *it << "' AND ";
|
||||
}
|
||||
for (std::vector<std::string>::const_iterator it = benchmarks.begin();
|
||||
it != benchmarks.end(); ++it) {
|
||||
ss << "benchmark LIKE '" << *it << "' AND ";
|
||||
}
|
||||
for (std::vector<std::string>::const_iterator it = benchmarks_exclude.begin();
|
||||
it != benchmarks_exclude.end(); ++it) {
|
||||
ss << "benchmark NOT LIKE '" << *it << "' AND ";
|
||||
}
|
||||
// dummy terminator to avoid special cases in query construction above
|
||||
ss << "1";
|
||||
m_variant_id_query = ss.str();
|
||||
|
||||
if (!(m_method_id = db->get_fspmethod_id(method_name()))) {
|
||||
return false;
|
||||
}
|
||||
LOG << "Pruning variant "
|
||||
<< variant << "/" << benchmark << " (ID: " << m_variant_id << ")"
|
||||
<< " with method " << method_name() << " (ID: " << m_method_id << ")"
|
||||
|
||||
LOG << "Pruning with method " << method_name() << " (ID: " << m_method_id << ")"
|
||||
<< std::endl;
|
||||
return true;
|
||||
}
|
||||
@ -28,37 +53,37 @@ bool Pruner::create_database() {
|
||||
" id int(11) NOT NULL AUTO_INCREMENT,"
|
||||
" known_outcome tinyint(4) NOT NULL,"
|
||||
" variant_id int(11) NOT NULL,"
|
||||
" instr1 int(10) unsigned NOT NULL,"
|
||||
" instr2 int(10) unsigned NOT NULL,"
|
||||
" injection_instr int(10) unsigned NOT NULL,"
|
||||
" injection_instr_absolute int(10) unsigned,"
|
||||
" data_address int(10) unsigned NOT NULL,"
|
||||
" fspmethod_id int(11) NOT NULL,"
|
||||
" PRIMARY KEY (id),"
|
||||
" KEY fspmethod_id (fspmethod_id,variant_id,instr1,instr2,data_address)"
|
||||
") engine=MyISAM ";
|
||||
" KEY fspmethod_id (fspmethod_id,variant_id,instr2,data_address)"
|
||||
") engine=MyISAM ";
|
||||
bool success = (bool) db->query(create_statement.c_str());
|
||||
if (!success) return false;
|
||||
|
||||
create_statement = "CREATE TABLE IF NOT EXISTS fspgroup ("
|
||||
" variant_id int(11) NOT NULL,"
|
||||
" injection_instr int(10) unsigned NOT NULL,"
|
||||
" injection_instr_absolute int(10) unsigned,"
|
||||
" instr2 int(11) unsigned NOT NULL,"
|
||||
" data_address int(10) unsigned NOT NULL,"
|
||||
" fspmethod_id int(11) NOT NULL,"
|
||||
" pilot_id int(11) NOT NULL,"
|
||||
" PRIMARY KEY (variant_id, injection_instr, data_address, fspmethod_id, pilot_id),"
|
||||
" KEY joinresults (pilot_id,fspmethod_id))";
|
||||
" PRIMARY KEY (variant_id, instr2, data_address, fspmethod_id, pilot_id),"
|
||||
" KEY joinresults (pilot_id,fspmethod_id)) engine=MyISAM";
|
||||
|
||||
return db->query(create_statement.c_str());
|
||||
}
|
||||
|
||||
bool Pruner::clear_database() {
|
||||
std::stringstream ss;
|
||||
ss << "DELETE FROM fsppilot WHERE variant_id = " << m_variant_id << " AND fspmethod_id = " << m_method_id;
|
||||
ss << "DELETE FROM fsppilot WHERE variant_id IN (" << m_variant_id_query << ")";
|
||||
bool ret = (bool) db->query(ss.str().c_str());
|
||||
LOG << "deleted " << db->affected_rows() << " rows from fsppilot table" << std::endl;
|
||||
ss.str("");
|
||||
|
||||
ss << "DELETE FROM fspgroup WHERE variant_id = " << m_variant_id << " AND fspmethod_id = " << m_method_id;
|
||||
ss << "DELETE FROM fspgroup WHERE variant_id IN (" << m_variant_id_query << ")";
|
||||
ret = ret && (bool) db->query(ss.str().c_str());
|
||||
LOG << "deleted " << db->affected_rows() << " rows from fspgroup table" << std::endl;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user