tkdatabase.h

/********************************************************************************
 *   Copyright (c) : Université de Lyon 1, CNRS/IN2P3, UMR5822,                 *
 *                   IP2I, F-69622 Villeurbanne Cedex, France                   *
 *                   Normandie Université, ENSICAEN, UNICAEN, CNRS/IN2P3,       *
 *                   LPC Caen, F-14000 Caen, France                             *
 *   Contibutor(s) :                                                            *
 *      Jérémie Dudouet jeremie.dudouet@cnrs.fr [2020]                          *
 *      Diego Gruyer    diego.gruyer@cnrs.fr    [2020]                          *
 *                                                                              *
 *    This software is governed by the CeCILL-B license under French law and    *
 *    abiding by the  rules of distribution of free  software.  You can use,    *
 *    modify  and/ or  redistribute  the  software under  the  terms of  the    *
 *    CeCILL-B license as circulated by CEA, CNRS and INRIA at the following    *
 *    URL \"http://www.cecill.info\".                                           *
 *                                                                              *
 *    As a counterpart to the access  to the source code and rights to copy,    *
 *    modify  and redistribute granted  by the  license, users  are provided    *
 *    only with a limited warranty  and the software's author, the holder of    *
 *    the economic  rights, and the  successive licensors have  only limited    *
 *    liability.                                                                *
 *                                                                              *
 *    In this respect, the user's attention is drawn to the risks associated    *
 *    with loading,  using, modifying  and/or developing or  reproducing the    *
 *    software by the user in light of its specific status of free software,    *
 *    that  may mean that  it is  complicated to  manipulate, and  that also    *
 *    therefore  means that it  is reserved  for developers  and experienced    *
 *    professionals having in-depth  computer knowledge. Users are therefore    *
 *    encouraged  to load  and test  the software's  suitability  as regards    *
 *    their  requirements  in  conditions  enabling the  security  of  their    *
 *    systems  and/or data to  be ensured  and, more  generally, to  use and    *
 *    operate it in the same conditions as regards security.                    *
 *                                                                              *
 *    The fact that  you are presently reading this means  that you have had    *
 *    knowledge of the CeCILL-B license and that you accept its terms.          *
 ********************************************************************************/
 
#ifndef database_H
#define database_H
 
#include "tkn_config.h"
 
#ifdef HAS_ROOT
#include "TClass.h"
#endif
 
#include <vector>
#include <map>
#include <mutex>
 
#include <sqlite3.h>
#include <tklog.h>
#include <tkstring.h>
#include <tkdb_table.h>
 
namespace tkn {
 
class tkdatabase{
 
    static std::recursive_mutex& get_mutex() {
        static std::recursive_mutex mutex;
        return mutex;
    }
 
    static tkdatabase *g_database;
 
public:
    static tkdatabase *the_database();  
 
protected:
    sqlite3 * fDataBase = nullptr;
    std::map<tkstring,tkdb_table> fTables;
    int fMode{};
    bool fReading = false;
    std::vector<tkstring> fViews;
    std::map<tkstring, sqlite3_stmt*> fSQLStatement; // sql statements to select rows and loop over them (begin() next() end())
    tkstring fName{};
    tkstring fFullName{};
 
public:
    tkdatabase(bool opening_db=true);
    virtual ~tkdatabase() = default;
 
    void set_sql_db(sqlite3 * _sql_db){fDataBase = _sql_db;}
    sqlite3 * get_sql_db(){return fDataBase;}
 
    sqlite3* open(tkstring _db, tkstring _option="OPEN");
    void close();
    tkstring get_name(){return fName;}
    tkstring get_full_name(){return fFullName;}
 
    void load_tables();
    void load_views();
    void add_table(tkdb_table &_table);
    bool has_table(const tkstring &_table_name);
    bool has_view(const tkstring &_table_name);
 
    std::vector<tkstring>& get_views(){return fViews;}
 
    tkdb_table &new_table(tkstring _table_name);
    void remove_table(tkstring _table_name);
 
    tkdb_table* get_table(tkstring _table_name);
    static int getdir (tkstring _dir, std::vector<tkstring> &_files);
    static void list_files(tkstring _dir, tkstring _pattern);
 
    void print(const tkstring &_properties="", const tkstring &_opt="");
 
    tkdb_table &operator[](const char* _table) {
             return fTables[_table];
          }
 
    void begin(tkstring _selection, tkstring _from, tkstring _condition="", tkstring _loop_name="default", tkstring _extra_cmd=""); 
    bool next(const tkstring &_loop_name="default"); 
    void end(const tkstring &_loop_name="default"); 
 
    int count(const tkstring &_from, const tkstring &_condition="");
    tkstring get_value(tkstring _selection, tkstring _from, tkstring _condition=""); 
 
//private:
    int exec_sql(const char *_cmd);
 
#define gdatabase (tkn::tkdatabase::the_database())
 
#ifdef HAS_ROOT
    ClassDef(tkdatabase,0);
#endif
};
}
 
#endif