86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
#include "Subscription.h"
|
|
#include "Log.h"
|
|
#include "TCPSession.h"
|
|
#include <algorithm>
|
|
|
|
namespace core
|
|
{
|
|
|
|
Subscription::Subscription(std::string id, std::string mode)
|
|
: id(id), mode(mode), owner(NULL), handler(NULL) {}
|
|
|
|
Subscription::Subscription(std::string id, TCPSession &session, std::string mode)
|
|
: id(id), mode(mode), owner(&session), handler(NULL) {}
|
|
|
|
Subscription::Subscription(std::string id, TCPSession &session, std::string mode, SubscriptionHandler *handler)
|
|
: id(id), mode(mode), owner(&session), handler(handler) {}
|
|
|
|
Subscription::~Subscription()
|
|
{
|
|
std::stringstream out;
|
|
out << "cancel:" << id << std::endl;
|
|
for (auto subscriber : subscribers)
|
|
{
|
|
subscriber->write(out.str(), NULL);
|
|
}
|
|
}
|
|
|
|
int Subscription::subscribe(TCPSession &session)
|
|
{
|
|
if(handler)
|
|
handler->onSubscribe(session);
|
|
else
|
|
onSubscribe(session);
|
|
|
|
subscribers.push_back(&session);
|
|
return 1;
|
|
}
|
|
|
|
int Subscription::unsubscribe(TCPSession &session)
|
|
{
|
|
for (auto subscriber = subscribers.begin(); subscriber < subscribers.end(); ++subscriber)
|
|
{
|
|
if (*subscriber == &session)
|
|
{
|
|
subscribers.erase(subscriber++);
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int Subscription::process(coreutils::ZString &request, std::stringstream &out, TCPSession &session)
|
|
{
|
|
std::cout << "(" << handler << ")" << std::endl;
|
|
if (handler)
|
|
handler->process(request, out, session);
|
|
else
|
|
out << "event:" << request[1] << ":" << request[2] << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
int Subscription::event(std::stringstream &out)
|
|
{
|
|
for (auto subscriber = subscribers.begin(); subscriber < subscribers.end(); ++subscriber)
|
|
(*subscriber)->write(out.str(), NULL);
|
|
return 1;
|
|
}
|
|
|
|
bool Subscription::ifSubscriber(TCPSession &session)
|
|
{
|
|
return (std::find(subscribers.begin(), subscribers.end(), &session) != subscribers.end());
|
|
}
|
|
|
|
int Subscription::onSubscribe(TCPSession &session)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
// void Subscription::setHandler(SubscriptionHandler *handlers)
|
|
// {
|
|
// handler = handlers;
|
|
|
|
// coreutils::Log(coreutils::LOG_DEBUG_1) << "Test Test";
|
|
// }
|
|
}
|