ConnectionManager.hpp

Go to the documentation of this file.

#pragma once
#include <orbit/server/Connection.hpp>
#include <orbit/network/Proactor.hpp>
#include <orbit/routing/Router.hpp>
#include <orbit/concurrency/ThreadPool.hpp>
#include <orbit/server/TimerManager.hpp>
#include <orbit/network/TlsContext.hpp>
#include <unordered_map>
#include <memory>
#include <mutex>
 
namespace server {
 
class ConnectionManager {
public:
    ConnectionManager(network::Proactor& proactor, const routing::Router& router, concurrency::ThreadPool& thread_pool, TimerManager& timer_manager, size_t max_body_size, network::TlsContext* tls_context = nullptr);
 
    void add_connection(network::Socket socket, const std::string& client_ip);
    
    void remove_connection(int fd);
    
    size_t get_connection_count() const;
 
private:
    network::Proactor& proactor_;
    const routing::Router& router_;
    concurrency::ThreadPool& thread_pool_;
    TimerManager& timer_manager_;
    
    std::unordered_map<int, std::shared_ptr<Connection>> connections_;
    mutable std::mutex map_mutex_;
    size_t max_body_size_;
    network::TlsContext* tls_context_;
};
 
} // namespace server