src/db/db-mgr.cpp

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
 * Copyright (c) 2014, Regents of the University of California.
 *
 * This file is part of NDNS (Named Data Networking Domain Name Service).
 * See AUTHORS.md for complete list of NDNS authors and contributors.
 *
 * NDNS is free software: you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 *
 * NDNS is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 * PURPOSE.  See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * NDNS, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <iostream>
#include <fstream>

#include "db-mgr.hpp"

namespace ndn {
namespace ndns {

DBMgr::DBMgr()
  : m_dbfile("src/db/ndns-local.db")
  , m_conn(0)
  , m_reCode(-1)
  , m_status(DBClosed)
  , m_resultNum(0)
{
  std::fstream file;
  file.open(m_dbfile, std::ios::in);
  if (file) {
    std::cout << "database file " << m_dbfile << " does exist" << std::endl;
  } else {
    std::cout << "database file " << m_dbfile << " does not exist" << std::endl;
  }

}

DBMgr::~DBMgr()
{
}

void DBMgr::open()
{
  if (m_status == DBConnected)
    return;

  m_reCode = sqlite3_open(this->m_dbfile.c_str(), &(this->m_conn));
  if (m_reCode != SQLITE_OK) {
    m_err = "Cannot connect to the db: " + this->m_dbfile;
    m_status = DBError;
    //exit(1);
  } else {
    std::cout << "connect to the db: " << m_dbfile << std::endl;
  }
  m_status = DBConnected;
}

void DBMgr::close()
{
  if (m_status == DBClosed)
    return;

  m_reCode = sqlite3_close(this->m_conn);
  if (m_reCode != SQLITE_OK) {
    m_err = "Cannot close the db: " + this->m_dbfile;
    m_status = DBError;
  }
  m_status = DBClosed;
}

void DBMgr::execute(std::string sql,
    int (*callback)(void*, int, char**, char**), void * paras)
{
  if (m_status == DBClosed)
    this->open();

  clearResultNum();
  char *err_msg;
  m_reCode = sqlite3_exec(m_conn, sql.c_str(), callback, paras, &err_msg);
  if (m_reCode != SQLITE_OK) {
    m_status = DBError;
    this->m_err.append(err_msg);
    std::cout << this->m_err << std::endl;
  }
  this->close();
}

} //namespace ndns
} //namespace ndn