void Keeper::open( const QMap<QString,QString> &opt ) {...db.exec( "LISTEN reload_schema" );connect( drv, SIGNAL( sqlNotify( const QString & ) ), this, SLOT( sqlNotify(const QString &) ) );...}void Keeper::sqlNotify( const QString &str ){//qDebug( "We'v GOT NOTIFY: %s", qPrintable(str) ); if ( str.compare( "reload_schema" ) == 0 ) { reloadSchema(); }}
QSqlQuery query( keeper->database() );query.exec( "NOTIFY reload_schema" );
bool Blob::writeBlob( const QByteArray &ba ){ int lo = get( "lo" ).toInt(); QPSQLDriver *drv = (QPSQLDriver*)keeper->database().driver(); if ( !drv ) return false; if ( lo ) { drv->loUnlink( lo ); set( "lo", 0 ); lo = 0; } lo = drv->loCreate(); if ( !lo ) return false; if ( drv->loWrite( lo, ba ) ) { set( "lo", lo ); set( "size", ba.size() ); return true; } return false;}QByteArray Blob::readBlob() const{ QByteArray ba; int lo = get( "lo" ).toInt(); if ( !lo ) return ba; QPSQLDriver *drv = (QPSQLDriver*)keeper->database().driver(); if ( !drv ) return ba; ba = drv->loRead( lo ); return ba;}
#include "qsql_psql.h"#include <QCoreApplication>#include <QVariant>#include <QDateTime>#include <QRegExp>#include <QSqlError>#include <QSqlField>#include <QSqlIndex>#include <QSqlRecord>#include <QSqlQuery>#include <QStringList>//#include <libpq-fe.h>#include <server/postgres.h>#include <libpq-fe.h>#include <libpq/libpq-fs.h>#if defined(Q_CC_MSVC)#pragma warning(disable: 4273) // '_errno' : inconsistent dll linkage. dllexport assumed#endif#include <catalog/pg_type.h>#if defined(Q_CC_MSVC)#pragma warning(default: 4273)#endif#include <stdlib.h>#include <math.h>Q_DECLARE_METATYPE(QPSQLconn*);Q_DECLARE_METATYPE(QPSQLresult*);// workaround for postgres defining their OIDs in a private header file#define QBOOLOID 16#define QINT8OID 20#define QINT2OID 21#define QINT4OID 23#define QNUMERICOID 1700#define QFLOAT4OID 700#define QFLOAT8OID 701#define QABSTIMEOID 702#define QRELTIMEOID 703#define QDATEOID 1082#define QTIMEOID 1083#define QTIMETZOID 1266#define QTIMESTAMPOID 1114#define QTIMESTAMPTZOID 1184#define QOIDOID 2278#define QBYTEAOID 17#define QREGPROCOID 24#define QXIDOID 28#define QCIDOID 29class QPSQLDriverPrivate{ public: QPSQLDriverPrivate(): connection(0), isUtf8(false), pro(QPSQLDriver::Version6) {} QPSQLconn *connection; bool isUtf8; QPSQLDriver::Protocol pro; void appendTables(QStringList &tl, QSqlQuery &t, QChar type);};void QPSQLDriverPrivate::appendTables(QStringList &tl, QSqlQuery &t, QChar type){ QString query; if (pro >= QPSQLDriver::Version73) { query = QString::fromLatin1("select pg_class.relname, pg_namespace.nspname from pg_class " "left join pg_namespace on (pg_class.relnamespace = pg_namespace.oid) " "where (pg_class.relkind = '%1') and (pg_class.relname !~ '^Inv') " "and (pg_class.relname !~ '^pg_') " "and (pg_namespace.nspname != 'information_schema') ").arg(type); } else { query = QString::fromLatin1("select relname, null from pg_class where (relkind = 'r') " "and (relname !~ '^Inv') " "and (relname !~ '^pg_') "); } t.exec(query); while (t.next()) { QString schema = t.value(1).toString(); if (schema.isEmpty() || schema == QLatin1String("public")) tl.append(t.value(0).toString()); else tl.append(t.value(0).toString().prepend(QLatin1Char('.')).prepend(schema)); }}class QPSQLResultPrivate{ public: QPSQLResultPrivate(QPSQLResult *qq): q(qq), driver(0), result(0), currentSize(-1) {} QPSQLResult *q; const QPSQLDriverPrivate *driver; QPSQLresult *result; int currentSize; bool processResults();};static QSqlError qMakeError(const QString& err, QSqlError::ErrorType type, const QPSQLDriverPrivate *p){ const char *s = PQerrorMessage(p->connection); QString msg = p->isUtf8 ? QString::fromUtf8(s) : QString::fromLocal8Bit(s); return QSqlError(QLatin1String("QPSQL: ") + err, msg, type);}bool QPSQLResultPrivate::processResults(){ if (!result) return false; int status = PQresultStatus(result); if (status == PGRES_TUPLES_OK) { q->setSelect(true); q->setActive(true); currentSize = PQntuples(result); return true; } else if (status == PGRES_COMMAND_OK) { q->setSelect(false); q->setActive(true); currentSize = -1; return true; } q->setLastError(qMakeError(QCoreApplication::translate("QPSQLResult", "Unable to create query"), QSqlError::StatementError, driver)); return false;}static QVariant::Type qDecodePSQLType(int t){ QVariant::Type type = QVariant::Invalid; switch (t) { case QBOOLOID: type = QVariant::Bool; break; case QINT8OID: type = QVariant::LongLong; break; case QINT2OID: case QINT4OID: case QOIDOID: case QREGPROCOID: case QXIDOID: case QCIDOID: type = QVariant::Int; break; case QNUMERICOID: case QFLOAT4OID: case QFLOAT8OID: type = QVariant::Double; break; case QABSTIMEOID: case QRELTIMEOID: case QDATEOID: type = QVariant::Date; break; case QTIMEOID: case QTIMETZOID: type = QVariant::Time; break; case QTIMESTAMPOID: case QTIMESTAMPTZOID: type = QVariant::DateTime; break; case POINTOID: type = QVariant::Point; break; case BOXOID: type = QVariant::Rect; break; case POLYGONOID: type = QVariant::Polygon; break; case QBYTEAOID: type = QVariant::ByteArray; break; default: type = QVariant::String; break; } return type;}QPSQLResult::QPSQLResult(const QPSQLDriver* db, const QPSQLDriverPrivate* p) : QSqlResult(db){ d = new QPSQLResultPrivate(this); d->driver = p;}QPSQLResult::~QPSQLResult(){ cleanup(); delete d;}QVariant QPSQLResult::handle() const{ return qVariantFromValue(d->result);}QPSQLDriver* QPSQLResult::pgDriver(){ QPSQLDriver* dr = (QPSQLDriver*)this->driver(); return dr;}void QPSQLResult::cleanup(){ if (d->result) PQclear(d->result); d->result = 0; setAt(QSql::BeforeFirstRow); d->currentSize = -1; setActive(false);}bool QPSQLResult::fetch(int i){ if (!isActive()) return false; if (i < 0) return false; if (i >= d->currentSize) return false; if (at() == i) return true; setAt(i); return true;}bool QPSQLResult::fetchFirst(){ return fetch(0);}bool QPSQLResult::fetchLast(){ return fetch(PQntuples(d->result) - 1);}QVariant QPSQLResult::data(int i){ if (i >= PQnfields(d->result)) { qWarning("QPSQLResult::data: column %d out of range", i); return QVariant(); } int ptype = PQftype(d->result, i); QVariant::Type type = qDecodePSQLType(ptype); const char *val = PQgetvalue(d->result, at(), i); if (PQgetisnull(d->result, at(), i)) return QVariant(type); switch (type) { case QVariant::Bool: return QVariant((bool)(val[0] == 't')); case QVariant::String: return d->driver->isUtf8 ? QString::fromUtf8(val) : QString::fromAscii(val); case QVariant::LongLong: if (val[0] == '-') return QString::fromLatin1(val).toLongLong(); else return QString::fromLatin1(val).toULongLong(); case QVariant::Int: return atoi(val); case QVariant::Double: if (ptype == QNUMERICOID) return QString::fromAscii(val); return strtod(val, 0); case QVariant::Date: if (val[0] == '\0') { return QVariant(QDate()); } else { return QVariant(QDate::fromString(QString::fromLatin1(val), Qt::ISODate)); } case QVariant::Time: { const QString str = QString::fromLatin1(val); if (str.isEmpty()) return QVariant(QTime()); if (str.at(str.length() - 3) == QLatin1Char('+')) // strip the timezone return QVariant(QTime::fromString(str.left(str.length() - 3), Qt::ISODate)); return QVariant(QTime::fromString(str, Qt::ISODate)); } case QVariant::DateTime: { QString dtval = QString::fromLatin1(val); if (dtval.length() < 10) return QVariant(QDateTime()); // remove the timezone if (dtval.at(dtval.length() - 3) == QLatin1Char('+')) dtval.chop(3); // milliseconds are sometimes returned with 2 digits only if (dtval.at(dtval.length() - 3).isPunct()) dtval += QLatin1Char('0'); if (dtval.isEmpty()) return QVariant(QDateTime()); else return QVariant(QDateTime::fromString(dtval, Qt::ISODate)); } case QVariant::ByteArray: { size_t len; unsigned char *data = PQunescapeBytea((unsigned char*)val, &len); QByteArray ba((const char*)data, len); PQfreemem(data); return QVariant(ba); } default: case QVariant::Invalid: qWarning("QPSQLResult::data: unknown data type"); } return QVariant();}bool QPSQLResult::isNull(int field){ PQgetvalue(d->result, at(), field); return PQgetisnull(d->result, at(), field);}bool QPSQLResult::reset (const QString& query){ cleanup(); if (!driver()) return false; if (!driver()->isOpen() || driver()->isOpenError()) return false; d->result = PQexec(d->driver->connection, d->driver->isUtf8 ? query.toUtf8().constData() : query.toLocal8Bit().constData()); pgDriver()->checkNotify(); return d->processResults();}int QPSQLResult::size(){ return d->currentSize;}int QPSQLResult::numRowsAffected(){ return QString::fromLatin1(PQcmdTuples(d->result)).toInt();}QVariant QPSQLResult::lastInsertId() const{ if (isActive()) { Oid id = PQoidValue(d->result); if (id != InvalidOid) return QVariant(id); } return QVariant();}QSqlRecord QPSQLResult::record() const{ QSqlRecord info; if (!isActive() || !isSelect()) return info; int count = PQnfields(d->result); for (int i = 0; i < count; ++i) { QSqlField f; if (d->driver->isUtf8) f.setName(QString::fromUtf8(PQfname(d->result, i))); else f.setName(QString::fromLocal8Bit(PQfname(d->result, i))); f.setType(qDecodePSQLType(PQftype(d->result, i))); int len = PQfsize(d->result, i); int precision = PQfmod(d->result, i); // swap length and precision if length == -1 if (len == -1 && precision > -1) { len = precision - 4; precision = -1; } f.setLength(len); f.setPrecision(precision); f.setSqlType(PQftype(d->result, i)); info.append(f); } return info;}///////////////////////////////////////////////////////////////////static bool setEncodingUtf8(QPSQLconn* connection){ QPSQLresult* result = PQexec(connection, "SET CLIENT_ENCODING TO 'UNICODE'"); int status = PQresultStatus(result); PQclear(result); return status == PGRES_COMMAND_OK;}static void setDatestyle(QPSQLconn* connection){ QPSQLresult* result = PQexec(connection, "SET DATESTYLE TO 'ISO'"); int status = PQresultStatus(result); if (status != PGRES_COMMAND_OK) qWarning("%s", PQerrorMessage(connection)); PQclear(result);}static QPSQLDriver::Protocol getPSQLVersion(QPSQLconn* connection){ QPSQLresult* result = PQexec(connection, "select version()"); int status = PQresultStatus(result); if (status == PGRES_COMMAND_OK || status == PGRES_TUPLES_OK) { QString val = QString::fromAscii(PQgetvalue(result, 0, 0)); PQclear(result); QRegExp rx(QLatin1String("(\\d+)\\.(\\d+)")); rx.setMinimal(true); // enforce non-greedy RegExp if (rx.indexIn(val) != -1) { int vMaj = rx.cap(1).toInt(); int vMin = rx.cap(2).toInt(); if (vMaj < 6) { qWarning("This version of PostgreSQL is not supported and may not work."); return QPSQLDriver::Version6; } if (vMaj == 6) { return QPSQLDriver::Version6; } else if (vMaj == 7) { if (vMin < 1) return QPSQLDriver::Version7; else if (vMin < 3) return QPSQLDriver::Version71; } return QPSQLDriver::Version73; } } else { qWarning("This version of PostgreSQL is not supported and may not work."); } return QPSQLDriver::Version6;}QPSQLDriver::QPSQLDriver(QObject *parent) : QSqlDriver(parent){ init();}QPSQLDriver::QPSQLDriver(QPSQLconn * conn, QObject * parent) : QSqlDriver(parent){ init(); d->connection = conn; if (conn) { d->pro = getPSQLVersion(d->connection); setOpen(true); setOpenError(false); }}void QPSQLDriver::init(){ d = new QPSQLDriverPrivate();}QPSQLDriver::~QPSQLDriver(){ if (d->connection) PQfinish(d->connection); delete d;}QVariant QPSQLDriver::handle() const{ return qVariantFromValue(d->connection);}bool QPSQLDriver::hasFeature(DriverFeature f) const{ switch (f) { case Transactions: case QuerySize: case LastInsertId: return true; case BatchOperations: case PreparedQueries: case NamedPlaceholders: case PositionalPlaceholders: return false; case BLOB: return d->pro >= QPSQLDriver::Version71; case Unicode: return d->isUtf8; } return false;}/* Quote a string for inclusion into the connection string \ -> \\ ' -> \' surround string by single quotes */static QString qQuote(QString s){ s.replace(QLatin1Char('\\'), QLatin1String("\\\\")); s.replace(QLatin1Char('\''), QLatin1String("\\'")); s.append(QLatin1Char('\'')).prepend(QLatin1Char('\'')); return s;}bool QPSQLDriver::open(const QString & db, const QString & user, const QString & password, const QString & host, int port, const QString& connOpts){ if (isOpen()) close(); QString connectString; if (!host.isEmpty()) connectString.append(QLatin1String("host=")).append(qQuote(host)); if (!db.isEmpty()) connectString.append(QLatin1String(" dbname=")).append(qQuote(db)); if (!user.isEmpty()) connectString.append(QLatin1String(" user=")).append(qQuote(user)); if (!password.isEmpty()) connectString.append(QLatin1String(" password=")).append(qQuote(password)); if (port != -1) connectString.append(QLatin1String(" port=")).append(qQuote(QString::number(port))); // add any connect options - the server will handle error detection if (!connOpts.isEmpty()) { QString opt = connOpts; opt.replace(QLatin1Char(';'), QLatin1Char(' '), Qt::CaseInsensitive); connectString.append(QLatin1Char(' ')).append(opt); } d->connection = PQconnectdb(connectString.toLocal8Bit().constData()); if (PQstatus(d->connection) == CONNECTION_BAD) { setLastError(qMakeError(tr("Unable to connect"), QSqlError::ConnectionError, d)); setOpenError(true); return false; } d->pro = getPSQLVersion(d->connection); d->isUtf8 = setEncodingUtf8(d->connection); setDatestyle(d->connection); setOpen(true); setOpenError(false); return true;}void QPSQLDriver::checkNotify(){ PQconsumeInput( d->connection ); pgNotify* noti = PQnotifies( d->connection ); while ( noti != NULL ) { emit sqlNotify( QString(noti->relname) ); PQfreemem( noti ); noti = PQnotifies( d->connection ); } PQfreemem( noti );}void QPSQLDriver::close(){ if ( isOpen() ) { if ( d->connection ) { PQfinish( d->connection ); } d->connection = 0; setOpen(false); setOpenError(false); }}QSqlResult *QPSQLDriver::createResult() const{ return new QPSQLResult(this, d);}bool QPSQLDriver::beginTransaction(){ if (!isOpen()) { qWarning("QPSQLDriver::beginTransaction: Database not open"); return false; } QPSQLresult* res = PQexec(d->connection, "BEGIN"); if (!res || PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); setLastError(qMakeError(tr("Could not begin transaction"), QSqlError::TransactionError, d)); return false; } PQclear(res); return true;}bool QPSQLDriver::commitTransaction(){ if (!isOpen()) { qWarning("QPSQLDriver::commitTransaction: Database not open"); return false; } QPSQLresult* res = PQexec(d->connection, "COMMIT"); if (!res || PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); setLastError(qMakeError(tr("Could not commit transaction"), QSqlError::TransactionError, d)); return false; } PQclear(res); return true;}bool QPSQLDriver::rollbackTransaction(){ if (!isOpen()) { qWarning("QPSQLDriver::rollbackTransaction: Database not open"); return false; } QPSQLresult* res = PQexec(d->connection, "ROLLBACK"); if (!res || PQresultStatus(res) != PGRES_COMMAND_OK) { setLastError(qMakeError(tr("Could not rollback transaction"), QSqlError::TransactionError, d)); PQclear(res); return false; } PQclear(res); return true;}QStringList QPSQLDriver::tables(QSql::TableType type) const{ QStringList tl; if (!isOpen()) return tl; QSqlQuery t(createResult()); t.setForwardOnly(true); if (type & QSql::Tables) d->appendTables(tl, t, QLatin1Char('r')); if (type & QSql::Views) d->appendTables(tl, t, QLatin1Char('v')); if (type & QSql::SystemTables) { t.exec(QLatin1String("select relname from pg_class where (relkind = 'r') " "and (relname like 'pg_%') ")); while (t.next()) tl.append(t.value(0).toString()); } return tl;}static void qSplitTableName(QString &tablename, QString &schema){ int dot = tablename.indexOf(QLatin1Char('.')); if (dot == -1) return; schema = tablename.left(dot); tablename = tablename.mid(dot + 1);}QSqlIndex QPSQLDriver::primaryIndex(const QString& tablename) const{ QSqlIndex idx(tablename); if (!isOpen()) return idx; QSqlQuery i(createResult()); QString stmt; QString tbl = tablename; QString schema; qSplitTableName(tbl, schema); switch(d->pro) { case QPSQLDriver::Version6: stmt = QLatin1String("select pg_att1.attname, int(pg_att1.atttypid), pg_cl.relname " "from pg_attribute pg_att1, pg_attribute pg_att2, pg_class pg_cl, pg_index pg_ind " "where lower(pg_cl.relname) = '%1_pkey' " "and pg_cl.oid = pg_ind.indexrelid " "and pg_att2.attrelid = pg_ind.indexrelid " "and pg_att1.attrelid = pg_ind.indrelid " "and pg_att1.attnum = pg_ind.indkey[pg_att2.attnum-1] " "order by pg_att2.attnum"); break; case QPSQLDriver::Version7: case QPSQLDriver::Version71: stmt = QLatin1String("select pg_att1.attname, pg_att1.atttypid::int, pg_cl.relname " "from pg_attribute pg_att1, pg_attribute pg_att2, pg_class pg_cl, pg_index pg_ind " "where lower(pg_cl.relname) = '%1_pkey' " "and pg_cl.oid = pg_ind.indexrelid " "and pg_att2.attrelid = pg_ind.indexrelid " "and pg_att1.attrelid = pg_ind.indrelid " "and pg_att1.attnum = pg_ind.indkey[pg_att2.attnum-1] " "order by pg_att2.attnum"); break; case QPSQLDriver::Version73: stmt = QLatin1String("SELECT pg_attribute.attname, pg_attribute.atttypid::int, " "pg_class.relname " "FROM pg_attribute, pg_class " "WHERE %1 pg_class.oid = " "(SELECT indexrelid FROM pg_index WHERE indisprimary = true AND indrelid = " " (SELECT oid FROM pg_class WHERE lower(relname) = '%2')) " "AND pg_attribute.attrelid = pg_class.oid " "AND pg_attribute.attisdropped = false " "ORDER BY pg_attribute.attnum"); if (schema.isEmpty()) stmt = stmt.arg(QLatin1String("pg_table_is_visible(pg_class.oid) AND")); else stmt = stmt.arg(QString::fromLatin1("pg_class.relnamespace = (select oid from " "pg_namespace where pg_namespace.nspname = '%1') AND ").arg(schema.toLower())); break; } i.exec(stmt.arg(tbl.toLower())); while (i.isActive() && i.next()) { QSqlField f(i.value(0).toString(), qDecodePSQLType(i.value(1).toInt())); idx.append(f); idx.setName(i.value(2).toString()); } return idx;}QSqlRecord QPSQLDriver::record(const QString& tablename) const{ QSqlRecord info; if (!isOpen()) return info; QString tbl = tablename; QString schema; qSplitTableName(tbl, schema); QString stmt; switch(d->pro) { case QPSQLDriver::Version6: stmt = QLatin1String("select pg_attribute.attname, int(pg_attribute.atttypid), " "pg_attribute.attnotnull, pg_attribute.attlen, pg_attribute.atttypmod, " "int(pg_attribute.attrelid), pg_attribute.attnum " "from pg_class, pg_attribute " "where lower(pg_class.relname) = '%1' " "and pg_attribute.attnum > 0 " "and pg_attribute.attrelid = pg_class.oid "); break; case QPSQLDriver::Version7: stmt = QLatin1String("select pg_attribute.attname, pg_attribute.atttypid::int, " "pg_attribute.attnotnull, pg_attribute.attlen, pg_attribute.atttypmod, " "pg_attribute.attrelid::int, pg_attribute.attnum " "from pg_class, pg_attribute " "where lower(pg_class.relname) = '%1' " "and pg_attribute.attnum > 0 " "and pg_attribute.attrelid = pg_class.oid "); break; case QPSQLDriver::Version71: stmt = QLatin1String("select pg_attribute.attname, pg_attribute.atttypid::int, " "pg_attribute.attnotnull, pg_attribute.attlen, pg_attribute.atttypmod, " "pg_attrdef.adsrc " "from pg_class, pg_attribute " "left join pg_attrdef on (pg_attrdef.adrelid = " "pg_attribute.attrelid and pg_attrdef.adnum = pg_attribute.attnum) " "where lower(pg_class.relname) = '%1' " "and pg_attribute.attnum > 0 " "and pg_attribute.attrelid = pg_class.oid " "order by pg_attribute.attnum "); break; case QPSQLDriver::Version73: stmt = QLatin1String("select pg_attribute.attname, pg_attribute.atttypid::int, " "pg_attribute.attnotnull, pg_attribute.attlen, pg_attribute.atttypmod, " "pg_attrdef.adsrc " "from pg_class, pg_attribute " "left join pg_attrdef on (pg_attrdef.adrelid = " "pg_attribute.attrelid and pg_attrdef.adnum = pg_attribute.attnum) " "where %1 " "and lower(pg_class.relname) = '%2' " "and pg_attribute.attnum > 0 " "and pg_attribute.attrelid = pg_class.oid " "and pg_attribute.attisdropped = false " "order by pg_attribute.attnum "); if (schema.isEmpty()) stmt = stmt.arg(QLatin1String("pg_table_is_visible(pg_class.oid)")); else stmt = stmt.arg(QString::fromLatin1("pg_class.relnamespace = (select oid from " "pg_namespace where pg_namespace.nspname = '%1')").arg(schema.toLower())); break; } QSqlQuery query(createResult()); query.exec(stmt.arg(tbl.toLower())); if (d->pro >= QPSQLDriver::Version71) { while (query.next()) { int len = query.value(3).toInt(); int precision = query.value(4).toInt(); // swap length and precision if length == -1 if (len == -1 && precision > -1) { len = precision - 4; precision = -1; } QString defVal = query.value(5).toString(); if (!defVal.isEmpty() && defVal.at(0) == QLatin1Char('\'')) defVal = defVal.mid(1, defVal.length() - 2); QSqlField f(query.value(0).toString(), qDecodePSQLType(query.value(1).toInt())); f.setRequired(query.value(2).toBool()); f.setLength(len); f.setPrecision(precision); f.setDefaultValue(defVal); f.setSqlType(query.value(1).toInt()); info.append(f); } } else { // Postgres < 7.1 cannot handle outer joins while (query.next()) { QString defVal; QString stmt2 = QLatin1String("select pg_attrdef.adsrc from pg_attrdef where " "pg_attrdef.adrelid = %1 and pg_attrdef.adnum = %2 "); QSqlQuery query2(createResult()); query2.exec(stmt2.arg(query.value(5).toInt()).arg(query.value(6).toInt())); if (query2.isActive() && query2.next()) defVal = query2.value(0).toString(); if (!defVal.isEmpty() && defVal.at(0) == QLatin1Char('\'')) defVal = defVal.mid(1, defVal.length() - 2); int len = query.value(3).toInt(); int precision = query.value(4).toInt(); // swap length and precision if length == -1 if (len == -1 && precision > -1) { len = precision - 4; precision = -1; } QSqlField f(query.value(0).toString(), qDecodePSQLType(query.value(1).toInt())); f.setRequired(query.value(2).toBool()); f.setLength(len); f.setPrecision(precision); f.setDefaultValue(defVal); f.setSqlType(query.value(1).toInt()); info.append(f); } } return info;}QString QPSQLDriver::formatValue(const QSqlField &field, bool) const{ QString r; if (field.isNull()) { r = QLatin1String("NULL"); } else { switch (field.type()) { case QVariant::DateTime: if (field.value().toDateTime().isValid()) { QDate dt = field.value().toDateTime().date(); QTime tm = field.value().toDateTime().time(); // msecs need to be right aligned otherwise psql // interpretes them wrong r = QLatin1String("'") + QString::number(dt.year()) + QLatin1String("-") + QString::number(dt.month()) + QLatin1String("-") + QString::number(dt.day()) + QLatin1String(" ") + tm.toString() + QLatin1String(".") + QString::number(tm.msec()).rightJustified(3, QLatin1Char('0')) + QLatin1String("'"); } else { r = QLatin1String("NULL"); } break; case QVariant::Time: if (field.value().toTime().isValid()) { r = field.value().toTime().toString(Qt::ISODate); } else { r = QLatin1String("NULL"); } case QVariant::String: { // Escape '\' characters r = QSqlDriver::formatValue(field); r.replace(QLatin1String("\\"), QLatin1String("\\\\")); break; } case QVariant::Bool: if (field.value().toBool()) r = QLatin1String("TRUE"); else r = QLatin1String("FALSE"); break; case QVariant::ByteArray: { QByteArray ba(field.value().toByteArray()); size_t len; unsigned char *data = PQescapeBytea((unsigned char*)ba.constData(), ba.size(), &len); r += QLatin1Char('\''); r += QLatin1String((const char*)data); r += QLatin1Char('\''); PQfreemem(data); break; } default: r = QSqlDriver::formatValue(field); break; } } return r;}QString QPSQLDriver::escapeIdentifier(const QString &identifier, IdentifierType type) const{ Q_UNUSED(type); QString res = identifier; res.replace(QLatin1Char('"'), QLatin1String("\"\"")); res.prepend(QLatin1Char('"')).append(QLatin1Char('"')); int idx = identifier.indexOf(QLatin1Char('.')); if (idx != -1) res.replace(QLatin1Char('.'), QLatin1String("\".\"")); return res;}bool QPSQLDriver::isOpen() const{ return PQstatus(d->connection) == CONNECTION_OK;}QPSQLDriver::Protocol QPSQLDriver::protocol() const{ return d->pro;}QByteArray QPSQLDriver::loRead(int oid){ QByteArray ba; this->beginTransaction(); int fd = lo_open( d->connection, oid, INV_READ ); int size = 0; int retval = lo_lseek( d->connection, fd, 0L, SEEK_END ); if ( retval >= 0 ) { size = lo_tell( d->connection, fd ); lo_lseek( d->connection, fd, 0L, SEEK_SET ); } if ( size == 0 ) { lo_close( d->connection, fd ); this->commitTransaction(); return ba; } char *buf = (char*)malloc( size ); retval = lo_read( d->connection, fd, buf, size ); ba = QByteArray( buf, size ); free( buf ); if ( retval < size ) qWarning( "QPSQLDriver::readBlob: unable to read large object: read(%d), size(%d)", retval, size );// else qDebug( "QPSQLDriver::readBlob: read large object: read(%d), size(%d)", retval, size ); lo_close( d->connection, fd ); this->commitTransaction(); return ba;}int QPSQLDriver::loCreate(){ this->beginTransaction(); int id = lo_creat(d->connection, INV_READ|INV_WRITE); if ( id == InvalidOid ) { qWarning( "QPSQLDriver::creatBlob: unable to create large object" ); this->rollbackTransaction(); } this->commitTransaction(); return id;}bool QPSQLDriver::loUnlink( int oid ){ this->beginTransaction(); int id = lo_unlink(d->connection, oid ); if ( id == -1 ) { qWarning( "QPSQLDriver::dropBlob: unable to drop large object" ); this->rollbackTransaction(); return false; } this->commitTransaction(); return true;}bool QPSQLDriver::loWrite(int oid, QByteArray &blobData){ this->beginTransaction(); int fd = lo_open( d->connection, oid, INV_READ|INV_WRITE); if ( fd < 0) { qWarning( "QPSQLDriver::writeBlob: unable to open large object for read" ); this->rollbackTransaction(); return FALSE; } size_t len = blobData.size(); int nBytesWriten = lo_write(d->connection, fd, blobData.data(), len); if ( nBytesWriten < 0) { qWarning( "QPSQLDriver::writeBlob: unable to write large object" ); this->rollbackTransaction(); return FALSE; } this->commitTransaction(); return TRUE;}
#ifndef QPSQLDRIVER_H#define QPSQLDRIVER_H#include <QtSql/qsqlresult.h>#include <QtSql/qsqldriver.h>typedef struct pg_conn QPSQLconn;typedef struct pg_result QPSQLresult;class QPSQLResultPrivate;class QPSQLDriverPrivate;class QPSQLDriver;class QSqlRecordInfo;class QPSQLResult : public QSqlResult{ friend class QPSQLResultPrivate; public: QPSQLResult(const QPSQLDriver* db, const QPSQLDriverPrivate* p); ~QPSQLResult(); QVariant handle() const; QPSQLDriver* pgDriver(); protected: void cleanup(); bool fetch(int i); bool fetchFirst(); bool fetchLast(); QVariant data(int i); bool isNull(int field); bool reset (const QString& query); int size(); int numRowsAffected(); QSqlRecord record() const; QVariant lastInsertId() const; private: QPSQLResultPrivate *d;};class QPSQLDriver : public QSqlDriver{ Q_OBJECT public: enum Protocol { Version6 = 6, Version7 = 7, Version71 = 8, Version73 = 9 }; explicit QPSQLDriver(QObject *parent=0); explicit QPSQLDriver(QPSQLconn *conn, QObject *parent=0); ~QPSQLDriver(); bool hasFeature(DriverFeature f) const; bool open(const QString & db, const QString & user, const QString & password, const QString & host, int port, const QString& connOpts); bool isOpen() const; void close(); QSqlResult *createResult() const; QStringList tables(QSql::TableType) const; QSqlIndex primaryIndex(const QString& tablename) const; QSqlRecord record(const QString& tablename) const; Protocol protocol() const; QVariant handle() const; QString escapeIdentifier(const QString &identifier, IdentifierType type) const; QString formatValue(const QSqlField &field, bool trimStrings) const; int loCreate(); QByteArray loRead(int oid); bool loWrite(int oid, QByteArray &blobData); bool loUnlink(int oid); void checkNotify(); signals: void sqlNotify( const QString &str ); protected: bool beginTransaction(); bool commitTransaction(); bool rollbackTransaction(); private: void init(); QPSQLDriverPrivate *d;};#endif // QSQL_PSQL_H
