Denes Matetelki
13 years ago
parent
4fa576f959
commit
a7db838068
@ -0,0 +1,76 @@
|
||||
#ifndef MYSQL_CLIENT_HPP
|
||||
#define MYSQL_CLIENT_HPP
|
||||
|
||||
#include <mysql/mysql.h>
|
||||
|
||||
#include <string>
|
||||
#include <list>
|
||||
|
||||
class MysqlClient
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
/**
|
||||
* For more details about the params,
|
||||
* check the documentation of mysql_real_connect
|
||||
*
|
||||
* @note Call init_client_errs() / finish_client_errs() before / after.
|
||||
*
|
||||
* @param host May be either a host name or an IP address.
|
||||
* If host is NULL or the string "localhost",
|
||||
* a connection to the local host is assumed.
|
||||
* @param user If user is NULL or the empty string "",
|
||||
* the current user is assumed.
|
||||
* @param passwd If passwd is NULL, only entries in the user table
|
||||
* for the user that have a blank (empty) password field
|
||||
* are checked for a match.
|
||||
* @param db If db is not NULL, the connection sets the default
|
||||
* database to this value.
|
||||
* @param port If port is not 0, the value is used as the port number
|
||||
* for the TCP/IP connection. Note that the host parameter determines
|
||||
* the type of the connection.
|
||||
* @param unix_socket If unix_socket is not NULL, the string specifies
|
||||
* the socket or named pipe that should be used. Note that the host parameter
|
||||
* determines the type of the connection.
|
||||
* @param clientflag Usually 0, but can be set to a combination
|
||||
* of flags to enable certain features.
|
||||
*/
|
||||
MysqlClient ( const char *host = NULL,
|
||||
const char *user = NULL,
|
||||
const char *passwd = NULL,
|
||||
const char *db = NULL,
|
||||
unsigned int port = 0,
|
||||
const char *unix_socket = NULL,
|
||||
unsigned long clientflag = 0 );
|
||||
|
||||
~MysqlClient();
|
||||
|
||||
bool connect();
|
||||
|
||||
bool querty(const std::string queryLine,
|
||||
std::list<std::string> &result);
|
||||
|
||||
private:
|
||||
|
||||
MysqlClient(const MysqlClient&);
|
||||
MysqlClient& operator=(const MysqlClient&);
|
||||
|
||||
/// @todo optimize this
|
||||
void queryResultToStringList(MYSQL_RES *res_set,
|
||||
std::list<std::string> &result);
|
||||
|
||||
const char *m_host;
|
||||
const char *m_user;
|
||||
const char *m_passwd;
|
||||
const char *m_db;
|
||||
unsigned int m_port;
|
||||
const char *m_unix_socket;
|
||||
unsigned long m_clientflag;
|
||||
|
||||
bool m_connected;
|
||||
MYSQL *m_connection;
|
||||
};
|
||||
|
||||
|
||||
#endif // MYSQL_CLIENT_HPP
|
||||
@ -0,0 +1,130 @@
|
||||
#include "MysqlClient.hpp"
|
||||
|
||||
#include <mysql/errmsg.h>
|
||||
|
||||
#include "Logger.hpp"
|
||||
#include "Common.hpp"
|
||||
|
||||
|
||||
MysqlClient::MysqlClient( const char *host,
|
||||
const char *user,
|
||||
const char *passwd,
|
||||
const char *db,
|
||||
unsigned int port,
|
||||
const char *unix_socket,
|
||||
unsigned long clientflag )
|
||||
: m_host(host)
|
||||
, m_user(user)
|
||||
, m_passwd(passwd)
|
||||
, m_db(db)
|
||||
, m_port(port)
|
||||
, m_unix_socket(unix_socket)
|
||||
, m_clientflag(clientflag)
|
||||
, m_connected(false)
|
||||
, m_connection(0)
|
||||
{
|
||||
TRACE;
|
||||
}
|
||||
|
||||
|
||||
MysqlClient::~MysqlClient()
|
||||
{
|
||||
TRACE;
|
||||
mysql_close(m_connection);
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
MysqlClient::connect()
|
||||
{
|
||||
TRACE;
|
||||
m_connection = mysql_init(NULL);
|
||||
|
||||
if ( m_connection == NULL ) {
|
||||
LOG ( Logger::ERR, std::string("MySQL Initialization Failed!").c_str());
|
||||
return false;
|
||||
}
|
||||
|
||||
m_connection = mysql_real_connect( m_connection,
|
||||
m_host,
|
||||
m_user,
|
||||
m_passwd,
|
||||
m_db,
|
||||
m_port,
|
||||
m_unix_socket,
|
||||
m_clientflag );
|
||||
|
||||
if ( m_connection == NULL ) {
|
||||
LOG ( Logger::ERR, std::string("MySQL Connection Failed! ").
|
||||
append(mysql_error(m_connection)).c_str());
|
||||
return false;
|
||||
}
|
||||
|
||||
LOG ( Logger::DEBUG,
|
||||
std::string("MySQL client connected to ").append(m_host ? m_host : "NULL")
|
||||
.append(", on port: ").append(TToStr(m_port))
|
||||
.append(", as user: ").append(m_user ? m_user : "NULL")
|
||||
.append(", with passwd: ").append(m_passwd ? m_passwd : "NULL")
|
||||
.append(", to DB: ").append(m_db ? m_db : "NULL")
|
||||
.append(", with unix_socket: ").append(m_unix_socket ? m_unix_socket : "NULL")
|
||||
.append(", and flags: ").append(TToStr(m_clientflag))
|
||||
.c_str() );
|
||||
|
||||
m_connected = true;
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
bool
|
||||
MysqlClient::querty(const std::string queryLine,
|
||||
std::list<std::string> &result)
|
||||
{
|
||||
TRACE;
|
||||
|
||||
if ( !m_connected && !connect() )
|
||||
return false;
|
||||
|
||||
if ( mysql_query(m_connection,queryLine.c_str()) != 0 ) {
|
||||
LOG ( Logger::ERR, std::string("MySQL query failed! ").
|
||||
append(mysql_error(m_connection)).c_str());
|
||||
return false;
|
||||
}
|
||||
|
||||
MYSQL_RES *res_set;
|
||||
res_set = mysql_store_result(m_connection);
|
||||
|
||||
if ( res_set == NULL ) {
|
||||
LOG ( Logger::ERR, std::string("MySQL store result failed! ").
|
||||
append(mysql_error(m_connection)).c_str());
|
||||
return false;
|
||||
}
|
||||
|
||||
queryResultToStringList( res_set, result);
|
||||
|
||||
mysql_free_result( res_set );
|
||||
return true;
|
||||
}
|
||||
|
||||
void
|
||||
MysqlClient::queryResultToStringList(MYSQL_RES *res_set,
|
||||
std::list<std::string> &result)
|
||||
{
|
||||
unsigned int numrows = mysql_num_rows(res_set);
|
||||
LOG( Logger::DEBUG, std::string("MySQL query returned number of rows: ").
|
||||
append(TToStr(numrows)).c_str());
|
||||
|
||||
unsigned int num_fields = mysql_num_fields(res_set);
|
||||
MYSQL_ROW row;
|
||||
|
||||
while ((row = mysql_fetch_row(res_set)) != NULL) {
|
||||
std::string rowString;
|
||||
for( unsigned int i = 0; i < num_fields; ++i ) {
|
||||
rowString.append(row[i] ? row[i] : "NULL");
|
||||
rowString.append(",");
|
||||
}
|
||||
if (num_fields > 0)
|
||||
rowString.erase(rowString.end()-1);
|
||||
|
||||
result.push_back(rowString);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue