/*
*
* Copyright (C) 2012 Mattias Runge
*
* This program 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 2 of the License, or
* (at your option) any later version.
*
* This program 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 this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
*/
#include "MySql.h"
#include <mysql.h>
#include <boost/lexical_cast.hpp>
#include "vm/Manager.h"
#include "common/log.h"
#include "common/exception.h"
namespace atom {
namespace vm {
namespace plugin {
MySql::Resources MySql::resources_;
static const std::string log_module_ = "vm::plugin::mysql";
MySql::MySql(boost::asio::io_service& io_service) : Plugin(io_service)
{
this->name_ = "mysql";
this->ExportFunction("MySqlExport_Connect", MySql::Export_Connect);
this->ExportFunction("MySqlExport_Close", MySql::Export_Close);
this->ExportFunction("MySqlExport_Query", MySql::Export_Query);
this->ExportFunction("MySqlExport_SelectDb", MySql::Export_SelectDb);
this->ExportFunction("MySqlExport_AffectedRows", MySql::Export_AffectedRows);
this->ExportFunction("MySqlExport_InsertId", MySql::Export_InsertId);
}
MySql::~MySql()
{
}
void MySql::InitializeDone()
{
Plugin::InitializeDone();
}
void MySql::CallOutput(unsigned int request_id, std::string output)
{
atom::log::Info(log_module_, output);
}
MySql::ResourceId MySql::GetFreeResourceId()
{
ResourceId resource_id = 1;
while (NULL != MySql::resources_[resource_id])
{
resource_id++;
}
return resource_id;
}
Value MySql::Export_Connect(const v8::Arguments& args)
{
ATOM_VM_PLUGIN_SCOPE;
MYSQL* resource = NULL;
ResourceId resource_id;
//atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
try
{
ATOM_VM_PLUGIN_NUM_PARAMS(3);
v8::String::AsciiValue server(args[0]);
v8::String::AsciiValue username(args[1]);
v8::String::AsciiValue password(args[2]);
/* Initialize MySQL resource */
resource = mysql_init(NULL);
if (NULL == resource)
{
throw atom::exception::initialization_failed();
}
/* Connect to database */
if (mysql_real_connect(resource, *server, *username, *password, NULL, 0, NULL, 0) == NULL)
{
atom::log::Error(log_module_, "Failed to connect to database %s, error code %d, error message \"%s\".", *server, mysql_errno(resource), mysql_error(resource));
throw atom::exception::connect_failed();
}
/* Add resource to list */
resource_id = MySql::GetFreeResourceId();
MySql::resources_[resource_id] = resource;
return handle_scope.Close(v8::Uint32::New(resource_id));
}
catch (std::exception& exception)
{
if (NULL != resource)
{
mysql_close(resource);
resource = NULL;
}
atom::log::Exception(log_module_, exception);
return handle_scope.Close(v8::Boolean::New(false));
}
}
Value MySql::Export_Close(const v8::Arguments& args)
{
ATOM_VM_PLUGIN_SCOPE;
//atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
try
{
ATOM_VM_PLUGIN_NUM_PARAMS(1);
/* Check that the resource exist */
if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
{
throw atom::exception::missing_resource();
}
/* Close connection */
mysql_close(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
MySql::resources_.erase((ResourceId)args[0]->Uint32Value());
return handle_scope.Close(v8::Boolean::New(true));
}
catch (std::exception& exception)
{
atom::log::Exception(log_module_, exception);
return handle_scope.Close(v8::Boolean::New(false));
}
}
Value MySql::Export_Query(const v8::Arguments& args)
{
ATOM_VM_PLUGIN_SCOPE;
MYSQL_RES* result = NULL;
//atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
try
{
ArgumentListPointer result_array = ArgumentListPointer(new ArgumentList);
ATOM_VM_PLUGIN_NUM_PARAMS(2);
v8::String::AsciiValue query(args[1]);
/* Check that the resource exist */
if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
{
throw atom::exception::missing_resource();
}
/* Execute the query */
if (0 != mysql_query(MySql::resources_[(ResourceId)args[0]->Uint32Value()], *query))
{
throw atom::exception::action_failed();
}
/* Get the result */
result = mysql_store_result(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
if (NULL == result)
{
if (mysql_field_count(MySql::resources_[(ResourceId)args[0]->Uint32Value()]) > 0)
{
throw atom::exception::action_failed();
}
}
else
{
MYSQL_ROW row;
uint32_t number_of_fields = mysql_num_fields(result);
MYSQL_FIELD* fields = mysql_fetch_fields(result);
uint32_t row_index = 0;
v8::Local<v8::Array> result_array = v8::Array::New();
while (NULL != (row = mysql_fetch_row(result)))
{
v8::Local<v8::Object> vars = v8::Object::New();
for (uint32_t index = 0; index < number_of_fields; index++)
{
vars->Set(v8::String::New(fields[index].name), v8::String::New(row[index]));
// atom::log::Info(log_module_, "%s = %s\n", fields[index].name, row[index]);
}
result_array->Set(row_index, vars);
row_index++;
}
mysql_free_result(result);
result = NULL;
return handle_scope.Close(v8::Handle<v8::Value>(result_array));
}
return handle_scope.Close(v8::Boolean::New(true));
}
catch (std::exception& exception)
{
if (NULL != result)
{
mysql_free_result(result);
result = NULL;
}
atom::log::Exception(log_module_, exception);
return handle_scope.Close(v8::Boolean::New(false));
}
}
Value MySql::Export_SelectDb(const v8::Arguments& args)
{
ATOM_VM_PLUGIN_SCOPE;
//atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
try
{
ATOM_VM_PLUGIN_NUM_PARAMS(2);
v8::String::AsciiValue database(args[1]);
/* Check that the resource exist */
if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
{
throw atom::exception::missing_resource();
}
/* Select database */
if (0 != mysql_select_db(MySql::resources_[(ResourceId)args[0]->Uint32Value()], *database))
{
throw atom::exception::action_failed();
}
return handle_scope.Close(v8::Boolean::New(true));
}
catch (std::exception& exception)
{
atom::log::Exception(log_module_, exception);
return handle_scope.Close(v8::Boolean::New(false));
}
}
Value MySql::Export_AffectedRows(const v8::Arguments& args)
{
ATOM_VM_PLUGIN_SCOPE;
//atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
try
{
ATOM_VM_PLUGIN_NUM_PARAMS(1);
/* Check that the resource exist */
if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
{
throw atom::exception::missing_resource();
}
/* Select database */
uint32_t number_of_rows = mysql_affected_rows(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
return handle_scope.Close(v8::Uint32::New(number_of_rows));
}
catch (std::exception& exception)
{
atom::log::Exception(log_module_, exception);
return handle_scope.Close(v8::Boolean::New(false));
}
}
Value MySql::Export_InsertId(const v8::Arguments& args)
{
ATOM_VM_PLUGIN_SCOPE;
//atom::log::Debug(log_module_, "%s called!", __FUNCTION__);
try
{
ATOM_VM_PLUGIN_NUM_PARAMS(1);
/* Check that the resource exist */
if (NULL == MySql::resources_[(ResourceId)args[0]->Uint32Value()])
{
throw atom::exception::missing_resource();
}
/* Select database */
uint32_t insert_id = mysql_insert_id(MySql::resources_[(ResourceId)args[0]->Uint32Value()]);
return handle_scope.Close(v8::Uint32::New(insert_id));
}
catch (std::exception& exception)
{
atom::log::Exception(log_module_, exception);
return handle_scope.Close(v8::Boolean::New(false));
}
}
}; // namespace plugin
}; // namespace vm
}; // namespace atom