PipeSink.hpp

// SPDX-FileCopyrightText: 2025 Vincent Leroy
// SPDX-License-Identifier: MIT
//
// This file is part of dfx.
//
// Licensed under the MIT License. See the LICENSE file in the project root
// for full license information.
 
#pragma once
 
// Standard includes
#include <queue>
 
// Project includes
#include "Sink.hpp"
#include <dfx-fdwatch/PollerFd.hpp>
#include <dfx-utilities/FileSystem.hpp>
 
namespace dfx::Pcapng
{
class PipeSink : public Sink
{
public:
    PipeSink(fs::path path, uint32_t mode = 0600);

    ~PipeSink();

    void init(std::vector<uint8_t> initialHeader) override;

    void setMaxQueueSize(std::size_t maxQueueSize);

    std::size_t maxQueueSize() const noexcept { return _maxQueueSize; }
 
public:
    void write(std::vector<uint8_t> data) override;
 
private:
    void _handleWriteReady(FdWatch::EventTriggers events);
 
private:
    fs::path _pipePath;
    FdWatch::PollerFd _fifoFd;
 
private:
    bool _isPipeWriteReady = true;
    std::vector<uint8_t> _nextData;
    std::queue<std::vector<uint8_t>> _pendingData;
    std::size_t _maxQueueSize = 1024;
};
} // !namespace dfx::Pcapng