1
0
mirror of https://github.com/OpenNebula/one.git synced 2024-12-22 13:33:52 +03:00
one/include/SqliteDB.h
2023-01-09 12:23:19 +01:00

107 lines
3.2 KiB
C++

/* -------------------------------------------------------------------------- */
/* Copyright 2002-2023, 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 SQLITE_DB_H_
#define SQLITE_DB_H_
#include <string>
#include <sstream>
#include <stdexcept>
#include "SqlDB.h"
#include "ObjectSQL.h"
#ifdef SQLITE_DB
#include <sqlite3.h>
/**
* SqliteDB class. Provides a wrapper to the sqlite3 database interface. It also
* provides "global" synchronization mechanism to use it in a multithread
* environment.
*/
class SqliteDB : public SqlDB
{
public:
SqliteDB(const std::string& db_name, int timeout);
~SqliteDB();
/**
* This function returns a legal SQL string that can be used in an SQL
* statement.
* @param str the string to be escaped
* @return a valid SQL string or NULL in case of failure
*/
char * escape_str(const std::string& str) const override;
/**
* Frees a previously scaped string
* @param str pointer to the str
*/
void free_str(char * str) const override;
protected:
/**
* Wraps the sqlite3_exec function call, and locks the DB mutex.
* @param sql_cmd the SQL command
* @param callbak function to execute on each data returned, watch the
* mutex you block in the callback.
* @param arg to pass to the callback function
* @return 0 on success
*/
int exec_ext(std::ostringstream& cmd, Callbackable *obj, bool quiet) override;
private:
/**
* Fine-grain mutex for DB access
*/
std::mutex _mutex;
/**
* Pointer to the database.
*/
sqlite3 * db;
};
#else
//CLass stub
class SqliteDB : public SqlDB
{
public:
SqliteDB(const std::string& db_name, int timeout)
{
throw std::runtime_error("Aborting oned, Sqlite support not compiled!");
}
~SqliteDB() = default;
char * escape_str(const std::string& str) const override { return 0; }
void free_str(char * str) const override {};
protected:
int exec_ext(std::ostringstream& cmd, Callbackable *obj, bool quiet) override
{
return -1;
}
};
#endif
#endif /*SQLITE_DB_H_*/