ping_service.h

Go to the documentation of this file.

#pragma once
 
#include "util/rats_export.h"
#include "node/peer_network.h"
#include "peer/peer.h"
#include "peer/peer_id.h"
 
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <cstdint>
#include <mutex>
#include <optional>
#include <thread>
#include <unordered_map>
 
namespace librats {
 
class RATS_API PingService final : public Subsystem {
public:
    explicit PingService(std::chrono::milliseconds interval = std::chrono::seconds(10));
    ~PingService() override;
 
    void attach(NodeContext& ctx) override;
    void start() override;
    void stop() override;
 
    std::optional<std::chrono::milliseconds> last_rtt(const PeerId& id) const;
 
    size_t alive_peer_count() const;
 
private:
    void run();                                       
    void handle(const Peer& peer, ByteView payload);
    void ping_all();
 
    PeerNetwork*              network_ = nullptr;
    std::chrono::milliseconds interval_;
 
    std::thread             thread_;
    std::atomic<bool>       running_{false};
    std::mutex              wait_mutex_;
    std::condition_variable wake_;
 
    mutable std::mutex mutex_;
    std::unordered_map<PeerId, std::chrono::milliseconds, PeerId::Hash> rtt_;
};
 
} // namespace librats