AsioModule.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac

    This file is part of Leosac.

    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Leosac is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#include "BaseModule.hpp"
#include "tools/bs2.hpp"
#include "tools/service/ServiceRegistry.hpp"
#include <boost/asio/io_service.hpp>
#include <boost/asio/steady_timer.hpp>

namespace Leosac
{
namespace Module
{

class AsioModule : public BaseModule
{
  public:
    AsioModule(zmqpp::context &ctx, zmqpp::socket *pipe,
               const boost::property_tree::ptree &cfg, CoreUtilsPtr utils);

    ~AsioModule();

    virtual void run() override final;

    template <typename Callable>
    void post(Callable &&callable)
    {
        io_service_.post(std::forward<Callable>(callable));
    }

  protected:
    boost::asio::io_service io_service_;

    virtual void on_service_event(const service_event::Event &) = 0;

  private:
    std::unique_ptr<boost::asio::io_service::work> work_;
    bs2::connection service_event_listener_;

    void install_async_handlers();

    struct StopWatcher : public std::enable_shared_from_this<StopWatcher>
    {
        StopWatcher(AsioModule &self)
            : self_(self)
            , timer_(self.io_service_){};

        void schedule_wait();

      private:
        AsioModule &self_;
        boost::asio::steady_timer timer_;
        void wait(const boost::system::error_code &ec);
    };

    struct AsyncReactorPoller
        : public std::enable_shared_from_this<AsyncReactorPoller>
    {
        AsyncReactorPoller(AsioModule &self)
            : self_(self)
            , timer_(self.io_service_){};

        void schedule_wait();

      private:
        AsioModule &self_;
        boost::asio::steady_timer timer_;
        void wait_handler(const boost::system::error_code &ec);
    };
};
}
}