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
122
|
/*
* Copyright (C) 2014 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.
*/
#include "entitystorage.h"
static void scan(const Akonadi2::Storage::Transaction &transaction, const QByteArray &key, std::function<bool(const QByteArray &key, const Akonadi2::Entity &entity)> callback, const QByteArray &bufferType)
{
transaction.openDatabase(bufferType + ".main").scan(key, [=](const QByteArray &key, const QByteArray &value) -> bool {
//Skip internals
if (Akonadi2::Storage::isInternalKey(key)) {
return true;
}
//Extract buffers
Akonadi2::EntityBuffer buffer(value.data(), value.size());
//FIXME implement buffer.isValid()
// const auto resourceBuffer = Akonadi2::EntityBuffer::readBuffer<DummyEvent>(buffer.entity().resource());
// const auto localBuffer = Akonadi2::EntityBuffer::readBuffer<Akonadi2::ApplicationDomain::Buffer::Event>(buffer.entity().local());
// const auto metadataBuffer = Akonadi2::EntityBuffer::readBuffer<Akonadi2::Metadata>(buffer.entity().metadata());
// if ((!resourceBuffer && !localBuffer) || !metadataBuffer) {
// qWarning() << "invalid buffer " << QByteArray::fromRawData(static_cast<char*>(keyValue), keySize);
// return true;
// }
return callback(key, buffer.entity());
},
[](const Akonadi2::Storage::Error &error) {
qWarning() << "Error during query: " << error.message;
});
}
void EntityStorageBase::readValue(const Akonadi2::Storage::Transaction &transaction, const QByteArray &key, const std::function<void(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &)> &resultCallback)
{
scan(transaction, key, [=](const QByteArray &key, const Akonadi2::Entity &entity) {
const auto metadataBuffer = Akonadi2::EntityBuffer::readBuffer<Akonadi2::Metadata>(entity.metadata());
qint64 revision = metadataBuffer ? metadataBuffer->revision() : -1;
//This only works for a 1:1 mapping of resource to domain types.
//Not i.e. for tags that are stored as flags in each entity of an imap store.
//additional properties that don't have a 1:1 mapping (such as separately stored tags),
//could be added to the adaptor
auto domainObject = create(key, revision, mDomainTypeAdaptorFactory->createAdaptor(entity));
resultCallback(domainObject);
return true;
}, mBufferType);
}
static ResultSet fullScan(const Akonadi2::Storage::Transaction &transaction, const QByteArray &bufferType)
{
//TODO use a result set with an iterator, to read values on demand
QVector<QByteArray> keys;
scan(transaction, QByteArray(), [=, &keys](const QByteArray &key, const Akonadi2::Entity &) {
keys << key;
return true;
}, bufferType);
Trace() << "Full scan found " << keys.size() << " results";
return ResultSet(keys);
}
ResultSet EntityStorageBase::filteredSet(const ResultSet &resultSet, const std::function<bool(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject)> &filter, const Akonadi2::Storage::Transaction &transaction, qint64 baseRevision, qint64 topRevision)
{
auto resultSetPtr = QSharedPointer<ResultSet>::create(resultSet);
//Read through the source values and return whatever matches the filter
std::function<bool(std::function<void(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &)>)> generator = [this, resultSetPtr, &transaction, filter](std::function<void(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &)> callback) -> bool {
while (resultSetPtr->next()) {
readValue(transaction, resultSetPtr->id(), [this, filter, callback](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject) {
if (filter(domainObject)) {
callback(domainObject);
}
});
}
return false;
};
return ResultSet(generator);
}
ResultSet EntityStorageBase::getResultSet(const Akonadi2::Query &query, Akonadi2::Storage::Transaction &transaction, qint64 baseRevision, qint64 topRevision)
{
QSet<QByteArray> appliedFilters;
ResultSet resultSet = queryIndexes(query, mResourceInstanceIdentifier, appliedFilters, transaction);
const auto remainingFilters = query.propertyFilter.keys().toSet() - appliedFilters;
//We do a full scan if there were no indexes available to create the initial set.
if (appliedFilters.isEmpty()) {
resultSet = fullScan(transaction, mBufferType);
}
auto filter = [remainingFilters, query, baseRevision, topRevision](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject) -> bool {
if (topRevision > 0) {
Trace() << "filtering by revision " << domainObject->revision();
if (domainObject->revision() < baseRevision || domainObject->revision() > topRevision) {
return false;
}
}
for (const auto &filterProperty : remainingFilters) {
//TODO implement other comparison operators than equality
if (domainObject->getProperty(filterProperty) != query.propertyFilter.value(filterProperty)) {
return false;
}
}
return true;
};
return filteredSet(resultSet, filter, transaction, baseRevision, topRevision);
}
|
