You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
koffice/kexi/kexidb/drivers/pqxx/pqxxcursor.cpp

340 lines
9.8 KiB

/* This file is part of the KDE project
Copyright (C) 2003 Adam Pigg <adam@piggz.co.uk>
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Library General Public License for more details.
You should have received a copy of the GNU Library General Public License
along with this program; see the file COPYING. If not, write to
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#include "pqxxcursor.h"
#include "pqxxconnection.h"
#include "pqxxconnection_p.h"
#include <kexidb/error.h>
#include <kexidb/global.h>
#include <klocale.h>
#include <kdebug.h>
#include <cstdlib>
using namespace KexiDB;
unsigned int pqxxSqlCursor_trans_num=0; //!< debug helper
static TQByteArray pgsqlByteaToByteArray(const pqxx::result::field& r)
{
return KexiDB::pgsqlByteaToByteArray(r.c_str(), r.size());
}
//==================================================================================
//Constructor based on query statement
pqxxSqlCursor::pqxxSqlCursor(KexiDB::Connection* conn, const TQString& statement, uint options):
Cursor(conn,statement, options)
{
// KexiDBDrvDbg << "PTQXXSQLCURSOR: constructor for query statement" << endl;
my_conn = static_cast<pqxxSqlConnection*>(conn)->d->pqxxsql;
m_options = Buffered;
m_res = 0;
// m_tran = 0;
m_implicityStarted = false;
}
//==================================================================================
//Constructor base on query object
pqxxSqlCursor::pqxxSqlCursor(Connection* conn, QuerySchema& query, uint options )
: Cursor( conn, query, options )
{
// KexiDBDrvDbg << "PTQXXSQLCURSOR: constructor for query schema" << endl;
my_conn = static_cast<pqxxSqlConnection*>(conn)->d->pqxxsql;
m_options = Buffered;
m_res = 0;
// m_tran = 0;
m_implicityStarted = false;
}
//==================================================================================
//Destructor
pqxxSqlCursor::~pqxxSqlCursor()
{
close();
}
//==================================================================================
//Create a cursor result set
bool pqxxSqlCursor::drv_open()
{
// KexiDBDrvDbg << "pqxxSqlCursor::drv_open:" << m_sql << endl;
if (!my_conn->is_open())
{
//! @todo this check should be moved to Connection! when drv_prepareQuery() arrive
//should never happen, but who knows
setError(ERR_NO_CONNECTION,i18n("No connection for cursor open operation specified"));
return false;
}
TQCString cur_name;
//Set up a transaction
try
{
//m_tran = new pqxx::work(*my_conn, "cursor_open");
cur_name.sprintf("cursor_transaction%d", pqxxSqlCursor_trans_num++);
// m_tran = new pqxx::nontransaction(*my_conn, (const char*)cur_name);
if (!((pqxxSqlConnection*)connection())->m_trans) {
// my_conn->drv_beginTransaction();
// if (implicityStarted)
(void)new pqxxTransactionData((pqxxSqlConnection*)connection(), true);
m_implicityStarted = true;
}
m_res = new pqxx::result(((pqxxSqlConnection*)connection())->m_trans->data->exec(std::string(m_sql.utf8())));
((pqxxSqlConnection*)connection())
->drv_commitTransaction(((pqxxSqlConnection*)connection())->m_trans);
// my_conn->m_trans->commit();
// KexiDBDrvDbg << "pqxxSqlCursor::drv_open: trans. committed: " << cur_name <<endl;
//We should now be placed before the first row, if any
m_fieldCount = m_res->columns() - (m_containsROWIDInfo ? 1 : 0);
//js m_opened=true;
m_afterLast=false;
m_records_in_buf = m_res->size();
m_buffering_completed = true;
return true;
}
catch (const std::exception &e)
{
setError(ERR_DB_SPECIFIC, TQString::fromUtf8( e.what()) );
KexiDBDrvWarn << "pqxxSqlCursor::drv_open:exception - " << TQString::fromUtf8( e.what() ) << endl;
}
catch(...)
{
setError();
}
// delete m_tran;
// m_tran = 0;
if (m_implicityStarted) {
delete ((pqxxSqlConnection*)connection())->m_trans;
m_implicityStarted = false;
}
// KexiDBDrvDbg << "pqxxSqlCursor::drv_open: trans. rolled back! - " << cur_name <<endl;
return false;
}
//==================================================================================
//Delete objects
bool pqxxSqlCursor::drv_close()
{
//js m_opened=false;
delete m_res;
m_res = 0;
// if (m_implicityStarted) {
// delete m_tran;
// m_tran = 0;
// m_implicityStarted = false;
// }
return true;
}
//==================================================================================
//Gets the next record...does not need to do much, just return fetchend if at end of result set
void pqxxSqlCursor::drv_getNextRecord()
{
// KexiDBDrvDbg << "pqxxSqlCursor::drv_getNextRecord, size is " <<m_res->size() << " Current Position is " << (long)at() << endl;
if(at() < m_res->size() && at() >=0)
{
m_result = FetchOK;
}
else if (at() >= m_res->size())
{
m_result = FetchEnd;
}
else
{
m_result = FetchError;
}
}
//==================================================================================
//Check the current position is within boundaries
void pqxxSqlCursor::drv_getPrevRecord()
{
// KexiDBDrvDbg << "pqxxSqlCursor::drv_getPrevRecord" << endl;
if(at() < m_res->size() && at() >=0)
{
m_result = FetchOK;
}
else if (at() >= m_res->size())
{
m_result = FetchEnd;
}
else
{
m_result = FetchError;
}
}
//==================================================================================
//Return the value for a given column for the current record
TQVariant pqxxSqlCursor::value(uint pos)
{
if (pos < m_fieldCount)
return pValue(pos);
else
return TQVariant();
}
//==================================================================================
//Return the value for a given column for the current record - Private const version
TQVariant pqxxSqlCursor::pValue(uint pos)const
{
if (m_res->size() <= 0)
{
KexiDBDrvWarn << "pqxxSqlCursor::value - ERROR: result size not greater than 0" << endl;
return TQVariant();
}
if (pos>=(m_fieldCount+(m_containsROWIDInfo ? 1 : 0)))
{
// KexiDBDrvWarn << "pqxxSqlCursor::value - ERROR: requested position is greater than the number of fields" << endl;
return TQVariant();
}
KexiDB::Field *f = (m_fieldsExpanded && pos<TQMIN(m_fieldsExpanded->count(), m_fieldCount))
? m_fieldsExpanded->at(pos)->field : 0;
// KexiDBDrvDbg << "pqxxSqlCursor::value(" << pos << ")" << endl;
//from most to least frequently used types:
if (f) //We probably have a schema type query so can use kexi to determin the row type
{
if ((f->isIntegerType()) || (/*ROWID*/!f && m_containsROWIDInfo && pos==m_fieldCount))
{
return (*m_res)[at()][pos].as(int());
}
else if (f->isTextType())
{
return TQString::fromUtf8((*m_res)[at()][pos].c_str()); //utf8?
}
else if (f->isFPNumericType())
{
return (*m_res)[at()][pos].as(double());
}
else if (f->typeGroup() == Field::BLOBGroup)
{
// pqxx::result::field r = (*m_res)[at()][pos];
// kdDebug() << r.name() << ", " << r.c_str() << ", " << r.type() << ", " << r.size() << endl;
return ::pgsqlByteaToByteArray((*m_res)[at()][pos]);
}
}
else // We probably have a raw type query so use pqxx to determin the column type
{
return pgsqlCStrToVariant((*m_res)[at()][pos]);
}
return TQString::fromUtf8((*m_res)[at()][pos].c_str(), (*m_res)[at()][pos].size()); //utf8?
}
//==================================================================================
//Return the current record as a char**
//who'd have thought we'd be using char** in this day and age :o)
const char** pqxxSqlCursor::rowData() const
{
// KexiDBDrvDbg << "pqxxSqlCursor::recordData" << endl;
const char** row;
row = (const char**)malloc(m_res->columns()+1);
row[m_res->columns()] = NULL;
if (at() >= 0 && at() < m_res->size())
{
for(int i = 0; i < (int)m_res->columns(); i++)
{
row[i] = (char*)malloc(strlen((*m_res)[at()][i].c_str())+1);
strcpy((char*)(*m_res)[at()][i].c_str(), row[i]);
// KexiDBDrvDbg << row[i] << endl;
}
}
else
{
KexiDBDrvWarn << "pqxxSqlCursor::recordData: m_at is invalid" << endl;
}
return row;
}
//==================================================================================
//Store the current record in [data]
void pqxxSqlCursor::storeCurrentRow(RowData &data) const
{
// KexiDBDrvDbg << "pqxxSqlCursor::storeCurrentRow: POSITION IS " << (long)m_at<< endl;
if (m_res->size()<=0)
return;
const uint realCount = m_fieldCount + (m_containsROWIDInfo ? 1 : 0);
data.resize(realCount);
for( uint i=0; i<realCount; i++)
{
data[i] = pValue(i);
}
}
//==================================================================================
//
void pqxxSqlCursor::drv_clearServerResult()
{
//! @todo pqxxSqlCursor: stuff with server results
}
//==================================================================================
//Add the current record to the internal buffer
//Implementation required but no need in this driver
//Result set is a buffer so do not need another
void pqxxSqlCursor::drv_appendCurrentRecordToBuffer()
{
}
//==================================================================================
//Move internal pointer to internal buffer +1
//Implementation required but no need in this driver
void pqxxSqlCursor::drv_bufferMovePointerNext()
{
}
//==================================================================================
//Move internal pointer to internal buffer -1
//Implementation required but no need in this driver
void pqxxSqlCursor::drv_bufferMovePointerPrev()
{
}
//==================================================================================
//Move internal pointer to internal buffer to N
//Implementation required but no need in this driver
void pqxxSqlCursor::drv_bufferMovePointerTo(TQ_LLONG to)
{
Q_UNUSED(to);
}