xbot/watchdog_thread.cpp

#include "watchdog_thread.hpp"

#include "connection.hpp"
#include "irc_parse_thread.hpp"
#include "write_irc.hpp"

#include <boost/asio/steady_timer.hpp>

#include <chrono>
#include <memory>

WatchdogThread::WatchdogThread(Connection& connection)
: connection_{connection}
, timer_{connection.get_executor()}
, stalled_{false}
{
}

auto WatchdogThread::on_activity() -> void
{
    stalled_ = false;
    timer_.expires_from_now(WatchdogThread::TIMEOUT);
}

auto WatchdogThread::start_timer()
{
    timer_.async_wait([weak = weak_from_this()](auto const& error)
    {
        if (not error)
        {
            if (auto self = weak.lock())
            {
                self->on_timeout();
            }
        }
    });
}

auto WatchdogThread::on_timeout() -> void
{
    if (stalled_)
    {
        connection_.close();
    }
    else
    {
        send_ping(connection_, "watchdog");
        stalled_ = true;
        timer_.expires_from_now(WatchdogThread::TIMEOUT);
        start_timer();
    }
}

auto WatchdogThread::on_connect() -> void
{
    on_activity();
    start_timer();
}

auto WatchdogThread::on_disconnect() -> void
{
    timer_.cancel();
}

auto WatchdogThread::start(Connection& connection) -> void
{
    auto const thread = std::make_shared<WatchdogThread>(connection);
    connection.add_listener<ConnectEvent>([thread](auto&)
    {
        thread->on_connect();
    });
    connection.add_listener<DisconnectEvent>([thread](auto&)
    {
        thread->on_disconnect();
    });
    connection.add_listener<IrcMsgEvent>([thread](auto&)
    {
        thread->on_activity();
    });
}