HarpoDB.h

Go to the documentation of this file.

#ifndef HARPODB_H
#define HARPODB_H
//
// HarpoDB
//
// Harpo Database Manipulation
//

#include "TList.h"
#include "TObjString.h"
#include "TString.h"
#include <map>
#include <list>

class TSQLServer;

// Name,valus for one row
typedef std::map<TString,TString> dbFields;
//list of pams for all resulting rows
typedef std::list<dbFields> dbResults;

// One run record monipulation, fields updates
class HarpoDBRow {
  
 public:
  HarpoDBRow() : run(0) {}
  virtual ~HarpoDBRow() {}

  void list();

  Bool_t Lookup(const TString field);

  Bool_t GetVal(const TString field, TString &val);

  //Set value for existing field
  Bool_t SetVal(const TString field, TString val);
  Bool_t SetVal(const TString field, Long_t val);
  Bool_t SetVal(const TString field, Double_t val);

  // Get Map of fields types
  dbFields& GetFields() { return fields; }
  // Set Map of fields types
  void SetFields(dbFields f);

  //Erase field from map
  void Erase(const TString &field) { fields.erase(field); }

  Long_t GetNrun() { return run; }
  void SetRun(Long_t r) { run = r; SetVal("run",r); }


 private:
  Long_t run;
  dbFields fields;
};

// Work with one table , select, update, insert records, record iteretor 
class HarpoDBTable {

public:
  HarpoDBTable() {}
  ~HarpoDBTable() {}

  const TString GetName() { return name; }
  Bool_t GetRunRow(Long_t nrun,HarpoDBRow &row);

  Bool_t InsertRunRow(HarpoDBRow row,const TString option="");

  Bool_t UpdateRunRow(HarpoDBRow row,const TString option="");

  Bool_t DeleteRun(Long_t run);

  Bool_t Lookup(const TString field);

  Bool_t GetType(const TString field, TString &type);

  Bool_t TypeExpression(const TString field);

  // Get Map of fields types
  dbFields & GetFiledsTypes() { return ftypes; }

  //Erase field from map
  void Erase(const TString &field) { ftypes.erase(field); }

  // Connect to table and read type info ( use by HarpoDB)
  Bool_t Open(const char *dbnm, const char *tbl); 

  // list rows in table
  void list();
  // Make root class
  void MakeClass(TString cname,TString option="");
private:
  TString  name;
  dbFields ftypes;
};

// Open/Close DB, list tables , get table class pointer
class HarpoDB {

public:
  HarpoDB() : fTables(NULL) {}
  virtual~HarpoDB() { delete fTables;}

  static void SetUser(TString u, TString p);
  static void SetDBUrl(const TString &DbUrl) { dbUrl = DbUrl; }
  Bool_t Open(const TString *DbUrl=NULL);
  Bool_t Close();
  Bool_t GetTable(const TString name,HarpoDBTable *table)
                             {return GetTable(name,*table);}
  Bool_t GetTable(const TString name,HarpoDBTable &table);

  Bool_t ExecQuery(const TString sql,dbResults *list = NULL);

  // list tables in db
  void list();
  TList *GetListOfTables() { return fTables; }

  //static
  static  TSQLServer *GetDBServer() { return db; }
  static const char *GetDBName();

private:
  static TSQLServer *db;
  static TString dbUrl;
  static TString dbuser;
  static TString dbpw;
  TList *fTables;
};
#endif
// Local Variables:
// mode: c++
// End: