124 lines
4.4 KiB
C++
124 lines
4.4 KiB
C++
/////////////////////////////////////////////////////////////////////////
|
|
// storage.cpp //
|
|
// //
|
|
// Copyright (C) 2010 Brian Pritchett <batenkaitos@gmail.com> //
|
|
// //
|
|
// This library is free software; you can redistribute it and/or //
|
|
// modify it under the terms of the GNU Lesser General Public //
|
|
// License as published by the Free Software Foundation; either //
|
|
// version 2.1 of the License, or (at your option) any later version. //
|
|
// //
|
|
// This library 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 //
|
|
// Lesser General Public License for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU Lesser General Public //
|
|
// License along with this library; if not, write to the Free Software //
|
|
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA //
|
|
// 02110-1301 USA //
|
|
/////////////////////////////////////////////////////////////////////////
|
|
#include "private/storage_p.h"
|
|
|
|
//Qt
|
|
#include <QSqlError>
|
|
#include <QSqlQuery>
|
|
#include <QSqlRecord>
|
|
|
|
//KDE
|
|
#include <kdebug.h>
|
|
#include <kstandarddirs.h>
|
|
|
|
static uint connectionId = 0;
|
|
|
|
//Storage Job implentation
|
|
StorageJob::StorageJob(const QString& destination,
|
|
const QString& operation,
|
|
const QMap<QString, QVariant>& parameters,
|
|
QObject *parent)
|
|
: ServiceJob(destination, operation, parameters, parent)
|
|
{
|
|
m_db = QSqlDatabase::addDatabase("QSQLITE", QString("plasma-storage-%1").arg(++connectionId));
|
|
m_db.setDatabaseName(KStandardDirs::locateLocal("appdata", "plasma-storage.db"));
|
|
|
|
if (!m_db.open()) {
|
|
kWarning() << "Unable to open the plasma storage cache database: " << m_db.lastError();
|
|
} else if (!m_db.tables().contains("data")) {
|
|
QSqlQuery query(m_db);
|
|
query.exec("create table data (id varchar(100) primary key, source varchar(100), data clob, date datetime)");
|
|
}
|
|
}
|
|
|
|
void StorageJob::start()
|
|
{
|
|
if (!m_db.isOpen()) {
|
|
return;
|
|
}
|
|
|
|
QMap<QString, QVariant> params = parameters();
|
|
|
|
if (operationName() == "save") {
|
|
QSqlQuery query(m_db);
|
|
query.prepare("delete from data where key = :id");
|
|
query.bindValue(":id", params["key"].toString());
|
|
query.exec();
|
|
|
|
query.prepare("insert into data values(:id, :source, :datavalue, 'now')");
|
|
query.bindValue(":id", params["key"].toString());
|
|
query.bindValue(":source", params["source"].toString());
|
|
query.bindValue(":datavalue", params["data"]);
|
|
const bool success = query.exec();
|
|
setResult(success);
|
|
return;
|
|
|
|
} else if (operationName() == "retrieve") {
|
|
QSqlQuery query(m_db);
|
|
query.prepare("delete from data where date < :date");
|
|
QDateTime time(QDateTime::currentDateTime());
|
|
time.addDays(-2);
|
|
query.bindValue(":date", time.toTime_t());
|
|
query.exec();
|
|
|
|
query.prepare("select * from data where source=:source");
|
|
query.bindValue(":source", params["source"].toString());
|
|
const bool success = query.exec();
|
|
|
|
QHash<QString, QVariant> h;
|
|
if (success) {
|
|
QSqlRecord rec = query.record();
|
|
const int keyColumn = rec.indexOf("id");
|
|
const int dataColumn = rec.indexOf("data");
|
|
|
|
while (query.next()) {
|
|
h[query.value(keyColumn).toString()] = query.value(dataColumn).toString();
|
|
}
|
|
|
|
setResult(h);
|
|
return;
|
|
} else {
|
|
return;
|
|
}
|
|
|
|
} else {
|
|
setError(true);
|
|
}
|
|
}
|
|
|
|
Plasma::ServiceJob* Storage::createJob(const QString &operation, QMap<QString, QVariant> ¶meters)
|
|
{
|
|
return new StorageJob(m_serviceName, operation, parameters);
|
|
}
|
|
|
|
//Storage implementation
|
|
Storage::Storage(const QString& destination, QObject* parent) : Plasma::Service(parent)
|
|
{
|
|
m_serviceName = destination;
|
|
setName("storage");
|
|
}
|
|
|
|
Storage::~Storage()
|
|
{
|
|
}
|
|
|
|
#include "storage_p.moc"
|