RDBCTestString.C

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

#include <TError.h>
#include <TString.h>
#include <TFile.h>
#include <TF1.h>
#include <RDBC/TSQLDriverManager.h>
#include <RDBC/TSQLConnection.h>
#include <RDBC/TSQLDatabaseMetaData.h>
#include <RDBC/TSQLResultSet.h>
#include <RDBC/TSQLResultSetMetaData.h>
#include <RDBC/TSQLPreparedStatement.h>
#include <RDBC/TSQLCallableStatement.h>
#include <RDBC/TSQLTypes.h>
extern "C" {
#include <stdlib.h>
};
//___________________________________________________________________
Int_t RDBCTestInt(const Text_t* dsn,
               const Text_t* usr="",
               const Text_t* pwd="")
{
   //    Open a connection...
   TSQLConnection* myConnection = NULL;
   if(usr!="" && pwd !=""){
     if(getenv("VERBOSE"))
       printf( "connecting with: dsn= %s usr=%s pwd=%s\n",dsn,usr,pwd);
     myConnection = TSQLDriverManager::GetConnection( dsn, usr,  pwd );
   } else{
     if(getenv("VERBOSE"))
       printf( "connecting with: dsn= %s \n",dsn);
     myConnection = TSQLDriverManager::GetConnection( dsn );
   }

  if(!myConnection) {
     printf( "failed to connect: dsn= %s usr=%s pwd=%s\n",dsn,usr,pwd);
     printf("exiting...\n");
     return -1;  // return on error
   }else
     printf("connected!!!\n");
   TSQLStatement* stmt = myConnection->CreateStatement();

   stmt->ExecuteUpdate( "drop table string_table" );
   stmt->ExecuteUpdate( "create table string_table (a_string varchar(255) not null)" );
   TSQLPreparedStatement* pstmt = 
     myConnection->PrepareStatement("insert into string_table (a_string) values(?)");
   
   Text_t *  some_string = "a short string";    
   pstmt->SetString(1,some_string);
   pstmt->ExecuteUpdate(""); 
   
   some_string = "a much much longer string that may overrun internal buffers if the odbc drivers are not written correctly";
   //             123456789_123456789_123456789_123456789_123456789_123456789_123456789_123456789_123456789_123456789_12345  less than 255 chars so its legal!
   pstmt->SetString(1,some_string);
   pstmt->ExecuteUpdate(""); 
   
   TSQLResultSet* rs = stmt->ExecuteQuery("select a_string from string_table order by a_string");
   rs->Next(); // goto the first row
   TString  my_string = rs->GetString(1); 
   printf ("retrieved: %s\n",my_string.Data() );        
   rs->Next(); // goto the next row
   my_string = rs->GetString(1); 
   printf ("retrieved: %s\n",my_string.Data()); 


   myConnection->Close();
   return 0;
}

//___________________________________________________________________
void Catch(TSQLException* e)
{ 
   // handle exceptions
   
   TString str = e->GetMessage();
   printf("SQL Error: %s\n",str.Data()); 
}


#ifdef STANDALONE

#include <TROOT.h>
#include <TSystem.h>
#include <iostream>

//---- Main program ------------------------------------------------------------

TROOT root("RDBCTestInt","Test RDBC TSQLDriverManager and TSQLConnection");

int main(int argc, char **argv)
{

   gSystem->Load("libRDBC");
   Int_t ret = -1;

   if(argc < 2 || argc > 4){
     printf ("usage: RDBCTestInt dsn [usr] [password]\n");
     return ret;
   }

   if(argc==2) 
     ret=RDBCTestInt(argv[1]);
   if(argc==3) 
     ret=RDBCTestInt(argv[1],argv[2]);
   if(argc==4)
     ret=RDBCTestInt(argv[1],argv[2],argv[3]);

   return ret;
}
#endif