sendfile_operation.hpp

Go to the documentation of this file.

/*
    restinio
*/

/*!
    sendfile routine.
*/

#pragma once

#include <memory>

#include <restinio/sendfile.hpp>

namespace restinio
{

namespace impl
{

//
// sendfile_operation_base_t
//

//!Base class for storing sendfile operation context.
class sendfile_operation_base_t
    :   public std::enable_shared_from_this< sendfile_operation_base_t >
{
    public:
        virtual ~sendfile_operation_base_t() = default;

        virtual void
        start() = 0;
};

using sendfile_operation_shared_ptr_t = std::shared_ptr< sendfile_operation_base_t >;

//! Callback type for invocation when sendfile operation completes.
using after_sendfile_cb_t =
    std::function< void ( const asio_ns::error_code & , file_size_t ) >;

//
// sendfile_operation_runner_base_t
//

//! A base runner of sendfile operation (keeps all the data).
template < typename Socket >
class sendfile_operation_runner_base_t
    :   public sendfile_operation_base_t
{
    public:
        sendfile_operation_runner_base_t() = delete;

        sendfile_operation_runner_base_t(
            const sendfile_t & sf,
            default_asio_executor executor,
            Socket & socket,
            after_sendfile_cb_t after_sendfile_cb )
            :   m_file_descriptor{ sf.file_descriptor() }
            ,   m_next_write_offset{ sf.offset() }
            ,   m_remained_size{ sf.size() }
            ,   m_chunk_size{ sf.chunk_size() }
            ,   m_expires_after{ std::chrono::steady_clock::now() + sf.timelimit() }
            ,   m_executor{ std::move( executor )}
            ,   m_socket{ socket }
            ,   m_after_sendfile_cb{ std::move( after_sendfile_cb ) }
        {}

        auto expires_after() const noexcept { return m_expires_after; }

    protected:
        file_descriptor_t m_file_descriptor;
        file_offset_t m_next_write_offset;
        file_size_t m_remained_size;
        file_size_t m_transfered_size{ 0 };

        const file_size_t m_chunk_size;

        const std::chrono::steady_clock::time_point m_expires_after;

        default_asio_executor m_executor;
        Socket & m_socket;
        after_sendfile_cb_t m_after_sendfile_cb;
};

template<typename Error_Type>
auto
make_error_code( const Error_Type & e ) noexcept
{
    return asio_ns::error_code{ static_cast<int>(e), asio_ns::error::get_system_category() };
}

} /* namespace impl */

} /* namespace restinio */

/*
    Concrete implementations.
*/

#if defined(RESTINIO_OS_WINDOWS)
    #include "sendfile_operation_win.ipp"
#elif defined(RESTINIO_OS_UNIX) || defined(RESTINIO_OS_APPLE)
    #include "sendfile_operation_posix.ipp"
#else
    #if defined (RESTINIO_ENABLE_SENDFILE_DEFAULT_IMPL)
        #include "sendfile_operation_default.ipp"
    #else
        #error "Sendfile not supported, to enable default implementation define RESTINIO_ENABLE_SENDFILE_DEFAULT_IMPL macro"
    #endif
#endif