Logo Search packages:      
Sourcecode: poco version File versions

StatementImpl.cpp

//
// StatementImpl.cpp
//
// $Id: //poco/1.4/Data/src/StatementImpl.cpp#1 $
//
// Library: Data
// Package: DataCore
// Module:  StatementImpl
//
// Copyright (c) 2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
// 
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//


#include "Poco/Data/StatementImpl.h"
#include "Poco/Data/DataException.h"
#include "Poco/Data/AbstractBinder.h"
#include "Poco/Data/Extraction.h"
#include "Poco/Data/BLOB.h"
#include "Poco/SharedPtr.h"
#include "Poco/String.h"
#include "Poco/Exception.h"


namespace Poco {
namespace Data {


StatementImpl::StatementImpl():
      _state(ST_INITIALIZED),
      _extrLimit(upperLimit((Poco::UInt32) Limit::LIMIT_UNLIMITED, false)),
      _lowerLimit(0),
      _columnsExtracted(0),
      _ostr(),
      _bindings()
{
}


00062 StatementImpl::~StatementImpl()
{
}


00067 Poco::UInt32 StatementImpl::execute()
{
      resetExtraction();
      Poco::UInt32 lim = 0;
      if (_lowerLimit > _extrLimit.value())
            throw LimitException("Illegal Statement state. Upper limit must not be smaller than the lower limit.");

      if (_extrLimit.value() == Limit::LIMIT_UNLIMITED)
            lim = executeWithoutLimit();
      else
            lim = executeWithLimit();
            
      if (lim < _lowerLimit)
      {
            throw LimitException("Did not receive enough data.");
      }
      return lim;
}


00087 Poco::UInt32 StatementImpl::executeWithLimit()
{
      poco_assert (_state != ST_DONE);

      compile();

      Poco::UInt32 count = 0;
      Poco::UInt32 limit = _extrLimit.value();
      do
      {
            bind();
            while (hasNext() && count < limit)
            {
                  next();
                  ++count;
            }
      }
      while (canBind() && count < limit);

      if (!canBind() && (!hasNext() || 0 == limit))
            _state = ST_DONE;
      else if (hasNext() && limit == count && _extrLimit.isHardLimit())
            throw LimitException("HardLimit reached. We got more data than we asked for");

      return count;
}


00115 Poco::UInt32 StatementImpl::executeWithoutLimit()
{
      poco_assert (_state != ST_DONE);
      
      compile();

      Poco::UInt32 count = 0;
      do
      {
            bind();
            while (hasNext())
            {
                  next();
                  ++count;
            }
      }
      while (canBind());

      _state = ST_DONE;
      return count;
}


00138 void StatementImpl::compile()
{
      if (_state == ST_INITIALIZED)
      {
            compileImpl();
            _state = ST_COMPILED;

            if (!extractions().size())
            {
                  Poco::UInt32 cols = columnsReturned();
                  if (cols) makeExtractors(cols);
            }

            fixupExtraction();
            fixupBinding();
      }
      else if (_state == ST_RESET)
      {
            resetBinding();
            resetExtraction();
            _state = ST_COMPILED;
      }
}


00163 void StatementImpl::bind()
{
      if (_state == ST_COMPILED)
      {
            bindImpl();
            _state = ST_BOUND;
      }
      else if (_state == ST_BOUND)
      {
            if (!hasNext())
            {
                  if (canBind())
                  {
                        bindImpl();
                  }
                  else
                        _state = ST_DONE;
            }
      }
}


00185 void StatementImpl::reset()
{
      _state = ST_RESET;
      compile();
}


00192 void StatementImpl::setExtractionLimit(const Limit& extrLimit)
{
      if (!extrLimit.isLowerLimit())
            _extrLimit = extrLimit;
      else
            _lowerLimit = extrLimit.value();
}


00201 void StatementImpl::fixupExtraction()
{
      Poco::Data::AbstractExtractionVec::iterator it    = extractions().begin();
      Poco::Data::AbstractExtractionVec::iterator itEnd = extractions().end();
      AbstractExtractor& ex = extractor();
      _columnsExtracted = 0;
      for (; it != itEnd; ++it)
      {
            (*it)->setExtractor(&ex);
            (*it)->setLimit(_extrLimit.value()),
            _columnsExtracted += (int)(*it)->numOfColumnsHandled();
      }
}


00216 void StatementImpl::fixupBinding()
{
      // no need to call binder().reset(); here will be called before each bind anyway
      AbstractBindingVec::iterator it    = bindings().begin();
      AbstractBindingVec::iterator itEnd = bindings().end();
      AbstractBinder& bin = binder();
      std::size_t numRows = 0;
      if (it != itEnd)
            numRows = (*it)->numOfRowsHandled();
      for (; it != itEnd; ++it)
      {
            if (numRows != (*it)->numOfRowsHandled())
            {
                  throw BindingException("Size mismatch in Bindings. All Bindings MUST have the same size");
            }
            (*it)->setBinder(&bin);
      }
}


00236 void StatementImpl::resetBinding()
{
      AbstractBindingVec::iterator it    = bindings().begin();
      AbstractBindingVec::iterator itEnd = bindings().end();
      for (; it != itEnd; ++it)
      {
            (*it)->reset();
      }
}


00247 void StatementImpl::resetExtraction()
{
      Poco::Data::AbstractExtractionVec::iterator it = extractions().begin();
      Poco::Data::AbstractExtractionVec::iterator itEnd = extractions().end();
      for (; it != itEnd; ++it)
      {
            (*it)->reset();
      }
}


00258 void StatementImpl::makeExtractors(Poco::UInt32 count)
{
      for (int i = 0; i < count; ++i)
      {
            const MetaColumn& mc = metaColumn(i);
            switch (mc.type())
            {
                  case MetaColumn::FDT_BOOL:
                  case MetaColumn::FDT_INT8:   
                        addInternalExtract<Int8, std::vector<Int8> >(mc); 
                        break;
                  case MetaColumn::FDT_UINT8:  
                        addInternalExtract<UInt8, std::vector<UInt8> >(mc); 
                        break;
                  case MetaColumn::FDT_INT16:  
                        addInternalExtract<Int16, std::vector<Int16> >(mc); 
                        break;
                  case MetaColumn::FDT_UINT16: 
                        addInternalExtract<UInt16, std::vector<UInt16> >(mc); 
                        break;
                  case MetaColumn::FDT_INT32:  
                        addInternalExtract<Int32, std::vector<Int32> >(mc); 
                        break;
                  case MetaColumn::FDT_UINT32: 
                        addInternalExtract<UInt32, std::vector<UInt32> >(mc); 
                        break;
                  case MetaColumn::FDT_INT64:  
                        addInternalExtract<Int64, std::vector<Int64> >(mc); 
                        break;
                  case MetaColumn::FDT_UINT64: 
                        addInternalExtract<UInt64, std::vector<UInt64> >(mc); 
                        break;
                  case MetaColumn::FDT_FLOAT:  
                        addInternalExtract<float, std::vector<float> >(mc); 
                        break;
                  case MetaColumn::FDT_DOUBLE: 
                        addInternalExtract<double, std::vector<double> >(mc); 
                        break;
                  case MetaColumn::FDT_STRING: 
                        addInternalExtract<std::string, std::vector<std::string> >(mc); 
                        break;
                  case MetaColumn::FDT_BLOB:   
                        addInternalExtract<BLOB, std::vector<BLOB> >(mc); 
                        break;
                  default:
                        throw Poco::InvalidArgumentException("Data type not supported.");
            }
      }
}


00309 const MetaColumn& StatementImpl::metaColumn(const std::string& name) const
{
      Poco::UInt32 cols = columnsReturned();
      for (Poco::UInt32 i = 0; i < cols; ++i)
      {
            const MetaColumn& column = metaColumn(i);
            if (0 == icompare(column.name(), name)) return column;
      }

      throw NotFoundException(format("Invalid column name: %s", name));
}


} } // namespace Poco::Data

Generated by  Doxygen 1.6.0   Back to index