16 class OrmNoObjectException : OrmException
20 class OrmSqlException : OrmException
23 OrmSqlException(const QString& text) : mText(text) {}
24 QString text() const { return mText; }
31 class OrmRecord : protected QSqlRecord
35 static T hydrate(const QSqlRecord& record);
36 void update(QString col, QVariant value = QVariant()); // updates specified column 'col'
39 QVariant value(QString col) const;
40 void setValue(QString col, QVariant value);
42 static T loadOne(QSqlQuery query);
43 static QList<T> load(QSqlQuery query);
46 static QSqlRecord toRecord(const QList<QSqlField> & columnList);
47 // all record items/columns are in one table
48 static QString columnsForSelect(const QString& prefix = QString());
49 static QString selectQuery();
50 static QString updateQuery();
52 static QVariant convertToC(QVariant value, QVariant::Type colType);
53 static QVariant convertToDb(QVariant value, QVariant::Type colType);
57 OrmRecord<T>::OrmRecord()
59 QSqlRecord::operator=(T::sColumns);
63 T OrmRecord<T>::hydrate(const QSqlRecord& record)
66 object.QSqlRecord::operator=(record);
70 // updates specified column 'col'
71 // if the value is not specified as an argument,
72 // it's taken from the reford itself
73 // see also: setValue() method for more details
75 void OrmRecord<T>::update(QString col, QVariant value)
78 query.prepare(QString(updateQuery() + "SET %1 = :col WHERE id = :id").arg(col));
79 if(value.isValid()) // take 'col' value from the method's arguments
80 query.bindValue(":col", value);
81 else // take 'col' value from the record; see setValue()
82 query.bindValue(":col", convertToDb(this->value(col), this->value(col).type()));
83 query.bindValue(":id", this->value("id"));
84 //query.bindValue(":id", convertToDb(value("id"), QVariant::Int));
89 QVariant OrmRecord<T>::value(QString col) const
91 return convertToC(QSqlRecord::value(col), T::sColumns.field(col).type());
95 void OrmRecord<T>::setValue(QString col, QVariant value)
97 QSqlRecord::setValue(col, convertToDb(value, T::sColumns.field(col).type()));
100 template <typename T>
101 T OrmRecord<T>::loadOne(QSqlQuery query)
103 if (!query.isActive())
107 throw OrmSqlException(query.lastError().text());
113 throw OrmNoObjectException();
116 return hydrate(query.record());
119 template <typename T>
120 QList<T> OrmRecord<T>::load(QSqlQuery query)
122 if (!query.isActive())
126 qDebug() << "Error: " << query.lastError().driverText() << "; Type: " << query.lastError().type();
127 throw OrmSqlException(query.lastError().text());
131 qDebug() << "SQL OK";
138 objects << hydrate(query.record());
140 qDebug() << "Fetch done";
144 template <typename T>
145 QString OrmRecord<T>::columnsForSelect(const QString& prefix)
147 QStringList prefixedColumns;
148 for (int i=0; i<T::sColumns.count(); i++)
150 prefixedColumns.append(prefix.isEmpty() ?
151 T::sColumns.field(i).name() :
152 QString("%1.%2").arg(prefix, T::sColumns.field(i).name()));
154 return prefixedColumns.join(",");
157 template <typename T>
158 QString OrmRecord<T>::selectQuery()
160 return QString("SELECT %1 FROM %2 ").arg(columnsForSelect(), T::sTableName);
163 template <typename T>
164 QString OrmRecord<T>::updateQuery()
166 return QString("UPDATE %1 ").arg(T::sTableName);
169 template <typename T>
170 QSqlRecord OrmRecord<T>::toRecord(const QList<QSqlField> & columnList)
173 foreach (const QSqlField & col, columnList)
180 template <typename T>
181 QVariant OrmRecord<T>::convertToC(QVariant value, QVariant::Type colType)
183 if (colType == QVariant::DateTime && value.canConvert<uint>())
186 date.setTimeSpec(Qt::UTC);
187 date.setTime_t(value.toUInt());
194 template <typename T>
195 QVariant OrmRecord<T>::convertToDb(QVariant value, QVariant::Type colType)
197 if (colType == QVariant::DateTime && value.canConvert<QDateTime>())
199 return value.toDateTime().toTime_t();
205 #endif // ORMRECORD_H