sapphire/src/common/Database/PreparedStatement.cpp

#include "PreparedStatement.h"
#include "DbConnection.h"

#include <MySqlConnector.h>

#include <string.h>
#include <sstream>

Core::Db::PreparedStatement::PreparedStatement( uint32_t index ) :
  m_stmt( nullptr ),
  m_index( index )
{
}

Core::Db::PreparedStatement::~PreparedStatement()
{
}

void Core::Db::PreparedStatement::bindParameters()
{
  assert( m_stmt );

  uint8_t i = 1;
  for( ; i < m_statementData.size(); i++ )
  {
    switch( m_statementData[ i ].type )
    {
      case TYPE_BOOL:
        m_stmt->setBoolean( i, m_statementData[ i ].data.boolean );
        break;
      case TYPE_UI:
        m_stmt->setUInt( i, m_statementData[ i ].data.ui32 );
        break;
      case TYPE_I:
        m_stmt->setInt( i, m_statementData[ i ].data.i32 );
        break;
      case TYPE_UI64:
        m_stmt->setUInt64( i, m_statementData[ i ].data.ui64 );
        break;
      case TYPE_I64:
        m_stmt->setInt64( i, m_statementData[ i ].data.i64 );
        break;
      case TYPE_DOUBLE:
        m_stmt->setDouble( i, m_statementData[ i ].data.d );
        break;
      case TYPE_STRING:
        m_stmt->setString( i, std::string( reinterpret_cast< char* >( m_statementData[ i ].binary.data() ) ) );
        break;
      case TYPE_BINARY:
      {
        std::stringstream* is = new std::stringstream;

        for( auto entry : m_statementData[ i ].binary )
          is->rdbuf()->sputc( static_cast< char > ( entry ) );

        m_stmt->setBlob( i, is );
      }
        break;
      case TYPE_NULL:
        m_stmt->setNull( i, 0 );
        break;
    }
  }
}

//- Bind to buffer
void Core::Db::PreparedStatement::setBool( uint8_t index, const bool value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].data.boolean = value;
  m_statementData[ index ].type = TYPE_BOOL;
}

void Core::Db::PreparedStatement::setUInt( uint8_t index, uint32_t value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].data.ui32 = value;
  m_statementData[ index ].type = TYPE_UI;
}

void Core::Db::PreparedStatement::setUInt64( uint8_t index, uint64_t value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].data.ui64 = value;
  m_statementData[ index ].type = TYPE_UI64;
}

void Core::Db::PreparedStatement::setInt( uint8_t index, int32_t value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].data.i32 = value;
  m_statementData[ index ].type = TYPE_I;
}

void Core::Db::PreparedStatement::setInt64( uint8_t index, int64_t value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].data.i64 = value;
  m_statementData[ index ].type = TYPE_I64;
}

void Core::Db::PreparedStatement::setDouble( uint8_t index, double value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].data.d = value;
  m_statementData[ index ].type = TYPE_DOUBLE;
}

void Core::Db::PreparedStatement::setString( uint8_t index, const std::string& value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].binary.resize( value.length() + 1 );
  memcpy( m_statementData[ index ].binary.data(), value.c_str(), value.length() + 1 );
  m_statementData[ index ].type = TYPE_STRING;
}

void Core::Db::PreparedStatement::setBinary( uint8_t index, const std::vector< uint8_t >& value )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].binary = value;
  m_statementData[ index ].type = TYPE_BINARY;
}

void Core::Db::PreparedStatement::setNull( uint8_t index )
{
  if( index >= m_statementData.size() )
    m_statementData.resize( index + 1 );

  m_statementData[ index ].type = TYPE_NULL;
}

uint32_t Core::Db::PreparedStatement::getIndex() const
{
  return m_index;
}

void Core::Db::PreparedStatement::setMysqlPS( boost::shared_ptr< Mysql::PreparedStatement > pStmt )
{
  m_stmt = pStmt;
}