2010-04-17 02:25:04 +04:00
/* -------------------------------------------------------------------------- */
2012-01-12 15:29:18 +04:00
/* Copyright 2002-2012, OpenNebula Project Leads (OpenNebula.org) */
2010-04-17 02:25:04 +04:00
/* */
/* Licensed under the Apache License, Version 2.0 (the "License"); you may */
/* not use this file except in compliance with the License. You may obtain */
/* a copy of the License at */
/* */
/* http://www.apache.org/licenses/LICENSE-2.0 */
/* */
/* Unless required by applicable law or agreed to in writing, software */
/* distributed under the License is distributed on an "AS IS" BASIS, */
/* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. */
/* See the License for the specific language governing permissions and */
/* limitations under the License. */
/* -------------------------------------------------------------------------- */
# ifndef MYSQL_DB_H_
# define MYSQL_DB_H_
# include <string>
# include <sstream>
# include <stdexcept>
# include <sys/time.h>
# include <sys/types.h>
# include <unistd.h>
2010-04-21 02:40:16 +04:00
# include "NebulaLog.h"
2010-04-17 02:25:04 +04:00
# include "SqlDB.h"
# include "ObjectSQL.h"
2010-05-07 16:45:27 +04:00
2010-04-17 02:25:04 +04:00
using namespace std ;
2010-05-07 16:45:27 +04:00
# ifdef MYSQL_DB
# include <mysql.h>
2010-04-17 02:25:04 +04:00
/**
* SqliteDB class . Provides a wrapper to the mysql database interface .
*/
class MySqlDB : public SqlDB
{
public :
2011-05-06 19:06:57 +04:00
MySqlDB ( const string & _server ,
int _port ,
const string & _user ,
const string & _password ,
const string & _database ) ;
2010-04-17 02:25:04 +04:00
2010-05-07 16:45:27 +04:00
~ MySqlDB ( ) ;
2010-04-17 02:25:04 +04:00
/**
* Wraps the mysql_query function call
* @ param cmd the SQL command
* @ param obj Callbackable obj to call if the query succeeds
* @ return 0 on success
*/
2010-05-07 16:45:27 +04:00
int exec ( ostringstream & cmd , Callbackable * obj = 0 ) ;
2010-04-17 02:25:04 +04:00
/**
* This function returns a legal SQL string that can be used in an SQL
* statement . The string is encoded to an escaped SQL string , taking into
* account the current character set of the connection .
* @ param str the string to be escaped
* @ return a valid SQL string or NULL in case of failure
*/
2010-05-07 16:45:27 +04:00
char * escape_str ( const string & str ) ;
2010-04-17 02:25:04 +04:00
/**
* Frees a previously scaped string
* @ param str pointer to the str
*/
2010-05-07 16:45:27 +04:00
void free_str ( char * str ) ;
2010-04-17 02:25:04 +04:00
private :
/**
* The MySql connection handler
*/
MYSQL * db ;
2010-05-06 18:16:50 +04:00
2011-05-06 19:06:57 +04:00
/**
* MySQL Connection parameters
*/
string server ;
int port ;
string user ;
string password ;
string database ;
2010-05-06 18:16:50 +04:00
/**
* Fine - grain mutex for DB access
*/
pthread_mutex_t mutex ;
/**
* Function to lock the DB
*/
void lock ( )
{
pthread_mutex_lock ( & mutex ) ;
} ;
/**
* Function to unlock the DB
*/
void unlock ( )
{
pthread_mutex_unlock ( & mutex ) ;
} ;
2010-04-17 02:25:04 +04:00
} ;
2010-05-07 16:45:27 +04:00
# else
//CLass stub
class MySqlDB : public SqlDB
{
public :
MySqlDB (
string server ,
2010-11-10 20:44:13 +03:00
int port ,
2010-05-07 16:45:27 +04:00
string user ,
string password ,
2011-05-06 20:14:23 +04:00
string database )
2010-05-07 16:45:27 +04:00
{
throw runtime_error ( " Aborting oned, MySQL support not compiled! " ) ;
} ;
~ MySqlDB ( ) { } ;
int exec ( ostringstream & cmd , Callbackable * obj = 0 ) { return - 1 ; } ;
char * escape_str ( const string & str ) { return 0 ; } ;
void free_str ( char * str ) { } ;
} ;
# endif
2010-04-17 02:25:04 +04:00
2010-09-02 22:44:14 +04:00
# endif /*MYSQL_DB_H_*/