the table name depends from the plugin name of who is using storage

svn path=/trunk/KDE/kdelibs/; revision=1188938
This commit is contained in:
Marco Martin 2010-10-23 12:04:24 +00:00
parent 0bd34bdd36
commit 72110136bc
3 changed files with 53 additions and 24 deletions

View File

@ -199,7 +199,7 @@ void DataContainerPrivate::store()
q->setNeedsToBeStored(false); q->setNeedsToBeStored(false);
if (storage == NULL) { if (storage == NULL) {
storage = new Storage(de->name(), 0); storage = new Storage(q);
} }
KConfigGroup op = storage->operationDescription("save"); KConfigGroup op = storage->operationDescription("save");
@ -220,7 +220,7 @@ void DataContainerPrivate::store()
} }
++it; ++it;
if (storage == NULL) { if (storage == NULL) {
storage = new Storage(de->name(), 0); storage = new Storage(q);
} }
ServiceJob* job = storage->startOperationCall(op); ServiceJob* job = storage->startOperationCall(op);
storageCount++; storageCount++;
@ -244,7 +244,7 @@ void DataContainerPrivate::retrieve()
return; return;
} }
if (!storage) { if (!storage) {
storage = new Storage(de->name(), 0); storage = new Storage(q);
} }
KConfigGroup retrieveGroup = storage->operationDescription("retrieve"); KConfigGroup retrieveGroup = storage->operationDescription("retrieve");

View File

@ -30,6 +30,11 @@
#include <kdebug.h> #include <kdebug.h>
#include <kstandarddirs.h> #include <kstandarddirs.h>
//Plasma
#include "applet.h"
#include "dataengine.h"
#include "abstractrunner.h"
static uint connectionId = 0; static uint connectionId = 0;
//Storage Job implentation //Storage Job implentation
@ -37,16 +42,19 @@ StorageJob::StorageJob(const QString& destination,
const QString& operation, const QString& operation,
const QMap<QString, QVariant>& parameters, const QMap<QString, QVariant>& parameters,
QObject *parent) QObject *parent)
: ServiceJob(destination, operation, parameters, parent) : ServiceJob(destination, operation, parameters, parent),
m_clientName(destination)
{ {
m_db = QSqlDatabase::addDatabase("QSQLITE", QString("plasma-storage-%1").arg(++connectionId)); m_db = QSqlDatabase::addDatabase("QSQLITE", QString("plasma-storage-%1").arg(++connectionId));
m_db.setDatabaseName(KStandardDirs::locateLocal("appdata", "plasma-storage.db")); m_db.setDatabaseName(KStandardDirs::locateLocal("appdata", "plasma-storage.db"));
if (!m_db.open()) { if (!m_db.open()) {
kWarning() << "Unable to open the plasma storage cache database: " << m_db.lastError(); kWarning() << "Unable to open the plasma storage cache database: " << m_db.lastError();
} else if (!m_db.tables().contains("data")) { } else if (!m_db.tables().contains(m_clientName)) {
QSqlQuery query(m_db); QSqlQuery query(m_db);
query.exec("create table data (id varchar(100) primary key, source varchar(100), data clob, date datetime)"); //bindValue doesn't seem to be able to replace stuff in create table
query.prepare(QString("create table ")+m_clientName+" (valueGroup varchar(256), id varchar(256), data clob, date datetime, primary key (valueGroup, id))");
query.exec();
} }
} }
@ -60,13 +68,13 @@ void StorageJob::start()
if (operationName() == "save") { if (operationName() == "save") {
QSqlQuery query(m_db); QSqlQuery query(m_db);
query.prepare("delete from data where key = :id"); query.prepare("delete from "+m_clientName+" where key = :id");
query.bindValue(":id", params["key"].toString()); query.bindValue(":id", params["key"].toString());
query.exec(); query.exec();
query.prepare("insert into data values(:id, :source, :datavalue, 'now')"); query.prepare("insert into "+m_clientName+" values(:valueGroup, :id, :datavalue, 'now')");
query.bindValue(":id", params["key"].toString()); query.bindValue(":id", params["key"].toString());
query.bindValue(":source", params["source"].toString()); query.bindValue(":valueGroup", params["valueGroup"].toString());
query.bindValue(":datavalue", params["data"]); query.bindValue(":datavalue", params["data"]);
const bool success = query.exec(); const bool success = query.exec();
setResult(success); setResult(success);
@ -74,25 +82,22 @@ void StorageJob::start()
} else if (operationName() == "retrieve") { } else if (operationName() == "retrieve") {
QSqlQuery query(m_db); QSqlQuery query(m_db);
query.prepare("delete from data where date < :date"); query.prepare("delete from "+m_clientName+" where date < :date");
QDateTime time(QDateTime::currentDateTime()); QDateTime time(QDateTime::currentDateTime());
time.addDays(-2); time.addDays(-2);
query.bindValue(":date", time.toTime_t()); query.bindValue(":date", time.toTime_t());
query.exec(); query.exec();
//a bit redundant but should be the faster way with less string concatenation as possible //a bit redundant but should be the faster way with less string concatenation as possible
if (params["source"].isNull() && params["key"].isNull()) { if (params["valueGroup"].isNull()) {
setError(true); setError(true);
return; return;
} else if (params["source"].isNull()) {
query.prepare("select * from data where key=:key");
query.bindValue(":source", params["source"].toString());
} else if (params["key"].isNull()) { } else if (params["key"].isNull()) {
query.prepare("select * from data where source=:source"); query.prepare("select * from "+m_clientName+" where valueGroup=:valueGroup");
query.bindValue(":source", params["source"].toString()); query.bindValue(":valueGroup", params["valueGroup"].toString());
} else { } else {
query.prepare("select * from data where source=:source and key=:key"); query.prepare("select * from "+m_clientName+" where valueGroup=:valueGroup and key=:key");
query.bindValue(":source", params["source"].toString()); query.bindValue(":valueGroup", params["valueGroup"].toString());
query.bindValue(":key", params["key"].toString()); query.bindValue(":key", params["key"].toString());
} }
@ -116,7 +121,7 @@ void StorageJob::start()
} else if (operationName() == "expire") { } else if (operationName() == "expire") {
QSqlQuery query(m_db); QSqlQuery query(m_db);
query.prepare("delete from data where date < :date"); query.prepare("delete from "+m_clientName+" where date < :date");
QDateTime time(QDateTime::currentDateTime()); QDateTime time(QDateTime::currentDateTime());
time.addDays(-2); time.addDays(-2);
query.bindValue(":date", time.toTime_t()); query.bindValue(":date", time.toTime_t());
@ -129,13 +134,36 @@ void StorageJob::start()
Plasma::ServiceJob* Storage::createJob(const QString &operation, QMap<QString, QVariant> &parameters) Plasma::ServiceJob* Storage::createJob(const QString &operation, QMap<QString, QVariant> &parameters)
{ {
return new StorageJob(m_serviceName, operation, parameters); return new StorageJob(m_clientName, operation, parameters, this);
} }
//Storage implementation //Storage implementation
Storage::Storage(const QString& destination, QObject* parent) : Plasma::Service(parent) Storage::Storage(QObject* parent) : Plasma::Service(parent)
{ {
m_serviceName = destination; //search among parents for an applet or dataengine: if found call the table as its plugin name
QObject *parentObject = this;
QString clientName("data");
while ((parentObject = parentObject->parent())) {
Plasma::Applet *applet = qobject_cast<Plasma::Applet *>(parentObject);
if (applet) {
m_clientName = applet->pluginName();
break;
}
Plasma::DataEngine *engine = qobject_cast<Plasma::DataEngine *>(parentObject);
if (engine) {
m_clientName = engine->pluginName();
break;
}
Plasma::AbstractRunner *runner = qobject_cast<Plasma::AbstractRunner *>(parentObject);
if (runner) {
m_clientName = runner->id();
break;
}
}
setName("storage"); setName("storage");
} }

View File

@ -40,6 +40,7 @@ public:
void start(); void start();
private: private:
QSqlDatabase m_db; QSqlDatabase m_db;
QString m_clientName;
}; };
//End StorageJob //End StorageJob
@ -48,14 +49,14 @@ class Storage : public Plasma::Service
Q_OBJECT Q_OBJECT
public: public:
Storage(const QString& destination, QObject* parent); Storage(QObject* parent);
~Storage(); ~Storage();
protected: protected:
Plasma::ServiceJob* createJob(const QString &operation, QMap<QString, QVariant> &parameters); Plasma::ServiceJob* createJob(const QString &operation, QMap<QString, QVariant> &parameters);
private: private:
QString m_serviceName; QString m_clientName;
}; };