PgPostBankBackupLog.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PgPostBankBackupLog.cc

#include "PgPostBankBackupLog.h"

#include <RDBC/TSQL.h>
#include <RDBC/TSQLConnection.h>
#include <RDBC/TSQLDriverManager.h>
#include <RDBC/TSQLPreparedStatement.h>

#include <TString.h>

#include <cassert>
#include <cstdlib>
#include <exception>
#include <fstream>
#include <iostream>
#include <sstream>

using namespace std;

PgPostBankBackupLog::TSQLConnection_PTR PgPostBankBackupLog::con(
    static_cast<TSQLConnection*>(nullptr));

PgPostBankBackupLog::PgPostBankBackupLog(const std::string& TableName,
                                         const std::string& Tag)
  : verbosity(1)
  , tablename(TableName)
  , tag(Tag)
  , pstmt(nullptr)
{
}

PgPostBankBackupLog::~PgPostBankBackupLog()
{
}

void PgPostBankBackupLog::Init()
{
  try
  {
    if (!con)
    {
      const TString s_con = "dsn=calibrations_backup_log; uid=phnxrc; pwd= ";

      if (Verbosity() >= 1)
        cout << "PgPostBankBackupLog::Init - connect to " << s_con << endl;

      con = gSQLDriverManager->GetConnection(s_con);
      if (!con)
      {
        cout << "PgPostBankBackupLog::Init - Error - "
             << "cannot init connection to " << s_con << endl;
        exit(1);
      }
    }

    if (!pstmt)
    {
      assert(con);

      ostringstream sqlcmd;
      sqlcmd << "insert into calib_log (tablename,rid,ops,tag) values ('"
             << tablename << "',?,?,'" << tag << "');";

      if (Verbosity() >= 1)
        cout << "PgPostBankBackupLog::Init - make TSQLPreparedStatement "
             << sqlcmd.str() << endl;
      pstmt = con->PrepareStatement(sqlcmd.str().c_str());

      if (!pstmt)
      {
        cout << "PgPostBankBackupLog::Init - Error - "
             << "cannot prepare statement " << sqlcmd.str() << endl;
        exit(1);
      }
    }
  }
  catch (std::exception& e)
  {
    cout << "PgPostBankBackupLog::Init - Error - "
         << "Initialization error "
         << e.what() << endl;
    exit(1);
  }
}

void PgPostBankBackupLog::Log(const int rid, const enu_ops ops)
{
  Init();

  if (Verbosity() >= 2)
  {
    cout << "PgPostBankBackupLog::Log - log rid = " << rid << ", ops = "
         << ops << " for table " << tablename << " and tag " << tag << endl;
  }

  assert(con);
  assert(pstmt);

  pstmt->SetInt(1, rid);
  pstmt->SetInt(2, (int) ops);
  try
  {
    const int res = pstmt->ExecuteUpdate();

    if (res == 0)
    {
      cout << "PgPostBankBackupLog::Log - Error - "
           << "DATABASE: commit failed. "
           << "Make sure you commit to the writable database " << endl;
      exit(1);
    }

    con->Commit();
  }
  catch (TSQLException& e)
  {
    cout << "PgPostBankBackupLog::Log - Error - "
         << " Exception caught during connection->commit()" << endl;
    cout << e.GetMessage() << endl;
    exit(1);
  }
}