57 lines
2.0 KiB
C++
57 lines
2.0 KiB
C++
#include "SubscriptionManager.h"
|
|
#include "Log.h"
|
|
|
|
namespace core {
|
|
|
|
SubscriptionManager::SubscriptionManager() {}
|
|
|
|
int SubscriptionManager::removeSessionSubscriptions(TCPSession &session) {
|
|
int count = 0;
|
|
for(auto subscription = subscriptions.begin(); subscription < subscriptions.end(); ++subscription) {
|
|
count += (*subscription)->unsubscribe(session);
|
|
}
|
|
coreutils::Log(coreutils::LOG_DEBUG_2) << "Removed session from " << count << " subscriptions.";
|
|
return count;
|
|
}
|
|
|
|
int SubscriptionManager::processCommand(coreutils::ZString &request, TCPSession &session) {
|
|
|
|
if(request[0].equals("publish")) {
|
|
subscriptions.push_back(new Subscription(request[1].str(), session));
|
|
return 1;
|
|
} else if(request[0].equals("catalog")) {
|
|
session.out << ":catalog:";
|
|
for(auto subscription = subscriptions.begin(); subscription < subscriptions.end(); ++subscription) {
|
|
session.out << (*subscription)->id << "|";
|
|
(*subscription)->processCommand(request, session);
|
|
session.out << (";");
|
|
}
|
|
session.out << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
for(auto subscription = subscriptions.begin(); subscription < subscriptions.end(); ++subscription) {
|
|
if(request[1].equals((*subscription)->id)) {
|
|
if(request[0].equals("unpublish")) {
|
|
subscriptions.erase(subscription);
|
|
} else if(request[0].equals("subscribe")) {
|
|
(*subscription)->subscribe(session);
|
|
} else if(request[0].equals("unsubscribe")) {
|
|
(*subscription)->unsubscribe(session);
|
|
} else if(request[0].equals("event")) {
|
|
if((*subscription)->owner == &session) {
|
|
std::stringstream out;
|
|
(*subscription)->process(request, out);
|
|
(*subscription)->event(out);
|
|
}
|
|
else
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
}
|