---

RdbMySQLDumpTable.cc

Go to the documentation of this file.

//--------------------------------------------------------------------------
// File and Version Information:
//      $Id: RdbMySQLDumpTable.cc,v 1.3 2005/04/15 17:56:07 salnikov Exp $
//
// Description:
//      Example application for the RdbMySQL package
//
// Environment:
//      Software developed for the BaBar Detector at the SLAC B-Factory.
//
// Author List:
//      Andy Salnikov
//
// Copyright Information:
//      Copyright (C) 2005 SLAC
//
//------------------------------------------------------------------------
#include "BaBar/BaBar.hh"

//-------------
// C Headers --
//-------------
extern "C" {
}

//---------------
// C++ Headers --
//---------------
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <string>
#include <vector>

//-------------------------------
// Collaborating Class Headers --
//-------------------------------
#include "ErrLogger/ErrLog.hh"
#include "RdbMySQL/RdbMySQLQuery.hh"
#include "RdbMySQL/RdbMySQLResult.hh"
#include "RdbMySQL/RdbMySQLRow.hh"
#include "RdbMySQL/RdbMySQLRowIter.hh"
#include "RdbMySQL/RdbMySQLConn.hh"
#include "SysUtils/SysApp.hh"
#include "SysUtils/SysCmdArg.hh"
#include "SysUtils/SysCmdOpt.hh"

//-----------------------------------------------------------------------
// Local Macros, Typedefs, Structures, Unions and Forward Declarations --
//-----------------------------------------------------------------------

namespace {

  template <typename Iter>
  void dashes ( Iter begin, Iter end ) {
    cout.fill('-') ;
    for ( ; begin != end ; ++ begin ) {
      cout << "+" << setw(*begin+2) << "" ;
    }
    cout << "+\n" ;
    cout.fill(' ') ;
  }

}

using namespace std ;

class RdbMySQLDumpTable : public SysApp {

public:

  // Constructors
  RdbMySQLDumpTable( const std::string& appName, bool installErrLog = true ) ;

  // Destructor
  virtual ~RdbMySQLDumpTable() {}

protected:

  virtual int runApp () ;

private:

  // Friends

  // Data members
  SysCmdOpt<std::string>  _optHost ;
  SysCmdOpt<unsigned int> _optPort ;
  SysCmdArg<std::string>  _argDb ;
  SysCmdArg<std::string>  _argTable ;

};

// Constructors
RdbMySQLDumpTable::RdbMySQLDumpTable( const std::string& appName, bool installErrLog )
  : SysApp ( appName, installErrLog )
  , _optHost ( 's', "host", "string", "remote host name", "" )
  , _optPort ( 'p', "port", "number", "remote port number", 0 )
  , _argDb ( "database", "database name" ) 
  , _argTable ( "table", "table name" ) 
{
  addOption ( _optHost ) ;
  addOption ( _optPort ) ;
  addArgument ( _argDb ) ;
  addArgument ( _argTable ) ;
}

int
RdbMySQLDumpTable::runApp ()
{

  // make a conection object
  RdbMySQLConn conn( _optHost.value(), "", "", _argDb.value(), _optPort.value() ) ;

  // open connection
  if ( ! conn.open() ) {
    cerr << "Cannot open connection" << endl ;
    return 1 ;
  }

  // make a query object
  RdbMySQLQuery query( conn ) ;
  
  const char* q = "SELECT * FROM ?" ;
  std::auto_ptr<RdbMySQLResult> res ( query.executePar( q, _argTable.value() ) ) ;
  if ( ! res.get() ) {
    ErrMsg(error) << "query failed: " << conn.error() << endmsg ;
    return 2 ;
  }

  // result header
  const RdbMySQLHeader& header = res->header() ;
  int nf = header.size() ;
  if ( ! nf ) {
    ErrMsg(error) << "empty header" << endmsg ;
    return 3 ;
  }

  // calculate max field sizes
  std::vector<int> sizes( nf, 0 ) ;
  for ( int i = 0 ; i < nf ; ++ i ) {
    const RdbMySQLField& field = header.field(i) ;
    sizes[i] = strlen(field.name()) ;
    if ( sizes[i] < field.max_length() ) {
      sizes[i] = field.max_length() ;
    }
  }

  // print header
  cout.setf( ios::left, ios::adjustfield ) ;
  ::dashes ( sizes.begin(), sizes.end() ) ;
  for ( int i = 0 ; i < nf ; ++ i ) {
    const RdbMySQLField& field = header.field(i) ;
    cout << "| " << setw(sizes[i]) << field.name() << " " ;
  }
  cout << "|\n" ;
  ::dashes ( sizes.begin(), sizes.end() ) ;

  // print all rows
  RdbMySQLRowIter iter ( *res ) ;
  while ( iter.next() ) {
    RdbMySQLRow row = iter.row() ;
    for ( int i = 0 ; i < nf ; ++ i ) {
      const char* str = row.at(i) ;
      if ( ! str ) str = "NULL" ;
      cout << "| " << setw(sizes[i]) << str << " " ;
    }
    cout << "|\n" ;
  }

  ::dashes ( sizes.begin(), sizes.end() ) ;

  return 0 ;
}


//
//  Main function
//
int
main ( int argc, char* argv[] ) 
{
  RdbMySQLDumpTable app ( argv[0] ) ;
  return app.run ( argc, argv ) ;
}

---