301 lines
10 KiB
C++
301 lines
10 KiB
C++
#include "corax.h"
|
|
|
|
#include <iostream>
|
|
|
|
using std::cout;
|
|
using std::endl;
|
|
|
|
Corax* Corax::corax = 0;
|
|
|
|
Corax::Corax(QObject *parent):
|
|
QObject(parent),
|
|
started(false),
|
|
server(new W::Server(W::String(u"Corax"), this)),
|
|
logger(new W::Logger()),
|
|
parentReporter(new W::ParentReporter()),
|
|
attributes(new M::Attributes(W::Address({u"attributes"}))),
|
|
commands(new U::Commands(W::Address{u"management"})),
|
|
connector(0),
|
|
dispatcher(new W::Dispatcher()),
|
|
caches(),
|
|
parsers(),
|
|
players()
|
|
{
|
|
if (corax != 0)
|
|
{
|
|
throw SingletonError();
|
|
}
|
|
Corax::corax = this;
|
|
|
|
connector = new U::Connector(dispatcher, server, commands);
|
|
connector->addIgnoredNode(W::String(u"Lorgar"));
|
|
connector->addIgnoredNode(W::String(u"Roboute"));
|
|
|
|
connect(attributes, SIGNAL(serviceMessage(const QString&)), SLOT(onModelServiceMessage(const QString&)));
|
|
connect(commands, SIGNAL(serviceMessage(const QString&)), SLOT(onModelServiceMessage(const QString&)));
|
|
connect(connector, SIGNAL(serviceMessage(const QString&)), SLOT(onModelServiceMessage(const QString&)));
|
|
connect(connector, SIGNAL(nodeConnected(const W::String&)), SLOT(onNodeConnected(const W::String&)));
|
|
connect(connector, SIGNAL(nodeDisconnected(const W::String&)), SLOT(onNodeDisconnected(const W::String&)));
|
|
connect(server, SIGNAL(connectionCountChange(uint64_t)), SLOT(onConnectionCountChanged(uint64_t)));
|
|
|
|
dispatcher->registerDefaultHandler(parentReporter);
|
|
dispatcher->registerDefaultHandler(logger);
|
|
|
|
attributes->addAttribute(W::String(u"connectionsCount"), new M::String(W::String(u"0"), W::Address({u"attributes", u"connectionCount"})));
|
|
attributes->addAttribute(W::String(u"name"), new M::String(W::String(u"Corax"), W::Address({u"attributes", u"name"})));
|
|
attributes->addAttribute(W::String(u"version"), new M::String(W::String(u"0.0.3"), W::Address({u"attributes", u"version"})));
|
|
attributes->addAttribute(W::String(u"players"), new M::String(W::String(u"0"), W::Address({u"attributes", u"players"})));
|
|
|
|
createCaches();
|
|
createHandlers();
|
|
}
|
|
|
|
Corax::~Corax()
|
|
{
|
|
std::map<W::String, Parser*>::const_iterator pbeg = parsers.begin();
|
|
std::map<W::String, Parser*>::const_iterator pend = parsers.end();
|
|
|
|
for (; pbeg != pend; ++pbeg) {
|
|
delete pbeg->second;
|
|
}
|
|
|
|
std::map<W::String, ResourceCache*>::const_iterator beg = caches.begin();
|
|
std::map<W::String, ResourceCache*>::const_iterator end = caches.end();
|
|
|
|
for (; beg != end; ++beg) {
|
|
delete beg->second;
|
|
}
|
|
|
|
std::map<W::Address, M::Player*>::const_iterator plit = players.begin();
|
|
std::map<W::Address, M::Player*>::const_iterator plend = players.end();
|
|
|
|
for (; plit != plend; ++plit) {
|
|
delete plit->second;
|
|
}
|
|
|
|
delete connector;
|
|
|
|
dispatcher->unregisterDefaultHandler(logger);
|
|
|
|
delete commands;
|
|
delete attributes;
|
|
|
|
delete logger;
|
|
delete dispatcher;
|
|
|
|
Corax::corax = 0;
|
|
}
|
|
|
|
void Corax::onConnectionCountChanged(uint64_t count)
|
|
{
|
|
attributes->setAttribute(W::String(u"connectionsCount"), new W::String(std::to_string(count)));
|
|
}
|
|
|
|
void Corax::start()
|
|
{
|
|
if (started) {
|
|
throw 3;
|
|
}
|
|
std::map<W::String, ResourceCache*>::iterator beg = caches.begin();
|
|
std::map<W::String, ResourceCache*>::iterator end = caches.end();
|
|
|
|
cout << "Starting corax..." << endl;
|
|
server->listen(8080);
|
|
|
|
cout << "Registering models..." << endl;
|
|
attributes->getRegistered(connector);
|
|
commands->getRegistered(connector);
|
|
|
|
for (; beg != end; ++beg) {
|
|
beg->second->getRegistered(connector);
|
|
}
|
|
|
|
std::map<W::Address, M::Player*>::const_iterator plit = players.begin();
|
|
std::map<W::Address, M::Player*>::const_iterator plend = players.end();
|
|
|
|
for (; plit != plend; ++plit) {
|
|
plit->second->getRegistered(connector);;
|
|
}
|
|
|
|
cout << "Opening caches..." << endl;
|
|
|
|
beg = caches.begin();
|
|
for (; beg != end; ++beg) {
|
|
beg->second->open();
|
|
}
|
|
|
|
commands->enableCommand(W::String(u"clearCache"), true);
|
|
commands->enableCommand(W::String(u"givePlayer"), true);
|
|
|
|
started = true;
|
|
cout << "Corax is ready" << endl;
|
|
}
|
|
|
|
void Corax::stop()
|
|
{
|
|
if (!started) {
|
|
throw 3;
|
|
}
|
|
|
|
std::map<W::String, ResourceCache*>::iterator beg = caches.begin();
|
|
std::map<W::String, ResourceCache*>::iterator end = caches.end();
|
|
|
|
cout << "Stopping corax..." << endl;
|
|
commands->getUnregistered();
|
|
attributes->getUnregistered();
|
|
|
|
for (; beg != end; ++beg) {
|
|
beg->second->getUnregistered();
|
|
}
|
|
|
|
std::map<W::Address, M::Player*>::const_iterator plit = players.begin();
|
|
std::map<W::Address, M::Player*>::const_iterator plend = players.end();
|
|
|
|
for (; plit != plend; ++plit) {
|
|
plit->second->getUnregistered();
|
|
}
|
|
|
|
server->stop();
|
|
|
|
started = false;
|
|
cout << "Corax is stopped" << endl;
|
|
}
|
|
|
|
void Corax::onModelServiceMessage(const QString& msg)
|
|
{
|
|
cout << msg.toStdString() << endl;
|
|
}
|
|
|
|
void Corax::addCache(ResourceCache* cache)
|
|
{
|
|
attributes->addAttribute(cache->name, new M::String(W::String(u"0"), W::Address({u"attributes", cache->name})));
|
|
|
|
connect(cache, SIGNAL(serviceMessage(const QString&)), SLOT(onModelServiceMessage(const QString&)));
|
|
connect(cache, SIGNAL(countChange(uint64_t)), SLOT(onCacheCountChange(uint64_t)));
|
|
|
|
parentReporter->registerParent(cache->getAddress(), cache->subscribeMember);
|
|
|
|
caches.insert(std::make_pair(cache->name, cache));
|
|
}
|
|
|
|
void Corax::h_clearCache(const W::Event& ev)
|
|
{
|
|
const W::Vocabulary& vc = static_cast<const W::Vocabulary&>(ev.getData());
|
|
const W::String& name = static_cast<const W::String&>(vc.at(u"name"));
|
|
|
|
cout << "received command to clear cache " << name.toString() << endl;
|
|
|
|
std::map<W::String, ResourceCache*>::iterator itr = caches.find(name);
|
|
if (itr == caches.end()) {
|
|
cout << "cache " << name.toString() << " doesn't exist" << endl;
|
|
} else {
|
|
itr->second->clear();
|
|
}
|
|
}
|
|
|
|
void Corax::h_parseDirectory(const W::Event& ev)
|
|
{
|
|
const W::Vocabulary& vc = static_cast<const W::Vocabulary&>(ev.getData());
|
|
const W::String& path = static_cast<const W::String&>(vc.at(u"path"));
|
|
|
|
cout << "received command to parse directory " << path.toString() << endl;
|
|
|
|
std::map<W::String, Parser*>::const_iterator itr = parsers.find(path);
|
|
|
|
if (itr != parsers.end()) {
|
|
cout << "directory " << path.toString() << " is already being parsed" << endl;
|
|
} else {
|
|
const W::Socket* socket = connector->getNodeSocket(W::String(u"Perturabo"));
|
|
ResourceCache* music = caches.at(W::String(u"music"));
|
|
ResourceCache* images = caches.at(W::String(u"images"));
|
|
Parser* parser = new Parser(socket, dispatcher, music, images);
|
|
parsers.insert(std::make_pair(path, parser));
|
|
|
|
connect(parser, SIGNAL(serviceMessage(const QString&)), SLOT(onModelServiceMessage(const QString&)));
|
|
connect(parser, SIGNAL(done(const W::String&)), SLOT(onParserDone(const W::String&)));
|
|
|
|
parser->parse(path);
|
|
}
|
|
}
|
|
|
|
void Corax::createCaches()
|
|
{
|
|
ResourceCache* music = new ResourceCache(W::String(u"music"));
|
|
ResourceCache* images = new ResourceCache(W::String(u"images"));
|
|
|
|
addCache(music);
|
|
addCache(images);
|
|
}
|
|
|
|
void Corax::createHandlers()
|
|
{
|
|
W::Handler* clearCache = W::Handler::create(W::Address({u"management", u"clearCache"}), this, &Corax::_h_clearCache);
|
|
W::Vocabulary clearArgs;
|
|
clearArgs.insert(u"name", W::Uint64(W::Object::string));
|
|
commands->addCommand(W::String(u"clearCache"), clearCache, clearArgs);
|
|
|
|
W::Handler* parseDirectory = W::Handler::create(W::Address({u"management", u"parseDirectory"}), this, &Corax::_h_parseDirectory);
|
|
W::Vocabulary parseArgs;
|
|
parseArgs.insert(u"path", W::Uint64(W::Object::string));
|
|
commands->addCommand(W::String(u"parseDirectory"), parseDirectory, parseArgs);
|
|
|
|
W::Handler* givePlayer = W::Handler::create(W::Address({u"management", u"givePlayer"}), this, &Corax::_h_givePlayer);
|
|
commands->addCommand(W::String(u"givePlayer"), givePlayer, W::Vocabulary());
|
|
}
|
|
|
|
void Corax::onParserDone(const W::String& path)
|
|
{
|
|
std::map<W::String, Parser*>::const_iterator itr = parsers.find(path);
|
|
|
|
delete itr->second;
|
|
parsers.erase(itr);
|
|
}
|
|
|
|
void Corax::onCacheCountChange(uint64_t count)
|
|
{
|
|
ResourceCache* cache = static_cast<ResourceCache*>(sender());
|
|
|
|
attributes->setAttribute(cache->name, W::String(std::to_string(count)));
|
|
}
|
|
|
|
void Corax::onNodeConnected(const W::String& name)
|
|
{
|
|
cout << "connected node " << name.toString() << endl;
|
|
if (name == u"Perturabo") {
|
|
commands->enableCommand(W::String(u"parseDirectory"), true);
|
|
}
|
|
}
|
|
|
|
void Corax::onNodeDisconnected(const W::String& name)
|
|
{
|
|
cout << "disconnected node " << name.toString() << endl;
|
|
if (name == u"Perturabo") {
|
|
commands->enableCommand(W::String(u"parseDirectory"), false);
|
|
}
|
|
}
|
|
|
|
void Corax::h_givePlayer(const W::Event& ev)
|
|
{
|
|
W::String num(std::to_string(players.size()));
|
|
W::Address addr{u"players", num};
|
|
M::Player* pl = new M::Player(addr);
|
|
connect(pl, SIGNAL(serviceMessage(const QString&)), SLOT(onModelServiceMessage(const QString&)));
|
|
players.insert(std::make_pair(addr, pl));
|
|
|
|
attributes->setAttribute(W::String(u"players"), W::String(std::to_string(players.size())));
|
|
pl->getRegistered(connector); //it's a handler, so I assume corax is started here;
|
|
|
|
W::Vocabulary* vc = new W::Vocabulary();
|
|
vc->insert(u"address", addr);
|
|
|
|
const W::Vocabulary& svc = static_cast<const W::Vocabulary&>(ev.getData());
|
|
const W::Address& source = static_cast<const W::Address&>(svc.at(u"source"));
|
|
uint64_t id = ev.getSenderId();
|
|
vc->insert(u"source", W::Address{u"management"}); //TODO think about it, may be Corax should be a model?
|
|
|
|
W::Event res(source + W::Address{u"getPlayer"}, vc);
|
|
res.setSenderId(id);
|
|
connector->getConnection(id)->send(res);
|
|
}
|
|
|