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
123
124
125
|
/*
* Copyright (C) 2015 Christian Mollekopf <chrigi_1@fastmail.fm>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) version 3, or any
* later version accepted by the membership of KDE e.V. (or its
* successor approved by the membership of KDE e.V.), which shall
* act as a proxy defined in Section 6 of version 3 of the license.
*
* 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
*/
#include "resourcecontrol.h"
#include <QTime>
#include <QUuid>
#include <functional>
#include "resourceaccess.h"
#include "commands.h"
#include "log.h"
#include "notifier.h"
#undef DEBUG_AREA
#define DEBUG_AREA "client.resourcecontrol"
namespace Sink
{
KAsync::Job<void> ResourceControl::shutdown(const QByteArray &identifier)
{
Trace() << "shutdown " << identifier;
auto time = QSharedPointer<QTime>::create();
time->start();
return ResourceAccess::connectToServer(identifier).then<void, QSharedPointer<QLocalSocket>>([identifier, time](QSharedPointer<QLocalSocket> socket, KAsync::Future<void> &future) {
//We can't currently reuse the socket
socket->close();
auto resourceAccess = ResourceAccessFactory::instance().getAccess(identifier);
resourceAccess->open();
resourceAccess->sendCommand(Sink::Commands::ShutdownCommand).then<void>([&future, resourceAccess, time]() {
Trace() << "Shutdown complete." << Log::TraceTime(time->elapsed());
future.setFinished();
}).exec();
},
[](int, const QString &) {
Trace() << "Resource is already closed.";
//Resource isn't started, nothing to shutdown
});
}
KAsync::Job<void> ResourceControl::start(const QByteArray &identifier)
{
Trace() << "start " << identifier;
auto time = QSharedPointer<QTime>::create();
time->start();
auto resourceAccess = ResourceAccessFactory::instance().getAccess(identifier);
resourceAccess->open();
return resourceAccess->sendCommand(Sink::Commands::PingCommand).then<void>([resourceAccess, time]() {
Trace() << "Start complete." << Log::TraceTime(time->elapsed());
});
}
KAsync::Job<void> ResourceControl::flushMessageQueue(const QByteArrayList &resourceIdentifier)
{
Trace() << "flushMessageQueue" << resourceIdentifier;
return KAsync::iterate(resourceIdentifier)
.template each<void, QByteArray>([](const QByteArray &resource, KAsync::Future<void> &future) {
Trace() << "Flushing message queue " << resource;
auto resourceAccess = ResourceAccessFactory::instance().getAccess(resource);
resourceAccess->open();
resourceAccess->synchronizeResource(false, true).then<void>([&future, resourceAccess]() {
future.setFinished();
}).exec();
});
}
KAsync::Job<void> ResourceControl::flushReplayQueue(const QByteArrayList &resourceIdentifier)
{
return flushMessageQueue(resourceIdentifier);
}
template <class DomainType>
KAsync::Job<void> ResourceControl::inspect(const Inspection &inspectionCommand)
{
auto resource = inspectionCommand.resourceIdentifier;
auto time = QSharedPointer<QTime>::create();
time->start();
Trace() << "Sending inspection " << resource;
auto resourceAccess = ResourceAccessFactory::instance().getAccess(resource);
resourceAccess->open();
auto notifier = QSharedPointer<Sink::Notifier>::create(resourceAccess);
auto id = QUuid::createUuid().toByteArray();
return resourceAccess->sendInspectionCommand(id, ApplicationDomain::getTypeName<DomainType>(), inspectionCommand.entityIdentifier, inspectionCommand.property, inspectionCommand.expectedValue)
.template then<void>([resourceAccess, notifier, id, time](KAsync::Future<void> &future) {
notifier->registerHandler([&future, id, time](const Notification ¬ification) {
if (notification.id == id) {
Trace() << "Inspection complete." << Log::TraceTime(time->elapsed());
if (notification.code) {
future.setError(-1, "Inspection returned an error: " + notification.message);
} else {
future.setFinished();
}
}
});
});
}
#define REGISTER_TYPE(T) template KAsync::Job<void> ResourceControl::inspect<T>(const Inspection &); \
REGISTER_TYPE(ApplicationDomain::Event);
REGISTER_TYPE(ApplicationDomain::Mail);
REGISTER_TYPE(ApplicationDomain::Folder);
REGISTER_TYPE(ApplicationDomain::SinkResource);
} // namespace Sink
|
