summaryrefslogtreecommitdiffstats
path: root/common/datastorequery.h
blob: cc501e67e6ee7ca1640c851c7b5af7d8d6c4265c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
 *   Copyright (C) 2016 Christian Mollekopf <chrigi_1@fastmail.fm>
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or
 *   (at your option) any later version.
 *
 *   This program 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 General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the
 *   Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
 */
#pragma once

#include "query.h"
#include "resultset.h"
#include "log.h"
#include "storage/entitystore.h"


class Source;
class Bloom;
class Reduce;
class Filter;
class FilterBase;

class DataStoreQuery {
    friend class FilterBase;
    friend class Source;
    friend class Bloom;
    friend class Reduce;
    friend class Filter;
public:
    typedef QSharedPointer<DataStoreQuery> Ptr;

    struct State {
        typedef QSharedPointer<State> Ptr;
        QSharedPointer<FilterBase> mCollector;
        QSharedPointer<Source> mSource;
    };

    DataStoreQuery(const Sink::QueryBase &query, const QByteArray &type, Sink::Storage::EntityStore &store);
    DataStoreQuery(const DataStoreQuery::State &state, const QByteArray &type, Sink::Storage::EntityStore &store, bool incremental);
    ~DataStoreQuery();
    ResultSet execute();
    ResultSet update(qint64 baseRevision);
    void updateComplete();

    State::Ptr getState();

private:

    typedef std::function<bool(const Sink::ApplicationDomain::ApplicationDomainType &entity, Sink::Operation)> FilterFunction;
    typedef std::function<void(const Sink::ApplicationDomain::ApplicationDomainType &entity, Sink::Operation)> BufferCallback;

    QVector<QByteArray> indexLookup(const QByteArray &property, const QVariant &value);

    void readEntity(const QByteArray &key, const BufferCallback &resultCallback);

    ResultSet createFilteredSet(ResultSet &resultSet, const FilterFunction &);
    QVector<QByteArray> loadIncrementalResultSet(qint64 baseRevision);

    void setupQuery(const Sink::QueryBase &query_);
    QByteArrayList executeSubquery(const Sink::QueryBase &subquery);

    const QByteArray mType;
    QSharedPointer<FilterBase> mCollector;
    QSharedPointer<Source> mSource;

    Sink::Storage::EntityStore &mStore;
    Sink::Log::Context mLogCtx;
};


class FilterBase {
public:
    typedef QSharedPointer<FilterBase> Ptr;
    FilterBase(DataStoreQuery *store)
        : mDatastore(store)
    {

    }

    FilterBase(FilterBase::Ptr source, DataStoreQuery *store)
        : mSource(source),
        mDatastore(store)
    {
    }

    virtual ~FilterBase(){}

    void readEntity(const QByteArray &key, const std::function<void(const Sink::ApplicationDomain::ApplicationDomainType &entity, Sink::Operation)> &callback)
    {
        Q_ASSERT(mDatastore);
        mDatastore->readEntity(key, callback);
    }

    QVector<QByteArray> indexLookup(const QByteArray &property, const QVariant &value)
    {
        Q_ASSERT(mDatastore);
        return mDatastore->indexLookup(property, value);
    }

    virtual void skip() { mSource->skip(); }

    //Returns true for as long as a result is available
    virtual bool next(const std::function<void(const ResultSet::Result &)> &callback) = 0;

    virtual void updateComplete() { }

    FilterBase::Ptr mSource;
    DataStoreQuery *mDatastore;
    bool mIncremental = false;
};