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 } |
|