3 files changed, 14 insertions, 0 deletions

diff --git a/common/genericresource.cpp b/common/genericresource.cpp
index 7b8d9bb..3655a63 100644
--- a/common/genericresource.cpp
+++ b/common/genericresource.cpp
@@ -292,12 +292,22 @@ KAsync::Job<void> GenericResource::replay(const QByteArray &type, const QByteArr
 
 void GenericResource::removeFromDisk(const QByteArray &instanceIdentifier)
 {
+    Warning() << "Removing from generic resource";
     Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier, Akonadi2::Storage::ReadWrite).removeFromDisk();
     Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier + ".userqueue", Akonadi2::Storage::ReadWrite).removeFromDisk();
     Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier + ".synchronizerqueue", Akonadi2::Storage::ReadWrite).removeFromDisk();
     Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier + ".changereplay", Akonadi2::Storage::ReadWrite).removeFromDisk();
 }
 
+qint64 GenericResource::diskUsage(const QByteArray &instanceIdentifier)
+{
+    auto size = Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier, Akonadi2::Storage::ReadOnly).diskUsage();
+    size += Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier + ".userqueue", Akonadi2::Storage::ReadOnly).diskUsage();
+    size += Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier + ".synchronizerqueue", Akonadi2::Storage::ReadOnly).diskUsage();
+    size += Akonadi2::Storage(Akonadi2::storageLocation(), instanceIdentifier + ".changereplay", Akonadi2::Storage::ReadOnly).diskUsage();
+    return size;
+}
+
 void GenericResource::onProcessorError(int errorCode, const QString &errorMessage)
 {
     Warning() << "Received error from Processor: " << errorCode << errorMessage;
diff --git a/common/genericresource.h b/common/genericresource.h
index 97251a9..f756273 100644
--- a/common/genericresource.h
+++ b/common/genericresource.h
@@ -49,6 +49,7 @@ public:
     int error() const;
 
     static void removeFromDisk(const QByteArray &instanceIdentifier);
+    static qint64 diskUsage(const QByteArray &instanceIdentifier);
 private Q_SLOTS:
     void updateLowerBoundRevision();
 
diff --git a/tests/dummyresourcebenchmark.cpp b/tests/dummyresourcebenchmark.cpp
index ae15b9c..25fb96b 100644
--- a/tests/dummyresourcebenchmark.cpp
+++ b/tests/dummyresourcebenchmark.cpp
@@ -106,6 +106,9 @@ private Q_SLOTS:
         row.setValue("total", (qreal)num/allProcessedTime);
         dataset.insertRow(row);
         HAWD::Formatter::print(dataset);
+
+        auto diskUsage = DummyResource::diskUsage("org.kde.dummy.instance1");
+        qDebug() << "Database size [kb]: " << diskUsage/1024;
     }
 
     void testQueryByUid()