VSQLManager_SQLite.cc

Go to the documentation of this file.

#include <det/VSQLManager_SQLite.h>
#include <utl/Reader.h>
#include <utl/AugerUnits.h>
#include <fwk/CentralConfig.h>
#include <fwk/CommandLineOptions.h>
#include <boost/algorithm/string.hpp>

using namespace det;
using namespace std;
using namespace utl;


void
VSQLManager::Close()
{
  if (!fIsInitialized)
    return;

  const double time = fTime.GetTime();
  if (time) {
    ostringstream info;
    info << "Total time spent for SQLite database '" << fDatabaseName
         << "' queries is " << time/second << " sec.";
    // reinstate this INFO(info);
  }

  FreeResult(); // free resources from previous query if not yet freed

  sqlite3_exec(fDb, "END TRANSACTION", 0, 0, 0);
  sqlite3_close(fDb);

  fDb = nullptr;
  fIsInitialized = false;
}


int
VSQLManager::FreeResult()
  const
{
  const int rc = sqlite3_finalize(fResult);
  fResult = nullptr;
  return rc;
}


int
VSQLManager::FetchRow()
  const
{
  return sqlite3_step(fResult);
}


void
VSQLManager::Init(const std::string& configLink)
{
  VManager::Init(configLink);
  fIsInitialized = false;

  fBranch.GetChild("databaseName").GetData(fDatabaseName);
  fBranch.GetChild("softwareVersion").GetData(fDatabaseSoftwareVersion);

  if (sqlite3_open_v2(fDatabaseName.c_str(), &fDb, SQLITE_OPEN_READONLY, 0)) {
    ostringstream err;
    err << VSQLManager::GetName() << " could not open the requested SQLite database file " << fDatabaseName << endl;
    ERROR(err);
    throw IOFailureException(err.str());
  }
  ostringstream info;
  info << VSQLManager::GetName() << " connects to the database '" << fDatabaseName << '\'';
  INFO(info);
  fIsInitialized = true;

  sqlite3_exec(fDb, "PRAGMA cache_size=500000", 0, 0, 0);
  sqlite3_exec(fDb, "PRAGMA temp_store=MEMORY", 0, 0, 0);
  sqlite3_exec(fDb, "BEGIN TRANSACTION", 0, 0, 0);
}


VManager::Status
VSQLManager::Query(const std::string& query, const std::string& /*what*/ /*for error report*/)
  const
{
  if (!fIsInitialized) {
    const string err = "SQLite database is not opened.";
    ERROR(err);
    throw IOFailureException(err);
  }

  // before preparing the next query, make sure the memory used for preparing
  // the last one is freed (if it is already freed, no harm is done)
  FreeResult();

  Status stat = eNotFound;
  const char* s = nullptr;
  fTime.Start();
  const string queryMod = boost::replace_all_copy(query, "UNIX_TIMESTAMP(", "strftime('%s',");
  const int rc = sqlite3_prepare_v2(fDb, queryMod.c_str(), queryMod.size(), &fResult, &s);
  fTime.Stop();
  if (rc == SQLITE_OK)
    stat = eFound;
  else {
    FreeResult();
    cerr << "SQLite error " << rc
         << ": could not process the query \"" << query << "\"" << endl;
    // put proper error handling here (eg, check 'what' and IsReportingErrors).
  }
  return stat;
}


string
VSQLManager::MergeIndexMap(const IndexMap& componentIndex)
{
  ostringstream res;
  VManager::IndexMap::const_iterator it = componentIndex.begin();
  res << it->first << " = " << it->second;
  for (++it; it != componentIndex.end(); ++it)
    res << " AND " << it->first << " = " << it->second;
  return res.str();
}