source/dbms.mysql/OutputBind.cpp

   1 // ANNA - Anna is Not Nothingness Anymore                                                         //
   2 //                                                                                                //
   3 // (c) Copyright 2005-2015 Eduardo Ramos Testillano & Francisco Ruiz Rayo                         //
   4 //                                                                                                //
   5 // See project site at http://redmine.teslayout.com/projects/anna-suite                           //
   6 // See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE //
   7
   8
   9 #include <mysql/mysql.h>
  10
  11 #include <anna/config/defines.hpp>
  12 #include <anna/core/tracing/Logger.hpp>
  13 #include <anna/core/DataBlock.hpp>
  14
  15 #include <anna/dbms/Float.hpp>
  16 #include <anna/dbms/ShortBlock.hpp>
  17 #include <anna/dbms/LongBlock.hpp>
  18 #include <anna/dbms/String.hpp>
  19 #include <anna/dbms/Date.hpp>
  20 #include <anna/dbms/TimeStamp.hpp>
  21
  22 #include <anna/dbms.mysql/mysql.hpp>
  23
  24 using namespace anna;
  25 using namespace std;
  26
  27 OutputBind::OutputBind(const char* name, dbms::Data& data) :
  28   dbms::OutputBind(name, data),
  29   BaseBind(data) {
  30   a_blob = (data.getType() == Data::Type::LongBlock) ? new Blob : NULL;
  31 }
  32
  33 OutputBind::~OutputBind() {
  34   delete a_blob;
  35 }
  36
  37 /*
  38  * El dato de LONG BLOB se recoge según cuenta:
  39  * http://dev.mysql.com/doc/refman/4.1/en/mysql-stmt-fetch.html.
  40  */
  41 void OutputBind::prepare(anna::dbms::Statement* dbmsStmt, anna::dbms::Connection*, const int pos)
  42 throw(RuntimeException) {
  43   st_mysql_bind* bind = static_cast <dbms::mysql::Statement*>(dbmsStmt)->getBindResults() + pos;
  44   Data& data = anna::dbms::Bind::getData();
  45   BaseBind::setupBind(*bind, data);
  46
  47   if(data.getType() == Data::Type::LongBlock) {
  48     dbms::mysql::Statement* myStmt = static_cast <dbms::mysql::Statement*>(dbmsStmt);
  49     a_blob->stmt = *myStmt;
  50     a_blob->binds = myStmt->getBindResults();
  51     a_blob->pos = pos;
  52     bind->buffer_type = MYSQL_TYPE_BLOB;
  53     bind->buffer = NULL;
  54     bind->buffer_length = 0;
  55     bind->length = &a_length;
  56   }
  57 }
  58
  59 /*
  60  * Transfiere la información del los MYSQL_BIND del API C de MySQL a las
  61  * estructuras dbms::Data de nuestro programa C++.
  62  */
  63 void OutputBind::decode() const
  64 throw(RuntimeException) {
  65   OutputBind* _this = const_cast <OutputBind*>(this);
  66   char* str;
  67   Data& data = _this->getData();
  68   data.setNull(a_nullIndicator == true);
  69
  70   switch(data.getType()) {
  71   case Data::Type::String:
  72     str = (char*) data.getBuffer();
  73
  74     if(data.isNull() == true)
  75       *str = 0;
  76     else
  77       dbms::String::strip(str);
  78
  79     break;
  80   case Data::Type::Float:
  81
  82     if(data.isNull() == true)
  83       static_cast <dbms::Float&>(data) = 0.0;
  84
  85     break;
  86   case Data::Type::Date:
  87   case Data::Type::TimeStamp:
  88     _this->decodeDate(data);
  89     break;
  90   case Data::Type::Integer:
  91     throw RuntimeException("anna::dbms::mysql::OutputBind::decode not implemented for Data::Type::Integer", ANNA_FILE_LOCATION);
  92     break;
  93   case Data::Type::ShortBlock:
  94     throw RuntimeException("anna::dbms::mysql::OutputBind::decode not implemented for Data::Type::ShortBlock", ANNA_FILE_LOCATION);
  95     break;
  96   case Data::Type::LongBlock:
  97
  98     try {
  99       _this->decodeLongBlob(data);
 100     } catch(dbms::DatabaseException& edb) {
 101       throw RuntimeException(edb);
 102     }
 103
 104     break;
 105   }
 106 }
 107
 108 void OutputBind::do_write(const dbms::LongBlock&) const
 109 throw(RuntimeException, dbms::DatabaseException) {
 110 }
 111
 112 /*
 113  * El m�todo BaseBind::setupBind asocia� el contenido de la variable
 114  * a_time al buffer de salida de la sentencia SQL, as� que el contenido
 115  * de la columna est� contenido ah�. S�lo tendremos que copiar dichos
 116  * contenidos en la variable C++ de nuestro entorno.
 117  */
 118 void OutputBind::decodeDate(dbms::Data& data)
 119 throw() {
 120   if(data.isNull() == true)
 121     return;
 122
 123   Date& date = static_cast <Date&>(data);
 124   date.setYear(a_time->year);
 125   date.setMonth(a_time->month);
 126   date.setDay(a_time->day);
 127   date.setHour(a_time->hour);
 128   date.setMinute(a_time->minute);
 129   date.setSecond(a_time->second);
 130
 131   if(data.getType() == Data::Type::TimeStamp)
 132     static_cast <dbms::TimeStamp&>(data).setFractionalSecond(a_time->second_part);
 133 }
 134
 135 /*
 136  * Según http://dev.mysql.com/doc/refman/4.1/en/mysql-stmt-fetch.html
 137  *
 138  * Recoge el contenido total de BLOB en partes que va componiendo sobre el
 139  * DataBlock final asociado al anna::dbms::LongBlock sobre el que trabaja el
 140  * programador final.
 141  */
 142 void OutputBind::decodeLongBlob(dbms::Data& data) const
 143 throw(RuntimeException, dbms::DatabaseException) {
 144   const int bufferSize = a_blob->buffer.getMaxSize();
 145   st_mysql_bind& bind = a_blob->binds [a_blob->pos];
 146   DataBlock& target = static_cast <dbms::LongBlock&>(data).getValue();
 147   const int maxloop = *bind.length / bufferSize;
 148   const int remainder = *bind.length % bufferSize;
 149   int offset = 0;
 150   target.clear();
 151   bind.buffer = (void*) a_blob->buffer.getData();
 152
 153   for(int iloop = 0; iloop < maxloop; iloop ++) {
 154     bind.buffer_length = bufferSize;
 155     anna_dbms_mysql_check(
 156       mysql_stmt_fetch_column(a_blob->stmt, a_blob->binds, a_blob->pos, offset), a_blob->stmt
 157     );
 158     target += DataBlock((const char*) bind.buffer, bind.buffer_length, false);
 159     offset += bufferSize;
 160   }
 161
 162   if(remainder) {
 163     bind.buffer_length = remainder;
 164     anna_dbms_mysql_check(
 165       mysql_stmt_fetch_column(a_blob->stmt, a_blob->binds, a_blob->pos, offset), a_blob->stmt
 166     );
 167     target += DataBlock((const char*) bind.buffer, remainder, false);
 168   }
 169 }
 170
 171 OutputBind::Blob::Blob() :
 172   buffer(true) {
 173   buffer.allocate(64 * 1024);
 174 }