invalid_matches.cpp

#include "mine/invalid_matches.hpp"

#include "seq/seq_list.hpp"
#include "sys/file.hpp"
#include "sys/log.hpp"
#include "sys/setup.hpp"

const std::string FILENAME = "invalid_matches.txt";

InvalidMatches::InvalidMatches()
    : scheduler(1800)  // 30 minutes
{}

void InvalidMatches::load() {
  // Migrate file from lists folder to cache folder
  const auto lists_path = SequenceList::getListsHome() + FILENAME;
  const auto cache_path = Setup::getCacheHome() + FILENAME;
  if (isFile(lists_path) && !isFile(cache_path)) {
    Log::get().info("Migrating \"" + FILENAME + "\" from lists to cache folder");
    moveFile(lists_path, cache_path);
  }
  
  auto path = cache_path;
  try {
    SequenceList::loadMap(path, invalid_matches);
  } catch (const std::exception&) {
    Log::get().warn("Resetting corrupt file " + path);
    invalid_matches.clear();
    deleteFile();
  }
}

bool InvalidMatches::hasTooMany(UID id) const {
  auto it = invalid_matches.find(id);
  if (it != invalid_matches.end() && it->second > 0) {
    int64_t r = Random::get().gen() % it->second;
    return r >= 100;
  }
  return false;
}

void InvalidMatches::insert(UID id) {
  invalid_matches[id]++;
  if (scheduler.isTargetReached()) {
    scheduler.reset();
    Log::get().info("Saving invalid matches stats for " +
                    std::to_string(invalid_matches.size()) + " sequences");
    SequenceList::mergeMap(Setup::getCacheHome(), FILENAME, invalid_matches);
  }
}

void InvalidMatches::deleteFile() {
  auto path = Setup::getCacheHome() + FILENAME;
  std::remove(path.c_str());
}