Exporter.hpp

Go to the documentation of this file.

#pragma once
 
#include <chrono>
#include <condition_variable>
#include <mutex>
#include <queue>
#include <thread>
#include <tuple>
 
#include <asiolink/io_address.h>
#include <dhcp/hwaddr.h>
#include <dhcpsrv/lease.h>
#include <log/logger.h>
 
class HiredisAdapter;
class ArpwatchRPC;
 
namespace arpwatch
{
class Exporter
{
public:
    using mac_addr_t = isc::dhcp::HWAddr;
    using ip4_addr_t = isc::asiolink::IOAddress;
    using subnet_id_t = isc::dhcp::SubnetID;
    using lease4_t = std::tuple<mac_addr_t, ip4_addr_t, subnet_id_t, std::chrono::seconds>;
 
    constexpr static std::chrono::seconds REDIS_RECONNECT_WAIT{5};
    constexpr static std::chrono::seconds DEFAULT_VALID_LFT{600};
    constexpr static double REDIS_EXPIRE_FACT = 1.5F;
    constexpr static std::chrono::milliseconds LEASE_QUEUE_FLUSH_WAIT{10};
 
    explicit inline Exporter(HiredisAdapter& redis_adapter, ArpwatchRPC& arpwatch_rpc)
        : redis_adapter(redis_adapter), arpwatch_rpc(arpwatch_rpc){};
    Exporter(const Exporter&) = delete;
    Exporter& operator=(const Exporter&) = delete;
    Exporter(Exporter&&) = delete;
    Exporter& operator=(Exporter&&) = delete;
    ~Exporter();
 
    void start_worker();
    void stop_worker();
 
    void push_lease(const mac_addr_t& mac, const ip4_addr_t& ip4, subnet_id_t subnet_id = 0,
                    std::chrono::seconds valid_lft = DEFAULT_VALID_LFT);
    void flush_lease_queue();
 
private:
    void run_worker();
 
    void worker_send_lease_to_redis(const lease4_t& lease);
    void worker_send_lease_to_arpwatch(const lease4_t& lease);
 
    std::unique_ptr<std::thread> worker;
    volatile bool worker_keep_running{false};
 
    std::condition_variable worker_notify_cv;
    std::mutex lease_queue_mtx;
    std::queue<lease4_t> lease_queue;
 
    HiredisAdapter& redis_adapter;
    ArpwatchRPC& arpwatch_rpc;
 
    isc::log::Logger logger{"arpwatch-exporter.Exporter"};
};
} // namespace arpwatch