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
|
/*
SPDX-FileCopyrightText: 2010 Stephen Kelly <steveire@gmail.com>
SPDX-License-Identifier: LGPL-2.0-or-later
*/
#include "entitytreecreatejob.h"
#include <Akonadi/CollectionCreateJob>
#include <Akonadi/ItemCreateJob>
#include <QVariant>
#include "migration_debug.h"
using namespace Akonadi;
static const char collectionIdMappingProperty[] = "collectionIdMappingProperty";
EntityTreeCreateJob::EntityTreeCreateJob(const QList<Akonadi::Collection::List> &collections, const Akonadi::Item::List &items, QObject *parent)
: Akonadi::TransactionSequence(parent)
, m_collections(collections)
, m_items(items)
{
}
void EntityTreeCreateJob::doStart()
{
if (!m_collections.isEmpty()) {
createNextLevelOfCollections();
}
createReadyItems();
}
void EntityTreeCreateJob::createNextLevelOfCollections()
{
CollectionCreateJob *job = nullptr;
const Collection::List colList = m_collections.takeFirst();
for (const Collection &collection : colList) {
++m_pendingJobs;
job = new CollectionCreateJob(collection, this);
job->setProperty(collectionIdMappingProperty, collection.id());
connect(job, &CollectionCreateJob::result, this, &EntityTreeCreateJob::collectionCreateJobDone);
}
}
void EntityTreeCreateJob::createReadyItems()
{
Item::List::iterator it;
for (it = m_items.begin(); it != m_items.end();) {
Collection parentCollection = (*it).parentCollection();
if (parentCollection.isValid()) {
(void)new ItemCreateJob(*it, parentCollection, this);
it = m_items.erase(it);
} else {
++it;
}
}
if (m_items.isEmpty() && m_collections.isEmpty()) {
commit();
}
}
void EntityTreeCreateJob::collectionCreateJobDone(KJob *job)
{
Q_ASSERT(m_pendingJobs > 0);
--m_pendingJobs;
auto collectionCreateJob = qobject_cast<CollectionCreateJob *>(job);
Collection createdCollection = collectionCreateJob->collection();
if (job->error()) {
qCDebug(MIGRATION_LOG) << job->errorString();
return;
}
const Collection::Id creationId = job->property(collectionIdMappingProperty).toLongLong();
Item::List::iterator it;
const Item::List::iterator end = m_items.end();
for (it = m_items.begin(); it != end; ++it) {
qCDebug(MIGRATION_LOG) << "updating items";
if (it->parentCollection().id() == creationId) {
it->setParentCollection(createdCollection);
}
}
createReadyItems();
if (!m_collections.isEmpty()) {
Collection::List::iterator col_it;
const Collection::List::iterator col_end = m_collections[0].end();
for (col_it = m_collections[0].begin(); col_it != col_end; ++col_it) {
if (col_it->parentCollection().id() == creationId) {
col_it->setParentCollection(createdCollection);
}
}
if (m_pendingJobs == 0) {
createNextLevelOfCollections();
}
}
if (m_items.isEmpty() && m_collections.isEmpty()) {
commit();
}
}
#include "moc_entitytreecreatejob.cpp"
|
