#ifndef ORMRECORD_H #define ORMRECORD_H #include #include #include #include #include #include #include class OrmException { public: OrmException(const QString& text) : mText(text) {}; virtual ~OrmException(){}; virtual const QString& text() const { return mText; } private: QString mText; }; class OrmNoObjectException : public OrmException { public: OrmNoObjectException() : OrmException("No object exception"){}; ~OrmNoObjectException(){}; }; class OrmSqlException : public OrmException { public: OrmSqlException(const QString& text) : OrmException( QString("Sql error: ") + text ) {}; ~OrmSqlException(){}; }; template class OrmRecord : protected QSqlRecord { public: OrmRecord(); static T hydrate(const QSqlRecord& record); void update(QString col, QVariant value = QVariant()); // updates specified column 'col' protected: QVariant value(QString col) const; void setValue(QString col, QVariant value); static T loadOne(QSqlQuery query); static QList load(QSqlQuery query); // auxiliary methods static QSqlRecord toRecord(const QList & columnList); // all record items/columns are in one table static QString columnsForSelect(const QString& prefix = QString()); static QString selectQuery(); static QString updateQuery(); static QVariant convertToC(QVariant value, QVariant::Type colType); static QVariant convertToDb(QVariant value, QVariant::Type colType); }; template OrmRecord::OrmRecord() { QSqlRecord::operator=(T::sColumns); } template T OrmRecord::hydrate(const QSqlRecord& record) { T object; object.QSqlRecord::operator=(record); return object; } // updates specified column 'col' // if the value is not specified as an argument, // it's taken from the reford itself // see also: setValue() method for more details template void OrmRecord::update(QString col, QVariant value) { QSqlQuery query; query.prepare(QString(updateQuery() + "SET %1 = :col WHERE id = :id").arg(col)); if(value.isValid()) // take 'col' value from the method's arguments query.bindValue(":col", value); else // take 'col' value from the record; see setValue() query.bindValue(":col", convertToDb(this->value(col), this->value(col).type())); query.bindValue(":id", this->value("id")); //query.bindValue(":id", convertToDb(value("id"), QVariant::Int)); query.exec(); } template QVariant OrmRecord::value(QString col) const { return convertToC(QSqlRecord::value(col), T::sColumns.field(col).type()); } template void OrmRecord::setValue(QString col, QVariant value) { QSqlRecord::setValue(col, convertToDb(value, T::sColumns.field(col).type())); } template T OrmRecord::loadOne(QSqlQuery query) { if (!query.isActive()) { if (!query.exec()) { throw OrmSqlException(query.lastError().text()); } } if (!query.next()) { throw OrmNoObjectException(); } return hydrate(query.record()); } template QList OrmRecord::load(QSqlQuery query) { if (!query.isActive()) { if (!query.exec()) { qDebug() << "Error: " << query.lastError().driverText() << "; Type: " << query.lastError().type(); throw OrmSqlException(query.lastError().text()); } else { qDebug() << "SQL OK"; } } QList objects; while (query.next()) { objects << hydrate(query.record()); } qDebug() << "Fetch done"; return objects; } template QString OrmRecord::columnsForSelect(const QString& prefix) { QStringList prefixedColumns; for (int i=0; i QString OrmRecord::selectQuery() { return QString("SELECT %1 FROM %2 ").arg(columnsForSelect(), T::sTableName); } template QString OrmRecord::updateQuery() { return QString("UPDATE %1 ").arg(T::sTableName); } template QSqlRecord OrmRecord::toRecord(const QList & columnList) { QSqlRecord record; foreach (const QSqlField & col, columnList) { record.append(col); } return record; } template QVariant OrmRecord::convertToC(QVariant value, QVariant::Type colType) { if (colType == QVariant::DateTime && value.canConvert()) { QDateTime date; date.setTimeSpec(Qt::UTC); date.setTime_t(value.toUInt()); return date; } return value; } template QVariant OrmRecord::convertToDb(QVariant value, QVariant::Type colType) { if (colType == QVariant::DateTime && value.canConvert()) { return value.toDateTime().toTime_t(); } return value; } #endif // ORMRECORD_H