link_server.h

#pragma once
 
//
// Copyright (c) 2019-2023 Ivan Baidakou (basiliscos) (the dot dmol at gmail dot com)
//
// Distributed under the MIT Software License
//
 
#include "plugin_base.h"
#include <string>
#include <unordered_set>
#include <functional>
#include <optional>
 
#if defined(_MSC_VER)
#pragma warning(push)
#pragma warning(disable : 4251)
#endif
 
namespace rotor::plugin {

struct ROTOR_API link_server_plugin_t : public plugin_base_t {
    using plugin_base_t::plugin_base_t;

    static const std::type_index class_identity;
 
    const std::type_index &identity() const noexcept override;
 
    void activate(actor_base_t *actor) noexcept override;

    virtual void on_link_request(message::link_request_t &message) noexcept;

    virtual void on_unlink_response(message::unlink_response_t &message) noexcept;

    virtual void on_unlink_notify(message::unlink_notify_t &message) noexcept;
 
    bool handle_shutdown(message::shutdown_request_t *message) noexcept override;
    void handle_start(message::start_trigger_t *message) noexcept override;

    virtual bool has_clients() noexcept { return !linked_clients.empty(); }

    virtual void notify_shutdown() noexcept;

    template <typename T> auto &access() noexcept;
 
  private:
    enum class link_state_t { PENDING, OPERATIONAL, UNLINKING };
    using link_request_ptr_t = intrusive_ptr_t<message::link_request_t>;
    using request_option_t = std::optional<request_id_t>;
    struct link_info_t {
        link_info_t(link_state_t state_, link_request_ptr_t request_) noexcept : state{state_}, request{request_} {}
        link_state_t state;
        link_request_ptr_t request;
        request_option_t unlink_request;
        bool shutdown_notified = false;
    };
 
    using linked_clients_t = std::unordered_map<address_ptr_t, link_info_t>;
    linked_clients_t linked_clients;
};
 
} // namespace rotor::plugin
 
#if defined(_MSC_VER)
#pragma warning(pop)
#endif