HttpSession.hpp

#pragma once
 
#include "data/LedgerCacheInterface.hpp"
#include "util/Taggable.hpp"
#include "web/AdminVerificationStrategy.hpp"
#include "web/PlainWsSession.hpp"
#include "web/ProxyIpResolver.hpp"
#include "web/dosguard/DOSGuardInterface.hpp"
#include "web/impl/HttpBase.hpp"
#include "web/interface/Concepts.hpp"
#include "web/interface/ConnectionBase.hpp"
 
#include <boost/asio/ip/tcp.hpp>
#include <boost/beast/core/error.hpp>
#include <boost/beast/core/flat_buffer.hpp>
#include <boost/beast/core/tcp_stream.hpp>
 
#include <cstdint>
#include <functional>
#include <memory>
#include <string>
#include <utility>
 
namespace web {
 
using tcp = boost::asio::ip::tcp;

template <SomeServerHandler HandlerType>
class HttpSession : public impl::HttpBase<HttpSession, HandlerType>,
                    public std::enable_shared_from_this<HttpSession<HandlerType>> {
    boost::beast::tcp_stream stream_;
    std::reference_wrapper<util::TagDecoratorFactory const> tagFactory_;
    std::uint32_t maxWsSendingQueueSize_;
 
public:
    explicit HttpSession(
        tcp::socket&& socket,
        std::string const& ip,
        std::shared_ptr<AdminVerificationStrategy> const& adminVerification,
        std::shared_ptr<ProxyIpResolver> proxyIpResolver,
        std::reference_wrapper<util::TagDecoratorFactory const> tagFactory,
        std::reference_wrapper<dosguard::DOSGuardInterface> dosGuard,
        std::shared_ptr<HandlerType> const& handler,
        std::reference_wrapper<data::LedgerCacheInterface const> cache,
        boost::beast::flat_buffer buffer,
        std::uint32_t maxWsSendingQueueSize
    )
        : impl::HttpBase<HttpSession, HandlerType>(
              ip,
              tagFactory,
              adminVerification,
              std::move(proxyIpResolver),
              dosGuard,
              handler,
              cache,
              std::move(buffer)
          )
        , stream_(std::move(socket))
        , tagFactory_(tagFactory)
        , maxWsSendingQueueSize_(maxWsSendingQueueSize)
    {
    }
 
    ~HttpSession() override = default;

    boost::beast::tcp_stream&
    stream()
    {
        return stream_;
    }

    void
    run()
    {
        boost::asio::dispatch(
            stream_.get_executor(),
            boost::beast::bind_front_handler(
                &impl::HttpBase<HttpSession, HandlerType>::doRead, this->shared_from_this()
            )
        );
    }

    void
    doClose()
    {
        boost::beast::error_code ec;
        stream_.socket().shutdown(tcp::socket::shutdown_send, ec);
    }

    void
    upgrade()
    {
        std::make_shared<WsUpgrader<HandlerType>>(
            std::move(stream_),
            this->clientIp_,
            tagFactory_,
            this->dosGuard_,
            this->handler_,
            std::move(this->buffer_),
            std::move(this->req_),
            ConnectionBase::isAdmin(),
            maxWsSendingQueueSize_
        )
            ->run();
    }
};
}  // namespace web