1 files changed, 77 insertions, 0 deletions

diff --git a/framework/src/entitycache.h b/framework/src/entitycache.h
new file mode 100644
index 00000000..485bcbdd
--- /dev/null
+++ b/framework/src/entitycache.h
@@ -0,0 +1,77 @@
+/*
+    Copyright (c) 2018 Christian Mollekopf <mollekopf@kolabsys.com>
+
+    This library is free software; you can redistribute it and/or modify it
+    under the terms of the GNU Library General Public License as published by
+    the Free Software Foundation; either version 2 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 Library General Public
+    License for more details.
+
+    You should have received a copy of the GNU Library General Public License
+    along with this library; see the file COPYING.LIB.  If not, write to the
+    Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+    02110-1301, USA.
+*/
+#pragma once
+
+#include "kube_export.h"
+#include <sink/query.h>
+#include <sink/store.h>
+#include <QSharedPointer>
+#include <QAbstractItemModel>
+
+class KUBE_EXPORT EntityCacheInterface
+{
+public:
+    typedef QSharedPointer<EntityCacheInterface> Ptr;
+    EntityCacheInterface() = default;
+    virtual ~EntityCacheInterface() = default;
+
+    virtual QVariant getProperty(const QByteArray &identifier, const QByteArray &property) const = 0;
+};
+
+template<typename DomainType, typename Property>
+class KUBE_EXPORT EntityCache : public EntityCacheInterface
+{
+public:
+    typedef QSharedPointer<EntityCache> Ptr;
+
+    EntityCache();
+    virtual ~EntityCache() = default;
+
+    virtual QVariant getProperty(const QByteArray &, const QByteArray &) const override;
+
+private:
+    QHash<QByteArray, typename DomainType::Ptr> mCache;
+    QSharedPointer<QAbstractItemModel> mModel;
+};
+
+template<typename DomainType, typename Property>
+EntityCache<DomainType, Property>::EntityCache()
+    : EntityCacheInterface()
+{
+    Sink::Query query;
+    query.request<Property>();
+    query.setFlags(Sink::Query::LiveQuery);
+    mModel = Sink::Store::loadModel<DomainType>(query);
+    QObject::connect(mModel.data(), &QAbstractItemModel::rowsInserted, mModel.data(), [this] (const QModelIndex &, int start, int end) {
+        for (int row = start; row <= end; row++) {
+            auto entity = mModel->index(row, 0, QModelIndex()).data(Sink::Store::DomainObjectRole).template value<typename DomainType::Ptr>();
+            mCache.insert(entity->identifier(), entity);
+        }
+    });
+}
+
+template<typename DomainType, typename Property>
+QVariant EntityCache<DomainType, Property>::getProperty(const QByteArray &identifier, const QByteArray &property) const
+{
+    if (auto entity = mCache.value(identifier)) {
+        return entity->getProperty(property);
+    }
+    return {};
+}
+