AAerosolSQLManager.cc

Go to the documentation of this file.

#include <atm/AAerosolSQLManager.h>
#include <atm/AManagerRegister.h>
#include <det/Detector.h>
#include <utl/ErrorLogger.h>
#include <utl/TimeStamp.h>
#include <utl/AugerException.h>
#include <utl/StringCompare.h>

#include <boost/lexical_cast.hpp>
#include <sstream>

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


REGISTER_A_MANAGER("AAerosolSQLManager", AAerosolSQLManager);


AAerosolSQLManager::AAerosolSQLManager() :
  VSQLManager()
{
  fAvailableComponents.insert({
    "aerosol", "aerosol_zone", "attenuation", "attenuation_lambda", "authors",
    "db_header", "phase_func", "phase_func_lambda", "software",
    "zone", "zone_slices"
  });
}


VManager::Status
AAerosolSQLManager::GetDBResFundamental(const string& tableName,
                                        const string& columnName,
                                        const VManager::IndexMap& componentIndex)
  const
{
  const utl::TimeStamp detTime = Detector::GetInstance().GetTime();

  ostringstream query;

  // aerosol table is a special case, as it requires checking software version
  if (tableName == "aerosol") {

    if (StringEquivalent(fDatabaseSoftwareVersion, "default")) {

      query << "SELECT aerosol_id "
               "FROM aerosol "
               "WHERE start_time <= \"" << detTime.GetGPSSecond() << "\" "
               "AND (end_time > \"" << detTime.GetGPSSecond() << "\" "
               "OR end_time IS NULL) ORDER BY last_modified DESC, start_time DESC LIMIT 1";

    } else {

      query << "SELECT aerosol_id "
               "FROM aerosol, software "
               "WHERE start_time <= \"" << detTime.GetGPSSecond() << "\" "
               "AND (end_time > \"" << detTime.GetGPSSecond() << "\" "
               "OR end_time IS NULL) "
               "AND aerosol.software_id = software.software_id "
               "AND software.software_version = \"" << fDatabaseSoftwareVersion << "\" "
               "ORDER BY aerosol.last_modified DESC, start_time DESC LIMIT 1";
    }

  } else if (tableName == "zone") {

    if (componentIndex.find("zone_id") == componentIndex.end()) {
      ERROR("request for zone data called with no zone_id");
      throw DataNotFoundInDBException("zone", "zone_id", componentIndex,
                                      boost::lexical_cast<string>(Detector::GetInstance().GetTime()));
    }
    query << "SELECT " << columnName << " "
             "FROM zone "
             "WHERE zone_id = " << componentIndex.find("zone_id")->second;

  } else {

    if (fAvailableComponents.find(tableName) == fAvailableComponents.end())
      return eNotFound;

    // issue queries in form :
    //   SELECT columnName FROM tableName WHERE
    //   componentIndex[0].first = componentIndex[0].second AND
    //   componentIndex[1].first = componentIndex[1].second ...

    query << "SELECT " << columnName << " "
             "FROM " << tableName << " "
             "WHERE " << MergeIndexMap(componentIndex);
  }

  // Send the query
  return Query(query, QueryInfoMessage(tableName, columnName));
}


VManager::Status
AAerosolSQLManager::GetDBResVector(const string& tableName,
                                   const string& columnName,
                                   const VManager::IndexMap& componentIndex)
  const
{
  // special case (I think) : get all zones valid for a particular aerosol_id

  if (tableName == "zone") {

    // index map must specify the aerosol_id
    if (componentIndex.find("aerosol_id") == componentIndex.end()) {
      DEBUGLOG("no aerosol_id specified when looking up zone names");
      return VManager::eNotFound;
    }

    ostringstream query;
    query << "SELECT zone." << columnName << " FROM zone,aerosol_zone "
             "WHERE aerosol_id = " << componentIndex.find("aerosol_id")->second << " "
             "AND aerosol_zone.zone_id = zone.zone_id;";

    return Query(query, string("vector for ") + QueryInfoMessage(tableName, columnName));

  } else {

    if (fAvailableComponents.find(tableName) == fAvailableComponents.end())
      return eNotFound;

    ostringstream query;
    query << "SELECT " << columnName << " "
             "FROM " << tableName << " "
             "WHERE " << MergeIndexMap(componentIndex);

    return Query(query, string("vector for ") + QueryInfoMessage(tableName, columnName));

  }
}


VManager::Status
AAerosolSQLManager::GenericGetData(VManager::Handle& /*returnData*/,
                                   const string& /*component*/,
                                   const string& /*property*/,
                                   const VManager::IndexMap& /*indexMap*/)
  const
{
  return eNotFound;
}