89 lines
3.0 KiB
C++
89 lines
3.0 KiB
C++
|
/*
|
||
|
|
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <ListUtils.h>
|
||
|
|
#include <StandardTypes.h>
|
||
|
|
#include "StartTESClient.h"
|
||
|
|
#include "../../Helpers/RemoteEndpoint.h"
|
||
|
|
#include "../../MessageBroker/MessageBroker.h"
|
||
|
|
|
||
|
|
namespace distributed3 {
|
||
|
|
ListExpr StartTESClient::typeMapping(ListExpr args) {
|
||
|
|
if (!nl->HasLength(args, 3)) {
|
||
|
|
return listutils::typeError("You must provide 3 arguments.");
|
||
|
|
}
|
||
|
|
const ListExpr worker = nl->First(args);
|
||
|
|
const ListExpr host = nl->Second(args);
|
||
|
|
const ListExpr port = nl->Third(args);
|
||
|
|
|
||
|
|
if (!CcInt::checkType(worker)) {
|
||
|
|
return listutils::typeError(
|
||
|
|
"The first argument (worker) must be of type int");
|
||
|
|
}
|
||
|
|
if (!CcString::checkType(host)) {
|
||
|
|
return listutils::typeError(
|
||
|
|
"The second argument (host) must be of type string");
|
||
|
|
}
|
||
|
|
if (!CcInt::checkType(port)) {
|
||
|
|
return listutils::typeError(
|
||
|
|
"The second argument (port) must be of type int");
|
||
|
|
}
|
||
|
|
|
||
|
|
return nl->SymbolAtom(CcBool::BasicType());
|
||
|
|
}
|
||
|
|
|
||
|
|
int StartTESClient::valueMapping(Word *args, Word &result, int message,
|
||
|
|
Word &local, Supplier s) {
|
||
|
|
result = qp->ResultStorage(s);
|
||
|
|
CcInt *workerNoObj = (CcInt *) args[0].addr;
|
||
|
|
CcString *hostObj = (CcString *) args[1].addr;
|
||
|
|
CcInt *portWrapper = (CcInt *) args[2].addr;
|
||
|
|
|
||
|
|
PRECONDITION(workerNoObj->IsDefined(), "worker is undefined");
|
||
|
|
PRECONDITION(hostObj->IsDefined(), "host is undefined");
|
||
|
|
PRECONDITION(portWrapper->IsDefined(), "port is undefined");
|
||
|
|
const int worker = workerNoObj->GetValue();
|
||
|
|
const std::string &hostName = hostObj->GetValue();
|
||
|
|
const int port = portWrapper->GetValue();
|
||
|
|
PRECONDITION(port > 0, "port must be grater than 0")
|
||
|
|
PRECONDITION(worker >= 0, "worker must be grater or equal to 0")
|
||
|
|
PRECONDITION(!hostName.empty(), "host must not be empty")
|
||
|
|
|
||
|
|
const RemoteEndpoint host(hostName, port);
|
||
|
|
MessageBroker &broker = MessageBroker::get();
|
||
|
|
bool clientStarted = broker.startClient(worker, host);
|
||
|
|
|
||
|
|
((CcBool *) result.addr)->Set(true, clientStarted);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
OperatorSpec StartTESClient::operatorSpec(
|
||
|
|
"int x string x int -> bool",
|
||
|
|
"# (_,_,_)",
|
||
|
|
"This operator starts a message client locally as component of the "
|
||
|
|
"Pregel messaging system."
|
||
|
|
"It connects to a (possibly) remote host that runs a message server "
|
||
|
|
"as counterpart."
|
||
|
|
"The client acts as a sender for Pregel messages to other workers."
|
||
|
|
"It has an index, which designates the logical address of the connected "
|
||
|
|
"message server."
|
||
|
|
"Arguments are (<index>, <host>, <port>)",
|
||
|
|
"query startTESClient(2, \"localhost\", 8001);",
|
||
|
|
"This operator belongs to the Pregel API."
|
||
|
|
"It may require knowledge of the system to effectively understand and use "
|
||
|
|
"all the operators that are provided."
|
||
|
|
"CAUTION: This operator is used internally by the Pregel system. "
|
||
|
|
"Hence you must not use it in queries yourself."
|
||
|
|
"Doing so may lead to inconsistent states of the Pregel system."
|
||
|
|
);
|
||
|
|
|
||
|
|
Operator StartTESClient::startTESClient(
|
||
|
|
"startTESClient",
|
||
|
|
StartTESClient::operatorSpec.getStr(),
|
||
|
|
StartTESClient::valueMapping,
|
||
|
|
Operator::SimpleSelect,
|
||
|
|
StartTESClient::typeMapping
|
||
|
|
);
|
||
|
|
}
|