MJDatabaseConnection.cc

Go to the documentation of this file.

//---------------------------------------------------------------------------//
//bb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nu//
//                                                                           //
//                         MAJORANA Simulation                               //
//                                                                           //
//      This code implementation is the intellectual property of the         //
//      MAJORANA Collaboration. It is based on Geant4, an intellectual       //
//      property of the RD44 GEANT4 collaboration.                           //
//                                                                           //
//                        *********************                              //
//                                                                           //
//    Neither the authors of this software system, nor their employing       //
//    institutes, nor the agencies providing financial support for this      //
//    work  make  any representation or  warranty, express or implied,       //
//    regarding this software system or assume any liability for its use.    //
//    By copying, distributing or modifying the Program (or any work based   //
//    on on the Program) you indicate your acceptance of this statement,     //
//    and all its terms.                                                     //
//                                                                           //
//bb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nubb0nu//
//---------------------------------------------------------------------------//
//                                                          
// $Id: MJDatabaseConnection.cc,v 1.1.1.1 2004/11/08 17:53:42 gast Exp $ 
//      
// CLASS IMPLEMENTATION:  MJDatabaseConnection.cc
//
//---------------------------------------------------------------------------//
// 
//---------------------------------------------------------------------------//
//---------------------------------------------------------------------------//
//
#include <iostream>
#include <fstream>
#include <map>
using namespace std;


//---------------------------------------------------------------------------//

#include "database/MJDatabaseConnection.hh"         //Present MJ Class Headers 
#include "io/MJLogger.hh"
//---------------------------------------------------------------------------//

static void
exit_nicely(PGconn *conn)
{
        PQfinish(conn);
        exit(1);
}
string MJDatabaseConnection::PG_VERSION="";

MJDatabaseConnection::MJDatabaseConnection()
{
}
MJDatabaseConnection::MJDatabaseConnection(const MJDatabaseConnection & other)
{

}

MJDatabaseConnection::~MJDatabaseConnection()
{
}
void
MJDatabaseConnection::getConnectionInfo(){

  int size = 512;
  char buff[512];
  map<string,string>  propMap;
  ifstream infile("MJDBProperties.txt", ios::in);
  if(!infile){
    MJLog(trace) << "Cannot open properties file MJDBProperties.txt" 
                 << endl;
    MJLog(trace) << "Make sure you have this file exists in the current directory."
                 << endl;
  }
  char* token = "=";
  while(!infile.eof()){ 
      infile.getline(buff, size);
    while('#' !=buff[0]  
          && buff[0] != '\n'
          && buff[0] != '\0'
          && !infile.eof()){
      string key(strtok(buff, token));
      string value(strtok(0, token));
      propMap.insert(make_pair(key, value));
      MJLog(trace) << "Key: " << key << "  value: "<< value << endl;
      infile.getline(buff, size);
      }
    
  }
  string host = propMap["databaseURL"];
  string name = propMap["databaseName"];
  string user = propMap["userName"];
  string passwd = propMap["password"];
  string portNo = propMap["port"];
  string conninfo =  "hostaddr=" +  host + " dbname="+
    name + " user=" + user + " password=" + passwd +
    " port=" + portNo;

//  fconninfo = conninfo.c_str();
// a test 
  fconninfo="hostaddr=128.3.11.122 dbname=majorana user=akbarm password= port=5432";

  //the version of postgres that is used
  //this is important in reading data from database
  PG_VERSION = propMap["pg_version"];

  MJLog(trace)<< conninfo << endl;
  MJLog(trace)<< fconninfo << endl;
}

PGresult*
MJDatabaseConnection::qResult(string sqlCmd)
{
  if(! connect())exit_nicely(fconn);
  /* Start a transaction block */
  fres = query("BEGIN");
  PQclear(fres);

  string command = "DECLARE myportal BINARY CURSOR FOR " + sqlCmd;
  fres = query(command);
  PQclear(fres);
  fres = PQexec(fconn, "FETCH ALL in myportal");
  if(PQresultStatus(fres) != PGRES_TUPLES_OK){
    MJLog(trace) << "FETCH ALL failed: " << PQerrorMessage(fconn) << endl;
    PQclear(fres);
    exit_nicely(fconn);
  }
  return fres;
}
void
MJDatabaseConnection::qDone(PGresult* res)
{
  PQclear(res);
  res = query("CLOSE myportal");
  
  /* end the transaction */
  res = query("END");
  PQclear(res);
  
  /* close the connection to the database and cleanup */
  PQfinish(fconn);

}
PGresult*
MJDatabaseConnection::query(string cmd)
{
  PGresult* res;
  res = PQexec(fconn, cmd.c_str());
  if (PQresultStatus(res) != PGRES_COMMAND_OK){
    MJLog(trace)<< cmd << " command failed: " << PQerrorMessage(fconn) << endl;
    PQclear(res);
    exit_nicely(fconn);
  }
  return res;
  
}
bool
MJDatabaseConnection::connect()
{
  getConnectionInfo();
  bool retVal = true;
  /* Make a connection to the database */
  fconn = PQconnectdb(fconninfo);

  
  /* Check to see that the backend connection was successfully made */
  if (PQstatus(fconn) != CONNECTION_OK)
    {
      MJLog(trace)<<"Connection to database \n" 
                  << PQdb(fconn)<<" failed.\n"<<endl;
      MJLog(trace)<<"----------- more information ----------------- \n"<<endl;
      MJLog(trace)<<" dbname "<<PQdb(fconn)
                  <<" username "<<PQuser(fconn)
                  <<" host "<<PQhost(fconn)
                  <<" port "<<PQport(fconn)
                  <<" option "<<PQoptions(fconn)
                  <<" status "<<PQstatus(fconn)<<"\n"<<endl; 
      MJLog(trace)<<"----------------------------------------------- \n"<<endl;
      MJLog(trace) << PQerrorMessage(fconn)<< endl;
      retVal = false;
    }
  return retVal;
}

---