Removed one comment and fixed typos.

[toast/confclerk.git] / src / orm / ormrecord.h

diff --git a/src/orm/ormrecord.h b/src/orm/ormrecord.h
index 3c93461c379f790e798e9951c40aad50ee13c943..85d32c29fc10cd00c26f2f074a2abb4ef02cefca 100644 (file)
--- a/src/orm/ormrecord.h
+++ b/src/orm/ormrecord.h
@@ -1,3 +1,22 @@
+/*
+ * Copyright (C) 2010 Ixonos Plc.
+ * Copyright (C) 2011 Philipp Spitzer, gregor herrmann
+ *
+ * This file is part of ConfClerk.
+ *
+ * ConfClerk is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the Free
+ * Software Foundation, either version 2 of the License, or (at your option)
+ * any later version.
+ *
+ * ConfClerk 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 General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * ConfClerk.  If not, see <http://www.gnu.org/licenses/>.
+ */
 #ifndef ORMRECORD_H
 #define ORMRECORD_H
 
@@ -11,20 +30,26 @@
 
 class OrmException
 {
+public:
+    OrmException(const QString& text) : mText(text) {}
+    virtual ~OrmException(){}
+    virtual const QString& text() const { return mText; }
+private:
+    QString mText;
 };
 
-class OrmNoObjectException : OrmException
+class OrmNoObjectException : public OrmException
 {
+public:
+    OrmNoObjectException() : OrmException("No object exception"){}
+    ~OrmNoObjectException(){}
 };
 
-class OrmSqlException : OrmException
+class OrmSqlException : public OrmException
 {
 public:
-    OrmSqlException(const QString& text) : mText(text) {}
-    QString text() const { return mText; }
-
-private:
-    QString mText;
+    OrmSqlException(const QString& text) : OrmException( QString("Sql error: ") + text ) {}
+    ~OrmSqlException(){}
 };
 
 template <typename T>
@@ -33,6 +58,7 @@ 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;
@@ -42,9 +68,11 @@ protected:
     static QList<T> load(QSqlQuery query);
 
     // auxiliary methods
+    static QSqlRecord toRecord(const QList<QSqlField> & columnList);
+    // all record items/columns are in one table
     static QString columnsForSelect(const QString& prefix = QString());
     static QString selectQuery();
-    static QSqlRecord toRecord(const QList<QSqlField> & columnList);
+    static QString updateQuery();
 
     static QVariant convertToC(QVariant value, QVariant::Type colType);
     static QVariant convertToDb(QVariant value, QVariant::Type colType);
@@ -64,6 +92,23 @@ T OrmRecord<T>::hydrate(const QSqlRecord& record)
     return object;
 }
 
+// updates specified column 'col'
+// if the value is not specified as an argument,
+// it's taken from the record itself
+// see also: setValue() method for more details
+template <typename T>
+void OrmRecord<T>::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.exec();
+}
+
 template <typename T>
 QVariant OrmRecord<T>::value(QString col) const
 {
@@ -83,13 +128,13 @@ T OrmRecord<T>::loadOne(QSqlQuery query)
     {
         if (!query.exec())
         {
-            throw new OrmSqlException(query.lastError().text());
+            throw OrmSqlException(query.lastError().text());
         }
     }
 
     if (!query.next())
     {
-        throw new OrmNoObjectException();
+        throw OrmNoObjectException();
     }
 
     return hydrate(query.record());
@@ -102,13 +147,8 @@ QList<T> OrmRecord<T>::load(QSqlQuery query)
     {
         if (!query.exec())
         {
-            //TODO Palo: exception handling !
-               QString start = "START\n";
-               QString end = "\nEND\n";
-               QString message = start + query.lastError().text() + end;
-               const char *data = message.toLatin1().data();
-               qDebug(data);
-            throw new OrmSqlException(query.lastError().text());
+            qDebug() << "Error: " << query.lastError().driverText() << "; Type: " << query.lastError().type();
+            throw OrmSqlException(query.lastError().text());
         }
     }
 
@@ -117,7 +157,6 @@ QList<T> OrmRecord<T>::load(QSqlQuery query)
     {
         objects << hydrate(query.record());
     }
-
     return objects;
 }
 
@@ -140,13 +179,19 @@ QString OrmRecord<T>::selectQuery()
     return QString("SELECT %1 FROM %2 ").arg(columnsForSelect(), T::sTableName);
 }
 
+template <typename T>
+QString OrmRecord<T>::updateQuery()
+{
+    return QString("UPDATE %1 ").arg(T::sTableName);
+}
+
 template <typename T>
 QSqlRecord OrmRecord<T>::toRecord(const QList<QSqlField> & columnList)
 {
     QSqlRecord record;
-    foreach (const QSqlField & col, columnList)
+    for(int i=0; i< columnList.count(); i++)
     {
-        record.append(col);
+        record.append(columnList[i]);
     }
     return record;
 }
@@ -177,3 +222,4 @@ QVariant OrmRecord<T>::convertToDb(QVariant value, QVariant::Type colType)
 }
 
 #endif // ORMRECORD_H
+