Application.cpp 32.1 KB
Newer Older
legoc's avatar
legoc committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
/*
 * Copyright 2015 Institut Laue-Langevin
 *
 * Licensed under the EUPL, Version 1.1 only (the "License");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 *
 * http://joinup.ec.europa.eu/software/page/eupl
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 */

17
#include "Application.h"
legoc's avatar
legoc committed
18
19
20
21
22
23

#include <sstream>
#include <iostream>
#include <iostream>
#include <stdexcept>
#include <vector>
24
25

#include "JSON.h"
26
#include "EventStreamSocket.h"
27
#include "impl/ServicesImpl.h"
28
29
30
31
32
33
34
#include "impl/PublisherImpl.h"
#include "impl/RequesterImpl.h"
#include "impl/RequestImpl.h"
#include "impl/ResponderImpl.h"
#include "impl/SubscriberImpl.h"
#include "impl/WaitingImpl.h"
#include "impl/WaitingImplSet.h"
35
#include "impl/HandlerImpl.h"
36
#include "impl/StreamSocketImpl.h"
legoc's avatar
legoc committed
37
#include "impl/RequestSocketImpl.h"
38
#include "message/Message.h"
39
40
41
#include "Server.h"
#include "StarterServerException.h"
#include "StatusEvent.h"
42
43
44
45
#include "PublisherEvent.h"
#include "ResultEvent.h"
#include "PortEvent.h"
#include "CancelEvent.h"
legoc's avatar
legoc committed
46
47
48
49
50
51

using namespace std;

namespace cameo {
namespace application {

52
This This::m_instance;
legoc's avatar
legoc committed
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
const std::string This::RUNNING_STATE = "RUNNING";

State This::parseState(const std::string& value) {

	if (value == "UNKNOWN") {
		return UNKNOWN;
	} else if (value == "STARTING") {
		return STARTING;
	} else if (value == "RUNNING") {
		return RUNNING;
	} else if (value == "STOPPING") {
		return STOPPING;
	} else if (value == "KILLING") {
		return KILLING;
	} else if (value == "PROCESSING_ERROR") {
		return PROCESSING_ERROR;
	} else if (value == "ERROR") {
legoc's avatar
legoc committed
70
		return FAILURE;
legoc's avatar
legoc committed
71
72
73
74
75
76
77
78
79
80
81
82
	} else if (value == "SUCCESS") {
		return SUCCESS;
	} else if (value == "STOPPED") {
		return STOPPED;
	} else if (value == "KILLED") {
		return KILLED;
	}

	return UNKNOWN;
}

void This::init(int argc, char *argv[]) {
legoc's avatar
legoc committed
83
	if (m_instance.m_impl == nullptr) {
84
		m_instance.initApplication(argc, argv);
legoc's avatar
legoc committed
85
86
87
88
	}
}

std::string This::getReference() {
legoc's avatar
legoc committed
89
	if (m_instance.m_impl != nullptr) {
legoc's avatar
legoc committed
90
		ostringstream os;
91
		os << getName() << "." << getId() << "@" << getEndpoint();
legoc's avatar
legoc committed
92
93
94
95
96
97
		return os.str();
	}
	return "";
}

void This::terminate() {
98
99

	// Test if termination is already done.
legoc's avatar
legoc committed
100
	if (m_instance.m_impl == nullptr) {
101
102
103
104
105
106
		return;
	}

	// Terminate the unmanaged application.
	if (!m_instance.m_managed) {
		m_instance.terminateUnmanagedApplication();
legoc's avatar
legoc committed
107
108
	}

109
110
111
112
	// Terminate the services.
	m_instance.Services::terminate();

	// Ensure that it won't be done twice.
legoc's avatar
legoc committed
113
	m_instance.m_impl = nullptr;
legoc's avatar
legoc committed
114
115
}

116
117

This::This() :
legoc's avatar
legoc committed
118
	Services(),
legoc's avatar
legoc committed
119
120
121
	m_id(-1),
	m_managed(false),
	m_starterId(0) {
122
123
124
}

void This::initApplication(int argc, char *argv[]) {
legoc's avatar
legoc committed
125

126
	Services::init();
legoc's avatar
legoc committed
127
128

	if (argc == 0) {
129
		throw InvalidArgumentException("missing info argument");
legoc's avatar
legoc committed
130
131
132
133
134
135
	}

	string info(argv[argc - 1]);
	vector<string> tokens = split(info);

	if (tokens.size() < 4) {
136
		throw InvalidArgumentException(info + " is not a valid argument");
legoc's avatar
legoc committed
137
138
139
140
141
142
143
144
	}

	m_url = tokens[0] + ":" + tokens[1];

	string port = tokens[2];
	istringstream is(port);
	is >> m_port;

145
146
147
	// We separated host endpoint and server in the past (server being tcp://localhost)
	// but that generates troubles when two applications communicate remotely.
	// However leave the same value seems to be ok.
legoc's avatar
legoc committed
148
149
	m_serverEndpoint = m_url + ":" + port;

150
151
152
	// Create the request socket. The server endpoint has been defined.
	Services::initRequestSocket();

153
154
155
	// Retrieve the server version.
	Services::retrieveServerVersion();

legoc's avatar
legoc committed
156
157
158
	string nameId = tokens[3];

	int index = nameId.find_last_of('.');
legoc's avatar
legoc committed
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173

	// Search for the . character meaning that the application is managed and already has an id.
	if (index != string::npos) {
		m_managed = true;
		m_name = nameId.substr(0, index);
		string sid = nameId.substr(index + 1);
		{
			istringstream is(sid);
			is >> m_id;
		}
	}
	else {
		m_managed = false;
		m_name = nameId;
		m_id = initUnmanagedApplication();
174
175
176
177

		if (m_id == -1) {
			throw UnmanagedApplicationException(string("Maximum number of applications ") + m_name + " reached");
		}
legoc's avatar
legoc committed
178
179
180
181
182
183
184
185
	}

	if (tokens.size() >= 7) {
		index = tokens[4].find_last_of('@');
		m_starterEndpoint = tokens[4].substr(index + 1) + ":" + tokens[5] + ":" + tokens[6];
		string starterNameId = tokens[4].substr(0, index);
		index = starterNameId.find_last_of('.');
		m_starterName = starterNameId.substr(0, index);
legoc's avatar
legoc committed
186
		string sid = starterNameId.substr(index + 1);
legoc's avatar
legoc committed
187
188
189
190
191
192
		{
			istringstream is(sid);
			is >> m_starterId;
		}
	}

legoc's avatar
legoc committed
193
194
	// Must be here because the server endpoint is required.
	initStatus();
legoc's avatar
legoc committed
195
196

	// Create the local server
legoc's avatar
legoc committed
197
	m_server = unique_ptr<Server>(new Server(m_serverEndpoint));
legoc's avatar
legoc committed
198
199
200

	// Create the starter server
	if (m_starterEndpoint != "") {
legoc's avatar
legoc committed
201
		m_starterServer = unique_ptr<Server>(new Server(m_starterEndpoint));
legoc's avatar
legoc committed
202
203
	}

legoc's avatar
legoc committed
204
	m_waitingSet = unique_ptr<WaitingImplSet>(new WaitingImplSet());
205
206
207
208

	// Init listener.
	setName(m_name);
	m_server->registerEventListener(this);
legoc's avatar
legoc committed
209
210
211
212
}

This::~This() {
	// Do not delete the impl here because there will be order trouble.
213
214

	// Terminate the unmanaged application.
legoc's avatar
legoc committed
215
	if (m_impl != nullptr && !m_managed) {
216
217
		terminateUnmanagedApplication();
	}
legoc's avatar
legoc committed
218
219
220
}

const std::string& This::getName() {
221
	return m_instance.m_name;
legoc's avatar
legoc committed
222
223
224
}

int This::getId() {
225
	return m_instance.m_id;
legoc's avatar
legoc committed
226
227
228
}

void This::setTimeout(int timeout) {
229
	m_instance.Services::setTimeout(timeout);
legoc's avatar
legoc committed
230
231
232
}

int This::getTimeout() {
233
	return m_instance.Services::getTimeout();
legoc's avatar
legoc committed
234
235
}

236
const std::string& This::getEndpoint() {
legoc's avatar
legoc committed
237
	if (m_instance.m_impl != nullptr) {
238
		return m_instance.m_serverEndpoint;
legoc's avatar
legoc committed
239
240
241
242
243
244
	}
	static string result;
	return result;
}

Server& This::getServer() {
245
	return *m_instance.m_server;
legoc's avatar
legoc committed
246
247
248
249
}

Server& This::getStarterServer() {

legoc's avatar
legoc committed
250
	if (m_instance.m_starterServer.get() == nullptr) {
legoc's avatar
legoc committed
251
252
253
		throw StarterServerException();
	}

254
	return *m_instance.m_starterServer;
legoc's avatar
legoc committed
255
256
257
}

const std::string& This::getUrl() {
258
	return m_instance.Services::getUrl();
legoc's avatar
legoc committed
259
260
261
}

bool This::isAvailable(int timeout) {
262
	return m_instance.Services::isAvailable(timeout);
legoc's avatar
legoc committed
263
264
265
}

bool This::isStopping() {
266
	return m_instance.getState(m_instance.m_id) == STOPPING;
legoc's avatar
legoc committed
267
268
269
}

void This::cancelWaitings() {
270
	m_instance.m_waitingSet->cancelAll();
legoc's avatar
legoc committed
271
272
}

legoc's avatar
legoc committed
273
int This::initUnmanagedApplication() {
legoc's avatar
legoc committed
274

275
	unique_ptr<zmq::message_t> reply = m_requestSocket->request(m_impl->createStartedUnmanagedRequest(m_name));
legoc's avatar
legoc committed
276

277
278
279
	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());
legoc's avatar
legoc committed
280

281
	return response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
282
283
284
285
}

void This::terminateUnmanagedApplication() {

286
	m_requestSocket->request(m_impl->createTerminatedUnmanagedRequest(m_id));
legoc's avatar
legoc committed
287
288
}

289
bool This::setRunning() {
legoc's avatar
legoc committed
290

291
	unique_ptr<zmq::message_t> reply = m_instance.m_requestSocket->request(m_instance.m_impl->createSetStatusRequest(m_instance.m_id, RUNNING));
legoc's avatar
legoc committed
292

293
294
295
	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());
legoc's avatar
legoc committed
296

297
298
	int value = response[message::RequestResponse::VALUE].GetInt();
	if (value == -1) {
299
		return false;
legoc's avatar
legoc committed
300
	}
301
302

	return true;
legoc's avatar
legoc committed
303
304
305
306
}

void This::setBinaryResult(const std::string& data) {

307
	m_instance.m_requestSocket->request(m_instance.m_impl->createSetResultRequest(m_instance.m_id), data);
legoc's avatar
legoc committed
308
309
310
311
312
313
314
315
316
317
318
}

void This::setResult(const std::string& data) {

	string resultData;
	serialize(data, resultData);
	setBinaryResult(resultData);
}

State This::getState(int id) const {

319
	unique_ptr<zmq::message_t> reply = m_requestSocket->request(m_impl->createGetStatusRequest(id));
legoc's avatar
legoc committed
320

321
322
323
	// Get the JSON response.
	json::Object event;
	json::parse(event, reply.get());
legoc's avatar
legoc committed
324

325
	return event[message::StatusEvent::APPLICATION_STATE].GetInt();
legoc's avatar
legoc committed
326
327
328
329
}

bool This::destroyPublisher(const std::string& name) const {

330
331
332
333
334
	unique_ptr<zmq::message_t> reply = m_requestSocket->request(m_impl->createTerminatePublisherRequest(m_id, name));

	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());
legoc's avatar
legoc committed
335

336
	int value = response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
337
338
339
340
341
342

	return (value != -1);
}

bool This::removePort(const std::string& name) const {

343
	unique_ptr<zmq::message_t> reply = m_requestSocket->request(m_impl->createRemovePortRequest(m_id, name));
legoc's avatar
legoc committed
344

345
346
347
348
349
	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());

	int value = response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364

	return (value != -1);
}

State This::waitForStop() {

	// test if stop was requested elsewhere
	State state = getState(m_id);
	if (state == STOPPING
		|| state == KILLING) {
		return state;
	}

	while (true) {
		// waits for a new incoming status
365
		unique_ptr<Event> event = popEvent();
legoc's avatar
legoc committed
366
367

		// The socket is canceled.
legoc's avatar
legoc committed
368
		if (event.get() == nullptr) {
legoc's avatar
legoc committed
369
370
371
372
373
374
			return UNKNOWN;
		}

		if (event->getId() == m_id) {
			StatusEvent * status = dynamic_cast<StatusEvent *>(event.get());

legoc's avatar
legoc committed
375
			if (status != nullptr) {
legoc's avatar
legoc committed
376
377
378
379
380
381
382
383
384
385
386
387
388
				state = status->getState();

				if (state == STOPPING
					|| state == KILLING) {
					return state;
				}
			}
		}
	}

	return UNKNOWN;
}

legoc's avatar
legoc committed
389
std::unique_ptr<Instance> This::connectToStarter() {
legoc's avatar
legoc committed
390

legoc's avatar
legoc committed
391
392
	if (m_instance.m_starterServer.get() == nullptr) {
		return unique_ptr<Instance>(nullptr);
legoc's avatar
legoc committed
393
394
395
	}

	// Iterate the instances to find the id
396
	InstanceArray instances = m_instance.m_starterServer->connectAll(m_instance.m_starterName);
legoc's avatar
legoc committed
397
398

	for (int i = 0; i < instances.size(); i++) {
399
		if (instances[i]->getId() == m_instance.m_starterId) {
legoc's avatar
legoc committed
400
			return unique_ptr<Instance>(std::move(instances[i]));
legoc's avatar
legoc committed
401
402
403
		}
	}

legoc's avatar
legoc committed
404
	return unique_ptr<Instance>(nullptr);
legoc's avatar
legoc committed
405
406
}

407
408
409
410
411
412
413
414
415
416
417
418
void This::storeKeyValue(const std::string& key, const std::string& value) {
	m_instance.m_server->storeKeyValue(m_instance.m_id, key, value);
}

std::string This::getKeyValue(const std::string& key) {
	return m_instance.m_server->getKeyValue(m_instance.m_id, key);
}

void This::removeKey(const std::string& key) {
	m_instance.m_server->removeKey(m_instance.m_id, key);
}

419
420
421
422
423
424
425
426
427
428
void This::stoppingFunction(StopFunctionType stop) {

	application::State state = waitForStop();

	// Only stop in case of STOPPING.
	if (state == application::STOPPING) {
		stop();
	}
}

legoc's avatar
legoc committed
429
void This::handleStopImpl(StopFunctionType function) {
430
	m_stopHandler = unique_ptr<HandlerImpl>(new HandlerImpl(bind(&This::stoppingFunction, this, function)));
legoc's avatar
legoc committed
431
432
433
434
435
}

///////////////////////////////////////////////////////////////////////////////
// Instance

436
Instance::Instance(Server * server) :
legoc's avatar
legoc committed
437
438
439
440
441
	m_server(server),
	m_id(-1),
	m_pastStates(0),
	m_initialState(UNKNOWN),
	m_lastState(UNKNOWN),
442
443
	m_hasResult(false),
	m_exitCode(-1) {
444
445

	m_waiting.reset(new GenericWaitingImpl(bind(&Instance::cancelWaitFor, this)));
legoc's avatar
legoc committed
446
447
448
}

Instance::~Instance() {
449
450
451
452
	// Unregister the instance.
	m_server->unregisterEventListener(this);

	// The destructor has been added to avoid blocking ZeroMQ, because the inner objects destructors were not called.
legoc's avatar
legoc committed
453
454
455
456
457
458
}

void Instance::setId(int id) {
	m_id = id;
}

legoc's avatar
legoc committed
459
460
461
462
const std::string& Instance::getName() const {
	return EventListener::m_name;
}

legoc's avatar
legoc committed
463
464
465
466
void Instance::setErrorMessage(const std::string& message) {
	m_errorMessage = message;
}

legoc's avatar
legoc committed
467
468
void Instance::setOutputStreamSocket(std::unique_ptr<OutputStreamSocket>& socket) {
	m_outputStreamSocket = std::move(socket);
469
	m_outputStreamSocket->setApplicationId(m_id);
legoc's avatar
legoc committed
470
471
}

legoc's avatar
legoc committed
472
473
474
475
476
477
void Instance::setPastStates(State pastStates) {
	m_pastStates = pastStates;
}

void Instance::setInitialState(State state) {
	m_initialState = state;
478
479
480

	// It is important to set the last state, because in case of a call to the function now without any incoming state.
	m_lastState = state;
legoc's avatar
legoc committed
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
}

int Instance::getId() const {
	return m_id;
}

const std::string& Instance::getUrl() const {
	return m_server->getUrl();
}

const std::string& Instance::getEndpoint() const {
	return m_server->getEndpoint();
}

std::string Instance::getNameId() const {
	stringstream os;
	os << m_name << "." << m_id;

	return os.str();
}

bool Instance::hasResult() const {
	return m_hasResult;
}

bool Instance::exists() const {
	return (m_id != -1);
}

const std::string& Instance::getErrorMessage() const {
	return m_errorMessage;
}

bool Instance::stop() {
	try {
		Response response = m_server->stopApplicationAsynchronously(m_id, false);

	} catch (const ConnectionTimeout& e) {
		m_errorMessage = e.what();
		return false;
	}

	return true;
}

bool Instance::kill() {
	try {
		Response response = m_server->stopApplicationAsynchronously(m_id, true);

	} catch (const ConnectionTimeout& e) {
		m_errorMessage = e.what();
		return false;
	}

	return true;
}

538
State Instance::waitFor(int states, const std::string& eventName, StateHandlerType handler, bool blocking) {
legoc's avatar
legoc committed
539
540

	if (!exists()) {
541
		// The application was not launched
legoc's avatar
legoc committed
542
543
544
		return m_lastState;
	}

545
	// Test the terminal state
legoc's avatar
legoc committed
546
547
548
	if (m_lastState == SUCCESS
		|| m_lastState == STOPPED
		|| m_lastState == KILLED
legoc's avatar
legoc committed
549
		|| m_lastState == FAILURE) {
550
		// The application is already terminated
legoc's avatar
legoc committed
551
552
553
		return m_lastState;
	}

554
	// Test the requested states
legoc's avatar
legoc committed
555
	if ((states & m_pastStates) != 0) {
556
		// The state is already received
legoc's avatar
legoc committed
557
558
559
560
		return m_lastState;
	}

	while (true) {
561
		// Waits for a new incoming status
562
		unique_ptr<Event> event = popEvent(blocking);
legoc's avatar
legoc committed
563

564
		// The non-blocking call returns a null message.
legoc's avatar
legoc committed
565
		if (event.get() == nullptr) {
legoc's avatar
legoc committed
566
567
568
569
570
571
			return m_lastState;
		}

		if (event->getId() == m_id) {
			StatusEvent * status = dynamic_cast<StatusEvent *>(event.get());

legoc's avatar
legoc committed
572
			if (status != nullptr) {
legoc's avatar
legoc committed
573
574
575
576
				State state = status->getState();
				m_pastStates = status->getPastStates();
				m_lastState = state;

577
578
579
580
581
				// Assign the exit code.
				if (status->getExitCode() != -1) {
					m_exitCode = status->getExitCode();
				}

582
				// Call the state handler.
legoc's avatar
legoc committed
583
				if (handler != nullptr) {
legoc's avatar
legoc committed
584
585
586
					handler(state);
				}

legoc's avatar
legoc committed
587
588
589
590
				// test the terminal state
				if (state == SUCCESS
					|| state == STOPPED
					|| state == KILLED
legoc's avatar
legoc committed
591
					|| state == FAILURE) {
legoc's avatar
legoc committed
592
593
594
595
596
597
598
					break;
				}

				// test the requested states
				if ((states & m_pastStates) != 0) {
					return m_lastState;
				}
599
600
			}
			else {
legoc's avatar
legoc committed
601
602
603
604

				if (ResultEvent * result = dynamic_cast<ResultEvent *>(event.get())) {
					m_hasResult = true;
					m_resultData = result->getData();
605
606
				}
				else if (PublisherEvent * publisher = dynamic_cast<PublisherEvent *>(event.get())) {
legoc's avatar
legoc committed
607
608
609
					if (publisher->getPublisherName() == eventName) {
						break;
					}
610
611
				}
				else if (PortEvent * port = dynamic_cast<PortEvent *>(event.get())) {
legoc's avatar
legoc committed
612
613
614
615
					if (port->getPortName() == eventName) {
						break;
					}
				}
616
617
618
				else if (CancelEvent * cancel = dynamic_cast<CancelEvent *>(event.get())) {
					break;
				}
legoc's avatar
legoc committed
619
620
621
622
623
624
625
			}
		}
	}

	return m_lastState;
}

legoc's avatar
legoc committed
626
State Instance::waitFor(int states, StateHandlerType handler) {
627
	return waitFor(states, "", handler, true);
legoc's avatar
legoc committed
628
629
630
}

State Instance::waitFor(StateHandlerType handler) {
631
632
633
634
635
	return waitFor(0, "", handler, true);
}

State Instance::waitFor(const std::string& eventName) {
	return waitFor(0, eventName, nullptr, true);
legoc's avatar
legoc committed
636
637
638
}

void Instance::cancelWaitFor() {
639
	cancel(m_id);
legoc's avatar
legoc committed
640
641
}

642
State Instance::now() {
643
644
645
646
647
648
649

	// First implementation used getLastState().
	return getActualState();
}

State Instance::getLastState() {

legoc's avatar
legoc committed
650
	return waitFor(0, "", nullptr, false);
651
652
}

653
State Instance::getActualState() const {
654
655
	return m_server->getActualState(m_id);
}
656

657
658
std::set<State> Instance::getPastStates() const {
	return m_server->getPastStates(m_id);
659
660
}

661
662
663
664
int Instance::getExitCode() const {
	return m_exitCode;
}

legoc's avatar
legoc committed
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
bool Instance::getBinaryResult(std::string& result) {

	waitFor();
	result = m_resultData;

	return m_hasResult;
}

bool Instance::getResult(std::string& result) {

	string bytes;
	getBinaryResult(bytes);
	parse(bytes, result);

	return m_hasResult;
}

legoc's avatar
legoc committed
682
683
684
685
std::shared_ptr<OutputStreamSocket> Instance::getOutputStreamSocket() {
	return m_outputStreamSocket;
}

686
687
688
689
690
std::string Instance::getKeyValue(const std::string& key) {
	// TODO catch exceptions and rethrow an exception: TerminatedException?
	return m_server->getKeyValue(m_id, key);
}

legoc's avatar
legoc committed
691
692
693
694
695
696
697
698
699
700
///////////////////////////////////////////////////////////////////////////
// InstanceArray

InstanceArray::InstanceArray() :
	m_size(0),
	m_array(0) {
}

InstanceArray::InstanceArray(const InstanceArray& array) :
	m_size(array.m_size),
legoc's avatar
legoc committed
701
	m_array(new unique_ptr<Instance>[m_size]) {
legoc's avatar
legoc committed
702
703
704

	// transferring pointers
	for (size_t i = 0; i < m_size; i++) {
legoc's avatar
legoc committed
705
		m_array[i] = std::move(array.m_array[i]);
legoc's avatar
legoc committed
706
707
708
709
710
711
712
713
714
	}
}

InstanceArray::~InstanceArray() {
	delete [] m_array;
}

void InstanceArray::allocate(std::size_t size) {
	m_size = size;
legoc's avatar
legoc committed
715
	m_array = new unique_ptr<Instance>[size];
legoc's avatar
legoc committed
716
717
718
719
720
721
}

std::size_t InstanceArray::size() const {
	return m_size;
}

legoc's avatar
legoc committed
722
std::unique_ptr<Instance>& InstanceArray::operator[](std::size_t index) {
legoc's avatar
legoc committed
723
724
725
726
727
728
	return m_array[index];
}

///////////////////////////////////////////////////////////////////////////////
// Publisher

legoc's avatar
legoc committed
729
Publisher::Publisher(This * application, int publisherPort, int synchronizerPort, const std::string& name, int numberOfSubscribers) :
legoc's avatar
legoc committed
730
731
732
733
734
735
736
737
738
	m_impl(new PublisherImpl(application, publisherPort, synchronizerPort, name, numberOfSubscribers)) {

	// Create the waiting here.
	m_waiting.reset(m_impl->waiting());
}

Publisher::~Publisher() {
}

legoc's avatar
legoc committed
739
std::unique_ptr<Publisher> Publisher::create(const std::string& name, int numberOfSubscribers) {
legoc's avatar
legoc committed
740

741
	unique_ptr<zmq::message_t> reply = This::m_instance.m_requestSocket->request(This::m_instance.m_impl->createCreatePublisherRequest(This::m_instance.m_id, name, numberOfSubscribers));
legoc's avatar
legoc committed
742

743
744
745
	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());
legoc's avatar
legoc committed
746

747
	int publisherPort = response[message::PublisherResponse::PUBLISHER_PORT].GetInt();
legoc's avatar
legoc committed
748
	if (publisherPort == -1) {
749
		throw PublisherCreationException(response[message::PublisherResponse::MESSAGE].GetString());
legoc's avatar
legoc committed
750
	}
751
	int synchronizerPort = response[message::PublisherResponse::SYNCHRONIZER_PORT].GetInt();;
legoc's avatar
legoc committed
752

legoc's avatar
legoc committed
753
	return unique_ptr<Publisher>(new Publisher(&This::m_instance, publisherPort, synchronizerPort, name, numberOfSubscribers));
legoc's avatar
legoc committed
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
}


const std::string& Publisher::getName() const {
	return m_impl->getName();
}

const std::string& Publisher::getApplicationName() const {
	return m_impl->getApplicationName();
}

int Publisher::getApplicationId() const {
	return m_impl->getApplicationId();
}

const std::string& Publisher::getApplicationEndpoint() const {
	return m_impl->getApplicationEndpoint();
}

bool Publisher::waitForSubscribers() const {
	return m_impl->waitForSubscribers();
}

void Publisher::cancelWaitForSubscribers() {
	m_waiting->cancel();
}

void Publisher::sendBinary(const std::string& data) const {
	m_impl->sendBinary(data);
}

void Publisher::send(const std::string& data) const {
	m_impl->send(data);
}

789
790
791
792
void Publisher::sendTwoBinaryParts(const std::string& data1, const std::string& data2) const {
	m_impl->sendTwoBinaryParts(data1, data2);
}

legoc's avatar
legoc committed
793
bool Publisher::hasEnded() const {
794
795
796
797
798
	return m_impl->isEnded();
}

bool Publisher::isEnded() const {
	return m_impl->isEnded();
legoc's avatar
legoc committed
799
800
801
802
803
804
805
806
807
}

void Publisher::sendEnd() const {
	m_impl->setEnd();
}

///////////////////////////////////////////////////////////////////////////
// Subscriber

808
Subscriber::Subscriber(Server * server, const std::string & url, int publisherPort, int synchronizerPort, const std::string & publisherName, int numberOfSubscribers, const std::string& instanceName, int instanceId, const std::string& instanceEndpoint, const std::string& statusEndpoint) :
legoc's avatar
legoc committed
809
810
811
812
813
814
	m_impl(new SubscriberImpl(server, url, publisherPort, synchronizerPort, publisherName, numberOfSubscribers, instanceName, instanceId, instanceEndpoint, statusEndpoint)) {
}

Subscriber::~Subscriber() {
}

legoc's avatar
legoc committed
815
std::unique_ptr<Subscriber> Subscriber::create(Instance & instance, const std::string& publisherName) {
legoc's avatar
legoc committed
816
817
818
819
820
821
822
823
	try {
		return instance.m_server->createSubscriber(instance.m_id, publisherName, instance.m_name);

	} catch (const SubscriberCreationException& e) {
		// the publisher does not exist, so we are waiting for it
	}

	// waiting for the publisher
824
	State lastState = instance.waitFor(publisherName);
legoc's avatar
legoc committed
825
826
827
828
829

	// state cannot be terminal or it means that the application has terminated that is not planned.
	if (lastState == SUCCESS
		|| lastState == STOPPED
		|| lastState == KILLED
legoc's avatar
legoc committed
830
		|| lastState == FAILURE) {
legoc's avatar
legoc committed
831
		return unique_ptr<Subscriber>(nullptr);
legoc's avatar
legoc committed
832
833
834
835
836
837
838
839
840
	}

	try {
		return instance.m_server->createSubscriber(instance.m_id, publisherName, instance.m_name);

	} catch (const SubscriberCreationException& e) {
		// that should not happen
	}

legoc's avatar
legoc committed
841
	return unique_ptr<Subscriber>(nullptr);
legoc's avatar
legoc committed
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
}

void Subscriber::init() {
	m_impl->init();

	// Create the waiting here.
	m_waiting.reset(m_impl->waiting());
}

const std::string& Subscriber::getPublisherName() const {
	return m_impl->m_publisherName;
}

const std::string& Subscriber::getInstanceName() const {
	return m_impl->m_instanceName;
}

int Subscriber::getInstanceId() const {
	return m_impl->m_instanceId;
}

const std::string& Subscriber::getInstanceEndpoint() const {
	return m_impl->m_instanceEndpoint;
}

bool Subscriber::hasEnded() const {
868
869
870
871
872
873
874
875
876
	return m_impl->isEnded();
}

bool Subscriber::isEnded() const {
	return m_impl->isEnded();
}

bool Subscriber::isCanceled() const {
	return m_impl->isCanceled();
legoc's avatar
legoc committed
877
878
879
880
881
882
883
884
885
886
}

bool Subscriber::receiveBinary(std::string& data) const {
	return m_impl->receiveBinary(data);
}

bool Subscriber::receive(std::string& data) const {
	return m_impl->receive(data);
}

887
888
889
890
bool Subscriber::receiveTwoBinaryParts(std::string& data1, std::string& data2) const {
	return m_impl->receiveTwoBinaryParts(data1, data2);
}

legoc's avatar
legoc committed
891
892
893
894
895
896
897
void Subscriber::cancel() {
	m_waiting->cancel();
}

///////////////////////////////////////////////////////////////////////////
// Request

legoc's avatar
legoc committed
898
899
Request::Request(std::unique_ptr<RequestImpl> & impl) :
	m_impl(std::move(impl)) {
legoc's avatar
legoc committed
900
901
902
903
904
}

Request::~Request() {
}

legoc's avatar
legoc committed
905
906
907
908
909
910
911
912
913
914
915
std::string Request::getObjectId() const {

	// Local id is missing.
	return "request:"
		+ m_impl->m_requesterApplicationName
		+ "."
		+ to_string(m_impl->m_requesterApplicationId)
		+ "@"
		+ m_impl->m_requesterServerEndpoint;
}

916
917
918
919
std::string Request::getRequesterEndpoint() const {
	return m_impl->m_requesterServerEndpoint;
}

legoc's avatar
legoc committed
920
921
void Request::setTimeout(int value) {
	m_impl->setTimeout(value);
922
923
}

924
const std::string& Request::getBinary() const {
legoc's avatar
legoc committed
925
926
927
	return m_impl->m_message;
}

928
std::string Request::get() const {
legoc's avatar
legoc committed
929
930
931
932
933
934
935

	string data;
	parse(m_impl->m_message, data);

	return data;
}

936
937
938
939
const std::string& Request::getSecondBinaryPart() const {
	return m_impl->m_message2;
}

legoc's avatar
legoc committed
940
941
bool Request::replyBinary(const std::string& response) {
	return m_impl->replyBinary(response);
legoc's avatar
legoc committed
942
943
}

legoc's avatar
legoc committed
944
945
bool Request::reply(const std::string& response) {
	return m_impl->reply(response);
legoc's avatar
legoc committed
946
947
}

legoc's avatar
legoc committed
948
std::unique_ptr<Instance> Request::connectToRequester() {
949

legoc's avatar
legoc committed
950
	// Instantiate the requester server if it does not exist.
legoc's avatar
legoc committed
951
	if (m_requesterServer.get() == nullptr) {
952
		m_requesterServer.reset(new Server(m_impl->m_requesterServerEndpoint, m_impl->m_timeout));
legoc's avatar
legoc committed
953
	}
954
955
956
957
958
959

	// Connect and find the instance.
	InstanceArray instances = m_requesterServer->connectAll(m_impl->m_requesterApplicationName);

	for (int i = 0; i < instances.size(); i++) {
		if (instances[i]->getId() == m_impl->m_requesterApplicationId) {
legoc's avatar
legoc committed
960
			return unique_ptr<Instance>(std::move(instances[i]));
961
962
963
964
		}
	}

	// Not found.
legoc's avatar
legoc committed
965
	return unique_ptr<Instance>(nullptr);
966
967
}

legoc's avatar
legoc committed
968
969
std::unique_ptr<Server> Request::getServer() {
	return std::move(m_requesterServer);
legoc's avatar
legoc committed
970
971
}

legoc's avatar
legoc committed
972
973
974
///////////////////////////////////////////////////////////////////////////
// Responder

legoc's avatar
legoc committed
975
Responder::Responder(application::This * application, int responderPort, const std::string& name) :
legoc's avatar
legoc committed
976
977
978
979
980
981
982
983
984
	m_impl(new ResponderImpl(application, responderPort, name)) {

	// Create the waiting here.
	m_waiting.reset(m_impl->waiting());
}

Responder::~Responder() {
}

legoc's avatar
legoc committed
985
std::unique_ptr<Responder> Responder::create(const std::string& name) {
legoc's avatar
legoc committed
986
987
988

	string portName = ResponderImpl::RESPONDER_PREFIX + name;

989
	unique_ptr<zmq::message_t> reply = This::m_instance.m_requestSocket->request(This::m_instance.m_impl->createRequestPortRequest(This::m_instance.m_id, portName));
legoc's avatar
legoc committed
990

991
992
993
	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());
legoc's avatar
legoc committed
994

995
	int responderPort = response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
996
	if (responderPort == -1) {
997
		throw ResponderCreationException(response[message::RequestResponse::MESSAGE].GetString());
legoc's avatar
legoc committed
998
999
	}

legoc's avatar
legoc committed
1000
	return unique_ptr<Responder>(new Responder(&This::m_instance, responderPort, name));
legoc's avatar
legoc committed
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
}

const std::string& Responder::getName() const {
	return m_impl->m_name;
}

void Responder::cancel() {
	m_impl->cancel();
}

legoc's avatar
legoc committed
1011
std::unique_ptr<Request> Responder::receive() {
legoc's avatar
legoc committed
1012

legoc's avatar
legoc committed
1013
1014
1015
	unique_ptr<RequestImpl> requestImpl = m_impl->receive();
	if (requestImpl.get() == nullptr) {
		return unique_ptr<Request>(nullptr);
legoc's avatar
legoc committed
1016
	}
legoc's avatar
legoc committed
1017
	return unique_ptr<Request>(new Request(requestImpl));
legoc's avatar
legoc committed
1018
1019
}

1020
1021
bool Responder::isCanceled() const {
	return m_impl->m_canceled;
legoc's avatar
legoc committed
1022
1023
1024
1025
1026
}

///////////////////////////////////////////////////////////////////////////
// Requester

legoc's avatar
legoc committed
1027
Requester::Requester(application::This * application, const std::string& url, int requesterPort, int responderPort, const std::string& name, int responderId, int requesterId) :
1028
	m_impl(new RequesterImpl(application, url, requesterPort, responderPort, name, responderId, requesterId)) {
legoc's avatar
legoc committed
1029
1030
1031
1032
1033
1034
1035
1036

	// Create the waiting here.
	m_waiting.reset(m_impl->waiting());
}

Requester::~Requester() {
}

legoc's avatar
legoc committed
1037
std::unique_ptr<Requester> Requester::create(Instance & instance, const std::string& name) {
legoc's avatar
legoc committed
1038
1039
1040
1041
1042

	int responderId = instance.getId();
	string responderUrl = instance.getUrl();
	string responderEndpoint = instance.getEndpoint();

legoc's avatar
legoc committed
1043
1044
1045
	// Create a request socket to the server of the instance.
	unique_ptr<RequestSocketImpl> instanceRequestSocket = This::m_instance.createRequestSocket(responderEndpoint);

legoc's avatar
legoc committed
1046
	string responderPortName = ResponderImpl::RESPONDER_PREFIX + name;
1047
1048
	int requesterId = RequesterImpl::newRequesterId();
	string requesterPortName = RequesterImpl::getRequesterPortName(name, responderId, requesterId);
legoc's avatar
legoc committed
1049

1050
1051
1052
	string request = This::m_instance.m_impl->createConnectPortRequest(responderId, responderPortName);

	unique_ptr<zmq::message_t> reply = instanceRequestSocket->request(request);
legoc's avatar
legoc committed
1053

1054
1055
1056
	// Get the JSON response.
	json::Object response;
	json::parse(response, reply.get());
legoc's avatar
legoc committed
1057

1058
	reply.reset();
legoc's avatar
legoc committed
1059

1060
	int responderPort = response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
1061
1062
	if (responderPort == -1) {
		// Wait for the responder port.
1063
		instance.waitFor(responderPortName);
legoc's avatar
legoc committed
1064
1065

		// Retry to connect.
1066
1067
		reply = instanceRequestSocket->request(request);
		json::parse(response, reply.get());
legoc's avatar
legoc committed
1068

1069
		responderPort = response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
1070
		if (responderPort == -1) {
1071
			throw RequesterCreationException(response[message::RequestResponse::MESSAGE].GetString());
legoc's avatar
legoc committed
1072
		}
1073
1074

		reply.reset();
legoc's avatar
legoc committed
1075
1076
	}

1077
	// Request a requester port.
1078
1079
	reply = This::m_instance.m_requestSocket->request(This::m_instance.m_impl->createRequestPortRequest(This::m_instance.m_id, requesterPortName));
	json::parse(response, reply.get());
legoc's avatar
legoc committed
1080

1081
	int requesterPort = response[message::RequestResponse::VALUE].GetInt();
legoc's avatar
legoc committed
1082
	if (requesterPort == -1) {
1083
		throw RequesterCreationException(response[message::RequestResponse::MESSAGE].GetString());
legoc's avatar
legoc committed
1084
1085
	}

legoc's avatar
legoc committed
1086
	return unique_ptr<Requester>(new Requester(&This::m_instance, responderUrl, requesterPort, responderPort, name, responderId, requesterId));
legoc's avatar
legoc committed
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
}

const std::string& Requester::getName() const {
	return m_impl->m_name;
}

void Requester::sendBinary(const std::string& request) {
	m_impl->sendBinary(request);
}

void Requester::send(const std::string& request) {
	m_impl->send(request);
}

1101
1102
1103
1104
void Requester::sendTwoBinaryParts(const std::string& request1, const std::string& request2) {
	m_impl->sendTwoBinaryParts(request1, request2);
}

legoc's avatar
legoc committed
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
bool Requester::receiveBinary(std::string& response) {
	return m_impl->receiveBinary(response);
}

bool Requester::receive(std::string& response) {
	return m_impl->receive(response);
}

void Requester::cancel() {
	m_impl->cancel();
}

1117
1118
1119
1120
bool Requester::isCanceled() const {
	return m_impl->m_canceled;
}

legoc's avatar
legoc committed
1121
1122
1123
///////////////////////////////////////////////////////////////////////////
// Configuration

1124
Configuration::Configuration(const std::string& name, const std::string& description, bool singleInstance, bool restart, int startingTime, int stoppingTime) {
legoc's avatar
legoc committed
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160

	m_name = name;
	m_description = description;
	m_singleInstance = singleInstance;
	m_restart = restart;
	m_startingTime = startingTime;
	m_stoppingTime = stoppingTime;
}

const std::string& Configuration::getName() const {
	return m_name;
}

const std::string& Configuration::getDescription() const {
	return m_description;
}

bool Configuration::hasSingleInstance() const {
	return m_singleInstance;
}

bool Configuration::canRestart() const {
	return m_restart;
}

int Configuration::getStartingTime() const {
	return m_startingTime;
}

int Configuration::getStoppingTime() const {
	return m_stoppingTime;
}

///////////////////////////////////////////////////////////////////////////
// Info

legoc's avatar
legoc committed
1161
Info::Info(const std::string& name, int id, int pid, State applicationState, State pastApplicationStates, const std::string& args) :
legoc's avatar
legoc committed
1162
	m_id(id),
legoc's avatar
legoc committed
1163
	m_pid(pid),
legoc's avatar
legoc committed
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
	m_applicationState(applicationState),
	m_pastApplicationStates(pastApplicationStates),
	m_args(args),
	m_name(name) {
}

int Info::getId() const {
	return m_id;
}

State Info::getState() const {
	return m_applicationState;
}

State Info::getPastStates() const {
	return m_pastApplicationStates;
}

const std::string& Info::getArgs() const {
	return m_args;
}

const std::string& Info::getName() const {
	return m_name;
}

legoc's avatar
legoc committed
1190
1191
1192
int Info::getPid() const {
	return m_pid;
}
legoc's avatar
legoc committed
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217

std::string toString(cameo::application::State applicationStates) {

	vector<string> states;

	if ((applicationStates & STARTING) != 0) {
		states.push_back("STARTING");
	}

	if ((applicationStates & RUNNING) != 0) {
		states.push_back("RUNNING");
	}

	if ((applicationStates & STOPPING) != 0) {
		states.push_back("STOPPING");
	}

	if ((applicationStates & KILLING) != 0) {
		states.push_back("KILLING");
	}

	if ((applicationStates & PROCESSING_ERROR) != 0) {
		states.push_back("PROCESSING_ERROR");
	}

legoc's avatar
legoc committed
1218
	if ((applicationStates & FAILURE) != 0) {
legoc's avatar
legoc committed
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
1300
1301
		states.push_back("ERROR");
	}

	if ((applicationStates & SUCCESS) != 0) {
		states.push_back("SUCCESS");
	}

	if ((applicationStates & STOPPED) != 0) {
		states.push_back("STOPPED");
	}

	if ((applicationStates & KILLED) != 0) {
		states.push_back("KILLED");
	}

	if (states.size() == 0) {
		return "UNKNOWN";
	}

	if (states.size() == 1) {
		return states.front();
	}

	ostringstream result;

	for (int i = 0; i < states.size() - 1; i++) {
		result << states[i] << ", ";
	}
	result << states.back();

	return result.str();
}

///////////////////////////////////////////////////////////////////////////////
// operator<<

std::ostream& operator<<(std::ostream& os, const cameo::application::This& application) {

	os << application.m_name << "." << application.m_id << "@" << application.m_serverEndpoint;

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Instance& instance) {

	os << instance.m_name << "." << instance.m_id << "@" << instance.m_server->getEndpoint();

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Publisher& publisher) {

	os << "pub." << publisher.getName()
		<< ":" << publisher.getApplicationName()
		<< "." << publisher.getApplicationId()
		<< "@" << publisher.getApplicationEndpoint();

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Subscriber& subscriber) {

	os << "sub." << subscriber.getPublisherName()
		<< ":" << subscriber.getInstanceName()
		<< "." << subscriber.getInstanceId()
		<< "@" << subscriber.getInstanceEndpoint();

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Request& request) {

	os << "[endpoint=" << request.m_impl->m_requesterEndpoint
			<< ", id=" << request.m_impl->m_requesterApplicationId << "]";

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Responder& responder) {

	os << "rep." << responder.getName()
		<< ":" << responder.m_impl->m_application->getName()
		<< "." << responder.m_impl->m_application->getId()
1302
		<< "@" << responder.m_impl->m_application->getEndpoint();
legoc's avatar
legoc committed
1303
1304
1305
1306
1307
1308
1309

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Requester& requester) {

	os << "req." << requester.getName()
1310
		<< "." << requester.m_impl->m_requesterId
legoc's avatar
legoc committed
1311
1312
		<< ":" << requester.m_impl->m_application->getName()
		<< "." << requester.m_impl->m_application->getId()
1313
		<< "@" << requester.m_impl->m_application->getEndpoint();
legoc's avatar
legoc committed
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333
1334
1335
1336
1337
1338
1339
1340
1341

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Configuration& info) {

	os << "[name=" << info.m_name
			<< ", description=" << info.m_description
			<< ", single instance=" << info.m_singleInstance
			<< ", restart=" << info.m_restart
			<< ", starting time=" << info.m_startingTime
			<< ", stopping time=" << info.m_stoppingTime << "]";

	return os;
}

std::ostream& operator<<(std::ostream& os, const application::Info& info) {

	os << "[name=" << info.m_name
			<< ", id=" << info.m_id
			<< ", state=" << info.m_applicationState
			<< ", args=" << info.m_args << "]";

	return os;
}

}
}