1 /*

     2 * Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

     3 * All rights reserved.

     4 * This component and the accompanying materials are made available

     5 * under the terms of "Eclipse Public License v1.0"

     6 * which accompanies this distribution, and is available

     7 * at the URL "http://www.eclipse.org/legal/epl-v10.html".

     8 *

     9 * Initial Contributors:

    10 * Nokia Corporation - initial contribution.

    11 *

    12 * Contributors:

    13 *

    14 * Description:

    15 *

    16 */

    17 #include "activitydatastorage.h"

    18 

    19 #include <QSqlDatabase>

    20 #include <QSqlQuery>

    21 #include <QSqlError>

    22 #include <QStringList>

    23 

    24 const char KConnectionName[] = "ActivityDataStorage";

    25 

    26 ActivityDataStorage::ActivityDataStorage()

    27 {

    28     QSqlDatabase database;

    29     if (QSqlDatabase::contains(KConnectionName)) {

    30         database = QSqlDatabase::database(KConnectionName);

    31     } else {

    32         database = QSqlDatabase::addDatabase("QSQLITE", KConnectionName);

    33         database.setDatabaseName("activitydatastorage.db");

    34         if (!database.open()) {

    35             qFatal(qPrintable(database.lastError().text()));

    36         }

    37     }

    38 

    39     if (!checkTables()) {

    40         recreateTables();

    41     }

    42 }

    43 

    44 ActivityDataStorage::~ActivityDataStorage()

    45 {

    46     QSqlDatabase::database(KConnectionName).close();

    47 }

    48 

    49 bool ActivityDataStorage::addActivity(const QString &activityId, const QVariant &data)

    50 {

    51     QSqlDatabase database = QSqlDatabase::database(KConnectionName);

    52 

    53     QByteArray streamedData;

    54     {

    55         QDataStream stream(&streamedData, QIODevice::WriteOnly);

    56         stream << data;

    57     }

    58 

    59     // insert data

    60     QSqlQuery query(database);

    61     if (!query.prepare("INSERT INTO Activities(Name, Data) VALUES(:Name, :Data)")) {

    62         qCritical(qPrintable(query.lastError().text()));

    63         return false;

    64     }

    65     query.bindValue(":Name", activityId);

    66     query.bindValue(":Data", streamedData);

    67     if (!query.exec()) {

    68         qCritical(qPrintable(query.lastError().text()));

    69         return false;

    70     }

    71 

    72     return true;

    73 }

    74 

    75 bool ActivityDataStorage::removeActivity(const QString &activityId)

    76 {

    77     QString sqlCommand = QString("DELETE FROM Activities WHERE Name = '%1'").arg(activityId);

    78     QSqlQuery query(QSqlDatabase::database(KConnectionName));

    79     if (!query.exec(sqlCommand)) {

    80         qCritical(qPrintable(query.lastError().text()));

    81         return false;

    82     }

    83     return query.numRowsAffected() > 0;

    84 }

    85 

    86 bool ActivityDataStorage::updateActivity(const QString &activityId, const QVariant &data)

    87 {

    88     QSqlDatabase database = QSqlDatabase::database(KConnectionName);

    89 

    90     QByteArray streamedData;

    91     {

    92         QDataStream stream(&streamedData, QIODevice::WriteOnly);

    93         stream << data;

    94     }

    95 

    96     // update data

    97     QSqlQuery query(database);

    98     if (!query.prepare("UPDATE Activities SET Data = :Data WHERE Name = :Name")) {

    99         qCritical(qPrintable(query.lastError().text()));

   100         return false;

   101     }

   102     query.bindValue(":Data", streamedData);

   103     query.bindValue(":Name", activityId);

   104     if (!query.exec()) {

   105         qCritical(qPrintable(query.lastError().text()));

   106         return false;

   107     }

   108 

   109     return query.numRowsAffected() > 0;

   110 }

   111 

   112 QVariant ActivityDataStorage::activityData(const QString &activityId) const

   113 {

   114     QSqlDatabase database = QSqlDatabase::database(KConnectionName);

   115     QSqlQuery query(database);

   116     if (!query.exec(QString("SELECT Data FROM Activities WHERE Name = '%1'").arg(activityId))) {

   117         qCritical(qPrintable(query.lastError().text()));

   118         return QVariant();

   119     }

   120 

   121     QVariant result;

   122     if (query.next()) {

   123         QByteArray data(query.value(0).toByteArray());

   124         QDataStream stream(&data, QIODevice::ReadOnly);

   125         stream >> result;

   126     }

   127     return result;

   128 }

   129 

   130 bool ActivityDataStorage::checkTables()

   131 {

   132     QStringList expectedTables("Activities");

   133     QStringList actualTables = QSqlDatabase::database(KConnectionName).tables();

   134     return (expectedTables == actualTables);

   135 }

   136 

   137 void ActivityDataStorage::recreateTables()

   138 {

   139     QSqlDatabase database = QSqlDatabase::database(KConnectionName);

   140 

   141     if (!database.transaction()) {

   142         qFatal(qPrintable(database.lastError().text()));

   143     }

   144 

   145     // drop any existing tables

   146     {

   147         QSqlQuery dropQuery(database);

   148         foreach(const QString &tableName, database.tables()) {

   149             if (!dropQuery.exec(QString("DROP TABLE %1").arg(tableName))) {

   150                 qFatal(qPrintable(dropQuery.lastError().text()));

   151             }

   152         }

   153     }

   154 

   155     // create new table

   156     {

   157         QSqlQuery createQuery(database);

   158         QString statement(

   159             "CREATE TABLE Activities("

   160             "Name TEXT NOT NULL PRIMARY KEY UNIQUE,"

   161             "Data BLOB NOT NULL)");

   162 

   163         if (!createQuery.exec(statement)) {

   164             qFatal(qPrintable(createQuery.lastError().text()));

   165         }

   166     }

   167 

   168     if (!database.commit()) {

   169         qFatal(qPrintable(database.lastError().text()));

   170     }

   171 }