storage_manager.h

Go to the documentation of this file.

#ifndef STORAGE_MANAGER_H
#define STORAGE_MANAGER_H

#include "vectors.h"
#include "timer.h"
#include "hazy_model.h"
#include "postgres_util.h"


// TODO, it can get the names as well...
template<class T>
struct sParse_IDvec {
  int id;
  T f;
  static int parse_tuple(PGresult *res, int index, sParse_IDvec<T> &c) {
    c.id = atoi(PQgetvalue(res, index, 0));
    std::string f_vec = PQgetvalue(res, index, 1);
    int r = c.f.fromPSQL(f_vec);
    return r;
  }
};


typedef int key; // primary key change it
typedef int sClass;

struct storage_spec {
  // This indicates whether the model needs to be rebuilt or is an existing model.
  bool bRebuild;
  // entity table name in the database
  std::string entity_table_name;
  
  hazy_model::strategy strategy;

  int unique_id;
};

template <class T> 
class Storage_Manager {
 protected:
  std::string entity_table_name;
  Hazy_Database *db_conn;
  
 public:
  Storage_Manager() {db_conn = NULL;}
  Storage_Manager(Hazy_Database *db_conn) {this->db_conn = db_conn;}
  
  virtual void resort(struct hazy_model &m) = 0;
  
  virtual void incrementalUpdate(struct hazy_model &m, double &waste_time) = 0;
  
  // ************** Load Methods
  // Insert tuple: This only populates the internal table.
  // THIS DOES NOT DO SORTING. If you build the table using this function
  // THEN YOU NEED TO CALL RESORT.
  virtual void insertEntity(struct hazy_model &m, key e, T f) = 0; 
  
  // this is the incremental maintenance call on adding in a new feature.
  // virtual void addEntity(struct hazy_model &m, key e, T f) = 0;
  
  // Repeated inserts. Overload for bulk loading.
  // TODO: Make strongly typed FVector/SVector!  
  virtual void loadFromFile(struct hazy_model &m, const char* szFileName); 
  // This fails if there is no db connection in the constructor...
  virtual int loadFromDatabase(struct hazy_model &m);
  
  // **********************
  // Read Methods
  // Single Entity Read
  virtual void getEntityClass(key e, sClass &c, struct hazy_model &m) = 0;
  // All Members Read
  virtual void getNumInClass(sClass c, int &nClass, struct hazy_model &m, double& waste_time) = 0;
  
  virtual ~Storage_Manager();
};

#include "storage_manager.hxx"
#endif

---