ws_listener_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_CORE_API_TRANSPORT_IMPL_WS_LISTENER_IMPL_HPP
#define KAGOME_CORE_API_TRANSPORT_IMPL_WS_LISTENER_IMPL_HPP

#include "api/transport/listener.hpp"

#include <atomic>

#include "api/transport/impl/ws/ws_session.hpp"
#include "log/logger.hpp"
#include "metrics/metrics.hpp"

namespace kagome::application {
  class AppStateManager;
}

namespace kagome::api {
  class WsListenerImpl : public Listener,
                         public std::enable_shared_from_this<WsListenerImpl> {
   public:
    using SessionImpl = WsSession;

    WsListenerImpl(
        const std::shared_ptr<application::AppStateManager> &app_state_manager,
        std::shared_ptr<Context> context,
        Configuration listener_config,
        SessionImpl::Configuration session_config);

    ~WsListenerImpl() override = default;

    bool prepare() override;

    bool start() override;

    void stop() override;

    void setHandlerForNewSession(NewSessionHandler &&on_new_session) override;

   private:
    void acceptOnce() override;

    std::shared_ptr<Context> context_;
    const Configuration config_;
    const SessionImpl::Configuration session_config_;
    const uint32_t max_ws_connections_;

    std::unique_ptr<Acceptor> acceptor_;
    std::unique_ptr<NewSessionHandler> on_new_session_;

    std::atomic<Session::SessionId> next_session_id_;
    std::shared_ptr<SessionImpl> new_session_;
    std::atomic<uint32_t> active_connections_;

    // metrics
    metrics::RegistryPtr registry_ = metrics::createRegistry();
    metrics::Counter *opened_session_;
    metrics::Counter *closed_session_;

    log::Logger log_;
  };

}  // namespace kagome::api

#endif  // KAGOME_CORE_API_TRANSPORT_IMPL_WS_LISTENER_IMPL_HPP