/*
 * rr-mgr.cpp
 *
 *  Created on: 23 Jul, 2014
 *      Author: shock
 */

#include "rr-mgr.hpp"

namespace ndn {
namespace ndns{
RRMgr::RRMgr(Zone& zone, Query& query, Response& response)
  : m_count(0U)
  , m_zone(zone)
  , m_query(query)
  , m_response(response)
{

}


RRMgr::~RRMgr() {
  // TODO Auto-generated destructor stub
}


int
RRMgr::count()
{
  std::string sql;
  sql =  "SELECT count(*) FROM rrs INNER JOIN rrsets ON rrs.rrset_id=rrsets.id";
  sql += " WHERE rrsets.zone_id=";
  sql += std::to_string(m_zone.getId());
  sql += " AND ";
  sql += "rrsets.type=\'";
  sql += RR::toString(m_query.getRrType());
  sql += "\' AND ";
  sql += "rrsets.label LIKE\'";
  sql += m_query.getRrLabel().toUri()+"/%\'";

  std::cout<<"sql="<<sql<<std::endl;
  this->execute(sql, static_callback_countRr, this);

  if (this->getStatus() == DBMgr::DBError)
  {
    return -1;
  }

  return this->m_count;
}
int
RRMgr::callback_countRr(int argc, char **argv, char **azColName)
{
  this->addResultNum();

  std::cout<<this->getResultNum()<<"th result: "<<"count="<<argv[0]<<std::endl;
  m_count = std::atoi(argv[0]);
  return 0;
}

int RRMgr::lookup()
{
  std::string sql;

  sql =  "SELECT rrs.ttl, rrs.rrdata, rrs.id FROM rrs INNER JOIN rrsets ON rrs.rrset_id=rrsets.id";
  sql += " WHERE rrsets.zone_id=";
  sql += std::to_string(m_zone.getId());
  sql += " AND ";
  sql += "rrsets.type=\'";
  sql += RR::toString(m_query.getRrType());
  sql += "\' AND ";
  sql += "rrsets.label=\'";
  sql += m_query.getRrLabel().toUri()+"\'";
  sql += " ORDER BY rrs.id";

  std::cout<<"sql="<<sql<<std::endl;
  this->execute(sql, static_callback_getRr, this);

  if (this->getStatus() == DBMgr::DBError)
  {
    return -1;
  }

  //m_response.setQueryName(m_query.getAuthorityZone());
  return 0;
}

int RRMgr::callback_getRr(int argc, char **argv, char **azColName)
{
  this->addResultNum();
  if (argc < 1)
  {
    this->setErr("No RRType="+RR::toString(m_query.getRrType())+
        " and label="+m_query.getRrLabel().toUri()+
        " and zone="+m_zone.getAuthorizedName().toUri());
    return 0;
  }

  std::cout<<this->getResultNum()<<"th result: "<<"id="<<argv[2]
           <<" ttl="<<argv[0]<<" rrdata="<<argv[1]<<std::endl;

  m_response.setFreshness(time::milliseconds(std::atoi(argv[0])));

  m_response.addRr(std::atoi(argv[2]), argv[1]);
  //std::cout<<"after add a Rr: current size="<<m_response.getRrs().size()<<std::endl;
  //m_response.setFreshness(time::milliseconds(std::atoi(argv[0])));
  return 0;
}

}//namespace ndnsn
} /* namespace ndn */