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
469
470
void Instance::setOutputStreamSocket(std::unique_ptr<OutputStreamSocket>& socket) {
	m_outputStreamSocket = std::move(socket);
}

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

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

	// 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
480
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
}

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;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

	return m_lastState;
}

625
626
627
628
State Instance::waitFor(int states, const std::string& eventName, StateHandlerType handler) {
	return waitFor(states, eventName, handler, true);
}

legoc's avatar
legoc committed
629
630
631
632
633
634
State Instance::waitFor(int states, StateHandlerType handler) {
	return waitFor(states, "", handler);
}

State Instance::waitFor(StateHandlerType handler) {
	return waitFor(0, "", handler);
legoc's avatar
legoc committed
635
636
637
}

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

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

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

State Instance::getLastState() {

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

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

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

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

legoc's avatar
legoc committed
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
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
681
682
683
684
std::shared_ptr<OutputStreamSocket> Instance::getOutputStreamSocket() {
	return m_outputStreamSocket;
}

685
686
687
688
689
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
690
691
692
693
694
695
696
697
698
699
///////////////////////////////////////////////////////////////////////////
// InstanceArray

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

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

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

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

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

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

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

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

legoc's avatar
legoc committed
728
Publisher::Publisher(This * application, int publisherPort, int synchronizerPort, const std::string& name, int numberOfSubscribers) :
legoc's avatar
legoc committed
729
730
731
732
733
734
735
736
737
	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
738
std::unique_ptr<Publisher> Publisher::create(const std::string& name, int numberOfSubscribers) {
legoc's avatar
legoc committed
739

740
	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
741

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

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

legoc's avatar
legoc committed
752
	return unique_ptr<Publisher>(new Publisher(&This::m_instance, publisherPort, synchronizerPort, name, numberOfSubscribers));
legoc's avatar
legoc committed
753
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
}


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);
}

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

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

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

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

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

807
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
808
809
810
811
812
813
	m_impl(new SubscriberImpl(server, url, publisherPort, synchronizerPort, publisherName, numberOfSubscribers, instanceName, instanceId, instanceEndpoint, statusEndpoint)) {
}

Subscriber::~Subscriber() {
}

legoc's avatar
legoc committed
814
std::unique_ptr<Subscriber> Subscriber::create(Instance & instance, const std::string& publisherName) {
legoc's avatar
legoc committed
815
816
817
818
819
820
821
822
823
824
825
826
827
828
	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
	State lastState = instance.waitFor(0, publisherName);

	// 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
829
		|| lastState == FAILURE) {
legoc's avatar
legoc committed
830
		return unique_ptr<Subscriber>(nullptr);
legoc's avatar
legoc committed
831
832
833
834
835
836
837
838
839
	}

	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
840
	return unique_ptr<Subscriber>(nullptr);
legoc's avatar
legoc committed
841
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
}

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 {
867
868
869
870
871
872
873
874
875
	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
876
877
878
879
880
881
882
883
884
885
}

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

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

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

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

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

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

Request::~Request() {
}

legoc's avatar
legoc committed
904
905
906
907
908
909
910
911
912
913
914
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;
}

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

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

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

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

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

	return data;
}

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

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

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

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

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

	// 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
959
			return unique_ptr<Instance>(std::move(instances[i]));
960
961
962
963
		}
	}

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

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

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

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

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

Responder::~Responder() {
}

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

	string portName = ResponderImpl::RESPONDER_PREFIX + name;

988
	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
989

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

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

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

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

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

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

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

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

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

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

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

Requester::~Requester() {
}

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

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

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

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

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

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

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

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

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

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

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

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

1076
	// Request a requester port.
1077
1078
	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
1079

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

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

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);
}

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

legoc's avatar
legoc committed
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
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();
}

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

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

1123
Configuration::Configuration(const std::string& name, const std::string& description, bool singleInstance, bool restart, int startingTime, int stoppingTime) {
legoc's avatar
legoc committed
1124
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

	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
1160
Info::Info(const std::string& name, int id, int pid, State applicationState, State pastApplicationStates, const std::string& args) :
legoc's avatar
legoc committed
1161
	m_id(id),
legoc's avatar
legoc committed
1162
	m_pid(pid),
legoc's avatar
legoc committed
1163
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
	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
1189
1190
1191
int Info::getPid() const {
	return m_pid;
}
legoc's avatar
legoc committed
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216

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
1217
	if ((applicationStates & FAILURE) != 0) {
legoc's avatar
legoc committed
1218
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
		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()
1301
		<< "@" << responder.m_impl->m_application->getEndpoint();
legoc's avatar
legoc committed
1302
1303
1304
1305
1306
1307
1308

	return os;
}

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

	os << "req." << requester.getName()
1309
		<< "." << requester.m_impl->m_requesterId
legoc's avatar
legoc committed
1310
1311
		<< ":" << requester.m_impl->m_application->getName()
		<< "." << requester.m_impl->m_application->getId()
1312
		<< "@" << requester.m_impl->m_application->getEndpoint();
legoc's avatar
legoc committed
1313
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

	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;
}

}
}