1
0
mirror of https://github.com/OpenNebula/one.git synced 2024-12-23 17:33:56 +03:00
one/include/MySqlDB.h
2018-01-02 18:27:37 +01:00

168 lines
4.3 KiB
C++

/* -------------------------------------------------------------------------- */
/* Copyright 2002-2018, OpenNebula Project, OpenNebula Systems */
/* */
/* 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 <queue>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include "NebulaLog.h"
#include "SqlDB.h"
#include "ObjectSQL.h"
using namespace std;
#ifdef MYSQL_DB
#include <mysql.h>
/**
* SqliteDB class. Provides a wrapper to the mysql database interface.
*/
class MySqlDB : public SqlDB
{
public:
MySqlDB(const string& _server,
int _port,
const string& _user,
const string& _password,
const string& _database);
~MySqlDB();
/**
* 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
*/
char * escape_str(const string& str);
/**
* Frees a previously scaped string
* @param str pointer to the str
*/
void free_str(char * str);
/**
* Returns true if the syntax INSERT VALUES (data), (data), (data)
* is supported
*
* @return true if supported
*/
bool multiple_values_support();
protected:
/**
* 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
*/
int exec(ostringstream& cmd, Callbackable* obj, bool quiet);
private:
/**
* Number of concurrent DB connections.
*/
static const int DB_CONNECT_SIZE;
/**
* The MySql connection pool handler
*/
queue<MYSQL *> db_connect;
/**
* Cached DB connection to escape strings (it uses the server character set)
*/
MYSQL * db_escape_connect;
/**
* MySQL Connection parameters
*/
string server;
int port;
string user;
string password;
string database;
/**
* Fine-grain mutex for DB access (pool of DB connections)
*/
pthread_mutex_t mutex;
/**
* Conditional variable to wake-up waiting threads.
*/
pthread_cond_t cond;
/**
* Gets a free DB connection from the pool.
*/
MYSQL * get_db_connection();
/**
* Returns the connection to the pool.
*/
void free_db_connection(MYSQL * db);
};
#else
//CLass stub
class MySqlDB : public SqlDB
{
public:
MySqlDB(
string server,
int port,
string user,
string password,
string database)
{
throw runtime_error("Aborting oned, MySQL support not compiled!");
};
~MySqlDB(){};
char * escape_str(const string& str){return 0;};
void free_str(char * str){};
bool multiple_values_support(){return true;};
protected:
int exec(ostringstream& cmd, Callbackable* obj, bool quiet){return -1;};
};
#endif
#endif /*MYSQL_DB_H_*/