From f2e3bfbe2ed1b8f35aa958041ee766f9d8ddf31e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Remko=20Tron=C3=A7on?= Date: Tue, 16 Jun 2009 22:41:48 +0200 Subject: Delay IQHandler removal in IQRouter during IQ handling. diff --git a/Swiften/Examples/TuneBot/TuneBot.cpp b/Swiften/Examples/TuneBot/TuneBot.cpp index d2bb100..d842798 100644 --- a/Swiften/Examples/TuneBot/TuneBot.cpp +++ b/Swiften/Examples/TuneBot/TuneBot.cpp @@ -28,13 +28,13 @@ class TuneBot { discoResponder_->setDiscoInfo(discoInfo); discoResponder_->setDiscoInfo(capsInfo_->getNode() + "#" + capsInfo_->getVersion(), discoInfo); - client_->onSessionStarted.connect(bind(&TuneBot::handleSessionStarted, this)); + client_->onConnected.connect(bind(&TuneBot::handleSessionStarted, this)); client_->onMessageReceived.connect(bind(&TuneBot::handleMessage, this, _1)); client_->connect(); } void handleSessionStarted() { - GetRosterRequest* rosterRequest = new GetRosterRequest(router_, Request::AutoDeleteAfterResponse); + boost::shared_ptr rosterRequest(new GetRosterRequest(router_)); rosterRequest->onResponse.connect(bind(&TuneBot::handleRosterReceived, this, _1)); rosterRequest->send(); } @@ -43,7 +43,7 @@ class TuneBot { boost::shared_ptr presence(new Presence()); presence->addPayload(capsInfo_); presence->setPriority(-1); - client_->send(presence); + client_->sendPresence(presence); } void handleMessage(shared_ptr message) { @@ -58,8 +58,7 @@ class TuneBot { }; -int main(int argc, char* argv[]) -{ +int main(int argc, char* argv[]) { if (argc != 3) { std::cerr << "Usage: " << argv[0] << " " << std::endl; return -1; @@ -67,6 +66,6 @@ int main(int argc, char* argv[]) SimpleEventLoop eventLoop; - TuneBot bot(argv[1], argv[2]); + TuneBot bot(JID(argv[1]), argv[2]); eventLoop.run(); } diff --git a/Swiften/Queries/IQRouter.cpp b/Swiften/Queries/IQRouter.cpp index 671566a..b20d344 100644 --- a/Swiften/Queries/IQRouter.cpp +++ b/Swiften/Queries/IQRouter.cpp @@ -10,20 +10,18 @@ namespace Swift { -namespace { - void noop(IQHandler*) {} - struct PointerEquals { - PointerEquals(IQHandler* handler) : handler_(handler) {} - bool operator()(boost::shared_ptr o) { return handler_ == o.get(); } - IQHandler* handler_; - }; -} +static void noop(IQHandler*) {} -IQRouter::IQRouter(IQChannel* channel) : channel_(channel) { +IQRouter::IQRouter(IQChannel* channel) : channel_(channel), queueRemoves_(false) { channel->onIQReceived.connect(boost::bind(&IQRouter::handleIQ, this, _1)); } +IQRouter::~IQRouter() { +} + void IQRouter::handleIQ(boost::shared_ptr iq) { + queueRemoves_ = true; + bool handled = false; foreach(boost::shared_ptr handler, handlers_) { handled |= handler->handleIQ(iq); @@ -34,22 +32,38 @@ void IQRouter::handleIQ(boost::shared_ptr iq) { if (!handled && (iq->getType() == IQ::Get || iq->getType() == IQ::Set) ) { channel_->sendIQ(IQ::createError(iq->getFrom(), iq->getID(), Error::FeatureNotImplemented, Error::Cancel)); } + + processPendingRemoves(); + + queueRemoves_ = false; } -void IQRouter::addHandler(IQHandler* handler) { - handlers_.push_back(boost::shared_ptr(handler, noop)); +void IQRouter::processPendingRemoves() { + foreach(boost::shared_ptr handler, queuedRemoves_) { + handlers_.erase(std::remove(handlers_.begin(), handlers_.end(), handler), handlers_.end()); + } + queuedRemoves_.clear(); } -void IQRouter::addHandler(boost::shared_ptr handler) { - handlers_.push_back(handler); +void IQRouter::addHandler(IQHandler* handler) { + addHandler(boost::shared_ptr(handler, noop)); } void IQRouter::removeHandler(IQHandler* handler) { - handlers_.erase(std::remove_if(handlers_.begin(), handlers_.end(), PointerEquals(handler)), handlers_.end()); + removeHandler(boost::shared_ptr(handler, noop)); +} + +void IQRouter::addHandler(boost::shared_ptr handler) { + handlers_.push_back(handler); } void IQRouter::removeHandler(boost::shared_ptr handler) { - handlers_.erase(std::remove(handlers_.begin(), handlers_.end(), handler)); + if (queueRemoves_) { + queuedRemoves_.push_back(handler); + } + else { + handlers_.erase(std::remove(handlers_.begin(), handlers_.end(), handler), handlers_.end()); + } } void IQRouter::sendIQ(boost::shared_ptr iq) { diff --git a/Swiften/Queries/IQRouter.h b/Swiften/Queries/IQRouter.h index ea80bf5..6de2ff9 100644 --- a/Swiften/Queries/IQRouter.h +++ b/Swiften/Queries/IQRouter.h @@ -14,10 +14,11 @@ namespace Swift { class IQRouter { public: IQRouter(IQChannel* channel); + ~IQRouter(); void addHandler(IQHandler* handler); - void addHandler(boost::shared_ptr handler); void removeHandler(IQHandler* handler); + void addHandler(boost::shared_ptr handler); void removeHandler(boost::shared_ptr handler); void sendIQ(boost::shared_ptr iq); @@ -25,10 +26,13 @@ namespace Swift { private: void handleIQ(boost::shared_ptr iq); + void processPendingRemoves(); private: IQChannel* channel_; std::vector< boost::shared_ptr > handlers_; + std::vector< boost::shared_ptr > queuedRemoves_; + bool queueRemoves_; }; } diff --git a/Swiften/Queries/UnitTest/IQRouterTest.cpp b/Swiften/Queries/UnitTest/IQRouterTest.cpp index 8fa8d2a..94b7de8 100644 --- a/Swiften/Queries/UnitTest/IQRouterTest.cpp +++ b/Swiften/Queries/UnitTest/IQRouterTest.cpp @@ -12,6 +12,8 @@ using namespace Swift; class IQRouterTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(IQRouterTest); + CPPUNIT_TEST(testRemoveHandler); + CPPUNIT_TEST(testRemoveHandler_AfterHandleIQ); CPPUNIT_TEST(testHandleIQ_SuccesfulHandlerFirst); CPPUNIT_TEST(testHandleIQ_SuccesfulHandlerLast); CPPUNIT_TEST(testHandleIQ_NoSuccesfulHandler); @@ -29,6 +31,31 @@ class IQRouterTest : public CppUnit::TestFixture delete channel_; } + void testRemoveHandler() { + IQRouter testling(channel_); + DummyIQHandler handler1(true, &testling); + DummyIQHandler handler2(true, &testling); + testling.removeHandler(&handler1); + + channel_->onIQReceived(boost::shared_ptr(new IQ())); + + CPPUNIT_ASSERT_EQUAL(0, handler1.called); + CPPUNIT_ASSERT_EQUAL(1, handler2.called); + } + + void testRemoveHandler_AfterHandleIQ() { + IQRouter testling(channel_); + DummyIQHandler handler1(true, &testling); + DummyIQHandler handler2(true, &testling); + + channel_->onIQReceived(boost::shared_ptr(new IQ())); + testling.removeHandler(&handler1); + channel_->onIQReceived(boost::shared_ptr(new IQ())); + + CPPUNIT_ASSERT_EQUAL(1, handler1.called); + CPPUNIT_ASSERT_EQUAL(1, handler2.called); + } + void testHandleIQ_SuccesfulHandlerFirst() { IQRouter testling(channel_); DummyIQHandler handler1(true, &testling); @@ -75,7 +102,7 @@ class IQRouterTest : public CppUnit::TestFixture CPPUNIT_ASSERT_EQUAL(1, handler1.called); CPPUNIT_ASSERT_EQUAL(2, handler2.called); } - + private: struct DummyIQHandler : public IQHandler { DummyIQHandler(bool handle, IQRouter* router) : handle(handle), router(router), called(0) { -- cgit v0.10.2-6-g49f6