#include "tconvert.h"
#include "tfarmcontroller.h"
#include "ttcpip.h"
#include "texception.h"
#include "tfarmproxy.h"
#include "tfilepath_io.h"
#include <strstream>
namespace {
//------------------------------------------------------------------------------
class Controller : public TFarmController, public TFarmProxy {
public:
Controller(const QString &hostName, const QString &addr, int port)
: TFarmProxy(hostName, addr, port) {}
// TFarmController interface implementation
QString addTask(const TFarmTask &task, bool suspended) override;
void removeTask(const QString &id) override;
void suspendTask(const QString &id) override;
void activateTask(const QString &id) override;
void restartTask(const QString &id) override;
void getTasks(vector<QString> &tasks) override;
void getTasks(const QString &parentId, vector<QString> &tasks) override;
void getTasks(const QString &parentId, vector<TaskShortInfo> &tasks) override;
void queryTaskInfo(const QString &id, TFarmTask &task) override;
void queryTaskShortInfo(const QString &id, QString &parentId, QString &name,
TaskState &status) override;
void attachServer(const QString &name, const QString &addr, int port) override;
void detachServer(const QString &name, const QString &addr, int port) override;
void taskSubmissionError(const QString &taskId, int errCode) override;
void taskProgress(const QString &taskId, int step, int stepCount,
int frameNumber, FrameState state) override;
void taskCompleted(const QString &taskId, int exitCode) override;
// fills the servers vector with the names of the servers
void getServers(vector<ServerIdentity> &servers) override;
// returns the state of the server whose id has been specified
ServerState queryServerState2(const QString &id) override;
// fills info with the infoes about the server whose id has been specified
void queryServerInfo(const QString &id, ServerInfo &info) override;
// activates the server whose id has been specified
void activateServer(const QString &id) override;
// deactivates the server whose id has been specified
// once deactivated, a server is not available for task rendering
void deactivateServer(const QString &id, bool completeRunningTasks = true) override;
};
//------------------------------------------------------------------------------
QString Controller::addTask(const TFarmTask &task, bool suspended) {
int i, count = task.getTaskCount();
QString data("addTask@TFarmTask_2");
data += ",";
data += task.m_name;
data += ",";
data += task.getCommandLine(true);
data += ",";
data += task.m_user;
data += ",";
data += task.m_hostName;
data += ",";
data += QString::number(suspended);
data += ",";
data += QString::number(task.m_stepCount);
data += ",";
data += QString::number(task.m_priority);
data += ",";
data += QString::number(task.m_platform);
data += ",";
int depCount = task.m_dependencies->getTaskCount();
data += QString::number(depCount);
i = 0;
for (; i < depCount; ++i) {
data += ",";
data += task.m_dependencies->getTaskId(i);
}
for (i = 0; i < count; ++i) {
TFarmTask &tt = const_cast<TFarmTask &>(task);
TFarmTask *subtask = tt.getTask(i);
data += ",";
data += subtask->m_name;
data += ",";
data += subtask->getCommandLine(true);
data += ",";
data += subtask->m_user;
data += ",";
data += subtask->m_hostName;
data += ",";
data += QString::number(subtask->m_stepCount);
data += ",";
data += QString::number(subtask->m_priority);
}
QString reply = sendToStub(data);
// MessageBox(NULL, data.c_str(), "ciao", 0);
return reply;
}
//------------------------------------------------------------------------------
void Controller::removeTask(const QString &id) {
QString data("removeTask");
data += ",";
data += id;
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::suspendTask(const QString &id) {
QString data("suspendTask");
data += ",";
data += id;
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::activateTask(const QString &id) {
QString data("activateTask");
data += ",";
data += id;
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::restartTask(const QString &id) {
QString data("restartTask");
data += ",";
data += id;
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::getTasks(vector<QString> &tasks) {
QString data("getTasks@vector");
QString reply = sendToStub(data);
// la stringa restituita contiene le informazioni desiderate separate da ","
vector<QString> argv;
extractArgs(reply, argv);
tasks.clear();
std::vector<QString>::iterator it = argv.begin();
for (; it != argv.end(); ++it) tasks.push_back(*it);
}
//------------------------------------------------------------------------------
void Controller::getTasks(const QString &parentId, vector<QString> &tasks) {
QString data("getTasks@string@vector");
data += ",";
data += parentId;
QString reply = sendToStub(data);
// la stringa restituita contiene le informazioni deiderate separate da ","
vector<QString> argv;
extractArgs(reply, argv);
tasks.clear();
std::vector<QString>::iterator it = argv.begin();
for (; it != argv.end(); ++it) tasks.push_back(*it);
}
//------------------------------------------------------------------------------
void Controller::getTasks(const QString &parentId,
vector<TaskShortInfo> &tasks) {
QString data("getTasks@string@vector$TaskShortInfo");
data += ",";
data += parentId;
QString reply = sendToStub(data);
// la stringa restituita contiene le informazioni desiderate separate da ","
vector<QString> argv;
extractArgs(reply, argv);
tasks.clear();
std::vector<QString>::iterator it = argv.begin();
for (; it != argv.end(); ++it) {
QString id = *it++;
QString name = *it++;
TaskState status;
status = (TaskState)(*it).toInt();
tasks.push_back(TaskShortInfo(id, name, status));
}
}
//------------------------------------------------------------------------------
void Controller::queryTaskInfo(const QString &id, TFarmTask &task) {
QString data("queryTaskInfo_2");
data += ",";
data += id;
QString reply = sendToStub(data);
// la stringa restituita contiene le informazioni desiderate separate da ","
vector<QString> argv;
int count = extractArgs(reply, argv);
if (reply == "") return;
assert(argv.size() > 15);
int incr = 0;
task.m_name = argv[incr++];
task.parseCommandLine(argv[incr++]);
task.m_priority = argv[incr++].toInt();
task.m_user = argv[incr++];
task.m_hostName = argv[incr++];
task.m_id = argv[incr++];
task.m_parentId = argv[incr++];
task.m_status = (TaskState)argv[incr++].toInt();
task.m_server = argv[incr++];
task.m_submissionDate = QDateTime::fromString(argv[incr++]);
task.m_startDate = QDateTime::fromString(argv[incr++]);
task.m_completionDate = QDateTime::fromString(argv[incr++]);
task.m_successfullSteps = argv[incr++].toInt();
task.m_failedSteps = argv[incr++].toInt();
task.m_stepCount = argv[incr++].toInt();
if (incr < count) {
task.m_platform = (TFarmPlatform)argv[incr++].toInt();
int depCount = 0;
depCount = argv[incr++].toInt();
if (depCount > 0) {
task.m_dependencies = new TFarmTask::Dependencies;
for (int i = 0; i < depCount; ++i) task.m_dependencies->add(argv[incr++]);
}
}
}
//------------------------------------------------------------------------------
void Controller::queryTaskShortInfo(const QString &id, QString &parentId,
QString &name, TaskState &status) {
QString data("queryTaskShortInfo");
data += ",";
data += id;
QString reply = sendToStub(data);
// la stringa restituita contiene le informazioni desiderate separate da ","
vector<QString> argv;
extractArgs(reply, argv);
assert(argv.size() == 3);
parentId = argv[0];
name = argv[1];
status = (TaskState)argv[2].toInt();
}
//------------------------------------------------------------------------------
void Controller::attachServer(const QString &name, const QString &addr,
int port) {
QString data("attachServer");
data += ",";
data += name;
data += ",";
data += addr;
data += ",";
data += QString::number(port);
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::detachServer(const QString &name, const QString &addr,
int port) {
QString data("detachServer");
data += ",";
data += name;
data += ",";
data += addr;
data += ",";
data += QString::number(port);
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::taskSubmissionError(const QString &taskId, int errCode) {
QString data("taskSubmissionError");
data += ",";
data += taskId;
data += ",";
data += QString::number(errCode);
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::taskProgress(const QString &taskId, int step, int stepCount,
int frameNumber, FrameState state) {
QString data("taskProgress");
data += ",";
data += taskId;
data += ",";
data += QString::number(step);
data += ",";
data += QString::number(stepCount);
data += ",";
data += QString::number(frameNumber);
data += ",";
data += QString::number(state);
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::taskCompleted(const QString &taskId, int exitCode) {
QString data("taskCompleted");
data += ",";
data += taskId;
data += ",";
data += QString::number(exitCode);
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::getServers(vector<ServerIdentity> &servers) {
QString data("getServers");
QString reply = sendToStub(data);
// la stringa restituita contiene le informazioni desiderate separate da ","
vector<QString> argv;
extractArgs(reply, argv);
servers.clear();
std::vector<QString>::iterator it = argv.begin();
for (; it != argv.end(); it += 2)
servers.push_back(ServerIdentity(*it, *(it + 1)));
}
//------------------------------------------------------------------------------
ServerState Controller::queryServerState2(const QString &name) {
QString data("queryServerState2");
data += ",";
data += name;
QString reply = sendToStub(data);
ServerState serverState = (ServerState)reply.toInt();
return serverState;
}
//------------------------------------------------------------------------------
void Controller::queryServerInfo(const QString &id, ServerInfo &info) {
QString data("queryServerInfo");
data += ",";
data += id;
QString reply = sendToStub(data);
if (reply != "") {
// la stringa restituita contiene le informazioni desiderate separate da ","
vector<QString> argv;
extractArgs(reply, argv);
info.m_name = argv[0];
info.m_ipAddress = argv[1];
info.m_portNumber = argv[2];
info.m_state = (ServerState)argv[3].toInt();
info.m_platform = argv[4];
int cpuCount, totPhysMem, totVirtMem, availPhysMem, availVirtMem;
cpuCount = argv[5].toInt();
totPhysMem = argv[6].toInt();
totVirtMem = argv[7].toInt();
availPhysMem = argv[8].toInt();
availVirtMem = argv[9].toInt();
info.m_currentTaskId = argv[10];
info.m_cpuCount = cpuCount;
info.m_totPhysMem = totPhysMem;
info.m_totVirtMem = totVirtMem;
info.m_availPhysMem = availPhysMem;
info.m_availVirtMem = availVirtMem;
}
}
//------------------------------------------------------------------------------
void Controller::activateServer(const QString &id) {
QString data("activateServer");
data += ",";
data += id;
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
void Controller::deactivateServer(const QString &id,
bool completeRunningTasks) {
QString data("deactivateServer");
data += ",";
data += id;
data += ",";
data += QString::number(completeRunningTasks);
QString reply = sendToStub(data);
}
//------------------------------------------------------------------------------
} // anonymous namespace
//==============================================================================
TFarmControllerFactory::TFarmControllerFactory() {}
//------------------------------------------------------------------------------
int TFarmControllerFactory::create(const ControllerData &data,
TFarmController **controller)
{
*controller = new Controller(data.m_hostName, data.m_ipAddress, data.m_port);
return 0;
}
//------------------------------------------------------------------------------
int TFarmControllerFactory::create(const QString &hostname, int port,
TFarmController **controller) {
*controller = new Controller(hostname, "", port);
return 0;
}
//------------------------------------------------------------------------------
void loadControllerData(const TFilePath &fp, ControllerData &data) {
Tifstream is(fp);
if (!is || !is.good()) {
throw TException(L"Unable to get Farm Controller Data (looking for '" +
fp.getWideString() + L"')");
}
while (!is.eof()) {
char line[1024];
is.getline(line, 1024);
if (line[0] != '#' && QString(line) != "") {
std::istrstream iss(line);
char hostName[512];
char ipAddr[80];
int port;
iss >> hostName >> ipAddr >> port;
data.m_hostName = hostName;
data.m_ipAddress = ipAddr;
data.m_port = port;
break;
}
}
}