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]                          *
 *                                                                              *
 *   Licensed under the MIT License <http://opensource.org/licenses/MIT>.       *   
 *   SPDX-License-Identifier: MIT                                               *
 ********************************************************************************/
 
#ifndef database_H
#define database_H
 
#include "tkn_config.h"
 
#ifdef HAS_ROOT
#include "TClass.h"
#endif
 
#include <vector>
#include <map>
 
#include "tkn_sqlite3.h"
#include <tklog.h>
#include <tkstring.h>
#include <tkdb_table.h>
 
namespace tkn {

class tkdatabase {
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())
   std::map<tkstring, std::vector<std::vector<tkdb_column *>>> fSQLColumnBindings;
   tkstring fName{};
   tkstring fFullName{};
 
public:
   tkdatabase(bool opening_db = true);
   virtual ~tkdatabase();
 
   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
};
} // namespace tkn
 
#endif