catalog/src/util/mysql-util.cpp

/** NDN-Atmos: Cataloging Service for distributed data originally developed
 *  for atmospheric science data
 *  Copyright (C) 2015 Colorado State University
 *
 *  NDN-Atmos 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.
 *
 *  NDN-Atmos 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 NDN-Atmos.  If not, see <http://www.gnu.org/licenses/>.
**/

#include "util/mysql-util.hpp"
#include <mysql/errmsg.h>
#include <stdexcept>
#include <iostream>

namespace atmos {
namespace util {

ConnectionDetails::ConnectionDetails(const std::string& serverInput, const std::string& userInput,
                                     const std::string& passwordInput, const std::string& databaseInput)
  : server(serverInput), user(userInput), password(passwordInput), database(databaseInput)
{
  // empty
}


std::shared_ptr<MYSQL>
MySQLConnectionSetup(const ConnectionDetails& details)
{
  MYSQL* conn = mysql_init(NULL);
  my_bool reconnect = 1;
  mysql_options(conn, MYSQL_OPT_RECONNECT, &reconnect);
  if(!mysql_real_connect(conn, details.server.c_str(), details.user.c_str(),
                         details.password.c_str(), details.database.c_str(), 0, NULL, 0)) {
    throw std::runtime_error(mysql_error(conn));
  }
  std::shared_ptr<MYSQL> connection(conn, &mysql_close);
  return connection;
}

std::shared_ptr<MYSQL_RES>
MySQLPerformQuery(std::shared_ptr<MYSQL> connection,
                  const std::string& sql_query,
                  DatabaseOperation op,
                  bool& success,
                  std::string& errMsg)
{
  switch (mysql_query(connection.get(), sql_query.c_str()))
  {
  case 0:
  {
    success = true;
    if (op == QUERY) {
      MYSQL_RES* resultPtr = mysql_store_result(connection.get());
      if (resultPtr != NULL)
      {
        return std::shared_ptr<MYSQL_RES>(resultPtr, &mysql_free_result);
      }
    }
    //for add, remove, we don't need the results, may be log the events
    break;
  }
  // Various error cases
  case CR_COMMANDS_OUT_OF_SYNC:
  case CR_SERVER_GONE_ERROR:
  case CR_SERVER_LOST:
  case CR_UNKNOWN_ERROR:
  default:
  {
    errMsg.assign(mysql_error(connection.get()));
    break;
  }
  }
  return nullptr;
}

} // namespace util
} // namespace atmos