SubscriptionContext.hpp

#pragma once
 
#include "util/CoroutineGroup.hpp"
#include "util/Taggable.hpp"
#include "web/SubscriptionContextInterface.hpp"
#include "web/ng/Connection.hpp"
#include "web/ng/Error.hpp"
#include "web/ng/impl/WsConnection.hpp"
 
#include <boost/asio/any_io_executor.hpp>
#include <boost/asio/spawn.hpp>
#include <boost/signals2/variadic_signal.hpp>
 
#include <atomic>
#include <cstddef>
#include <cstdint>
#include <functional>
#include <memory>
#include <optional>
#include <string>
 
namespace web::ng {

class SubscriptionContext : public web::SubscriptionContextInterface {
public:
    using ErrorHandler = std::function<bool(Error const&, Connection const&)>;
 
private:
    std::reference_wrapper<impl::WsConnectionBase> connection_;
    std::optional<size_t> maxSendQueueSize_;
    util::CoroutineGroup tasksGroup_;
    boost::asio::yield_context yield_;
    ErrorHandler errorHandler_;
 
    boost::signals2::signal<void(SubscriptionContextInterface*)> onDisconnect_;
    std::atomic_bool disconnected_{false};
    std::atomic_bool gotError_{false};

    std::atomic_uint32_t apiSubversion_ = 0u;
 
public:
    SubscriptionContext(
        util::TagDecoratorFactory const& factory,
        impl::WsConnectionBase& connection,
        std::optional<size_t> maxSendQueueSize,
        boost::asio::yield_context yield,
        ErrorHandler errorHandler
    );
 
    ~SubscriptionContext() override;

    void
    send(std::shared_ptr<std::string> message) override;

    void
    onDisconnect(OnDisconnectSlot const& slot) override;

    void
    setApiSubversion(uint32_t value) override;

    uint32_t
    apiSubversion() const override;

    void
    disconnect(boost::asio::yield_context yield);
};
 
}  // namespace web::ng