DbiOutRowStream.cxx

Go to the documentation of this file.

// $Id: DbiOutRowStream.cxx,v 1.19 2006/08/08 10:51:32 west Exp $
//
// DbiOutRowStream

#include <sstream>

#include "DatabaseInterface/DbiFieldType.h"
#include "DatabaseInterface/DbiOutRowStream.h"
#include "DatabaseInterface/DbiTableMetaData.h"
#include "LeakChecker/Lea.h"
#include "MessageService/MsgService.h"
#include "Util/UtilString.h"
#include "Validity/VldTimeStamp.h"

ClassImp(DbiOutRowStream)

#define OUT(t,v)                         \
  if ( ! StoreDefaultIfInvalid(t) ) {    \
    ostringstream out;                   \
    out << setprecision(16)<< v;         \
    Store(out.str());                    \
  }                                      \

// If writing unsigned dat as signed, convert bit pattern to signed, 
// extending sign bit if necessary.                                  
// For BIGINT (size 8) make an exception.  It's used only as
// an alternative to unsigned int so can written without conversion.
#define OUT2(t,v)                         \
  const DbiFieldType& fType = this->ColFieldType(this->CurColNum());             \
  if ( fType.IsSigned() && fType.GetSize() != 8 ) {                              \
    Int_t v_signed = (Int_t) v;                                                  \
    if ( fType.GetType() == Dbi::kTiny  && v & 0x80   ) v_signed |= 0xffffff00;  \
    if ( fType.GetType() == Dbi::kShort && v & 0x8000 ) v_signed |= 0xffff0000;  \
    OUT(Dbi::kInt,v_signed); }                                                   \
  else {                                                                         \
    OUT(t,v);                                                                    \
  }                                                                              \

//   Definition of static data members
//   *********************************

CVSID("$Id: DbiOutRowStream.cxx,v 1.19 2006/08/08 10:51:32 west Exp $");

//    Definition of all member functions (static or otherwise)
//    *******************************************************
//
//    -  ordered: ctors, dtor, operators then in alphabetical order.

//.....................................................................

DbiOutRowStream::DbiOutRowStream(const DbiTableMetaData* metaData) :
DbiRowStream(metaData),
fBadData(kFALSE)
{
//
//
//  Purpose:  Default constructor
//
//  Arguments:
//     metaData in  Meta data for table to be written to..

  LEA_CTOR    //Leak Checker

  MSG("Dbi", Msg::kVerbose) << "Creating DbiOutRowStream" << endl;

}


//.....................................................................

DbiOutRowStream::~DbiOutRowStream() {
//
//
//  Purpose: Destructor

  LEA_DTOR    //Leak Checker

  MSG("Dbi", Msg::kVerbose) << "Destroying DbiOutRowStream" << endl;

}

//.....................................................................

DbiOutRowStream& DbiOutRowStream::operator<<(Bool_t src) {
                                     OUT(Dbi::kBool,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(Char_t src) {
                                     OUT(Dbi::kChar,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(const Char_t* src) {
                                     OUT(Dbi::kString,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(Short_t src) {
                                     OUT(Dbi::kShort,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(UShort_t src) {
                                     OUT2(Dbi::kUShort,src); return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(Int_t src) {
                                     OUT(Dbi::kInt,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(UInt_t src) {
                                     OUT2(Dbi::kUInt,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(Float_t src) {
                                     OUT(Dbi::kFloat,src); return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(Double_t src) {
                                     OUT(Dbi::kDouble,src);  return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(const string& src) {
                                    OUT(Dbi::kString,src); return *this;}

DbiOutRowStream& DbiOutRowStream::operator<<(const VldTimeStamp& src) {
  if ( ! StoreDefaultIfInvalid(Dbi::kDate) )
                          Store(Dbi::MakeDateTimeString(src).c_str());
  return *this;
}




//.....................................................................

Bool_t DbiOutRowStream::StoreDefaultIfInvalid(Dbi::DataTypes type) {
//
//
//  Purpose:  Store default value if illegal type supplied.
//
//  Arguments: 
//    type         in    Type of supplied value.
//
//  Return:    kTRUE if illegal type supplied.
//
//  Contact:   N. West
//
//  Specification:-
//  =============
//
//  o If type of supplied value is incompatible with current column
//    type store default value, report error and return kTRUE, otherwise
//    return kFALSE.

//  Program Notes:-
//  =============

//  None.

  DbiFieldType typeSupplied(type);
  DbiFieldType typeRequired(CurColFieldType());
  if ( typeSupplied.IsCompatible(typeRequired) ) return kFALSE;

  string udef = typeRequired.UndefinedValue();
  MAXMSG("Dbi",Msg::kError,20) 
      << "In table " << TableNameTc() 
      << " column "<< CurColNum() 
      << " (" << CurColName() << ")"
      << " of type " << typeRequired.AsString()
      << " is incompatible with user type " << typeSupplied.AsString()
      << ", value \"" << udef
      << "\" will be substituted." <<  endl;
  Store(udef.c_str());
  fBadData = kTRUE;
  return kTRUE; 

}
//.....................................................................

void DbiOutRowStream::Store(const string& str)  {
//
//
//  Purpose: Store string value as comma separated values but exclude SeqNo.  
//
//  Arguments: 
//    str          in    Value to be stored.
//
//  Return:    
//
//  Contact:   N. West
//
//  Specification:-
//  =============
//
//  o Store string value in CSV with separator from 
//    previous value and increment fNumValues.  If required
//    enclose value in quotes. If storing the SeqNo column then
//    store a single '?' instead which will be replace during
//    SQL generation - see 

//  Program Notes:-
//  =============

//  None.

  UInt_t concept = CurColFieldType().GetConcept();
  string delim = "";
  if (    concept == Dbi::kString 
       || concept == Dbi::kDate 
       || concept == Dbi::kChar ) delim = "\'";  

  if ( CurColNum()> 1 ) fCSV += ',';
  fCSV += delim;
  if ( concept != Dbi::kString ) fCSV += str;
//  When exporting strings, take care of special characters.
  else {
    UtilString::MakePrintable(str.c_str(),fCSV);
  }
  fCSV += delim;
  IncrementCurCol();
}

---