Per-client connection limiter. More...

#include <rate_limiter.h>

Collaboration diagram for kcenon::network::per_client_connection_limiter:

Public Member Functions
	per_client_connection_limiter (size_t max_per_client=10, size_t max_total=1000)
	Construct limiter.

bool	try_accept (std::string_view client_id)
	Try to accept a connection from a client.

void	release (std::string_view client_id)
	Release a connection from a client.

size_t	client_connections (std::string_view client_id) const
	Get connection count for a client.

size_t	total_connections () const noexcept
	Get total connection count.

Private Attributes
size_t	max_per_client_

connection_limiter	total_limiter_

std::mutex	mutex_

std::unordered_map< std::string, size_t >	client_connections_

Detailed Description

Per-client connection limiter.

Limits connections per client identifier (e.g., IP address).

Definition at line 481 of file rate_limiter.h.

Constructor & Destructor Documentation

◆ per_client_connection_limiter()

kcenon::network::per_client_connection_limiter::per_client_connection_limiter	(	size_t	max_per_client = 10,
		size_t	max_total = 1000 )

inlineexplicit

Construct limiter.

Parameters

max_per_client	Maximum connections per client
max_total	Maximum total connections

Examples: /home/runner/work/network_system/network_system/src/internal/utils/rate_limiter.h.

Definition at line 488 of file rate_limiter.h.

491 : max_per_client_(max_per_client)

492 , total_limiter_(max_total) {}

kcenon::network::per_client_connection_limiter::total_limiter_

connection_limiter total_limiter_

Definition rate_limiter.h:560

kcenon::network::per_client_connection_limiter::max_per_client_

size_t max_per_client_

Definition rate_limiter.h:559

Member Function Documentation

◆ client_connections()

size_t kcenon::network::per_client_connection_limiter::client_connections ( std::string_view client_id ) const

inlinenodiscard

Get connection count for a client.

Parameters

client_id Client identifier

Returns: Number of connections from client

Examples: /home/runner/work/network_system/network_system/src/internal/utils/rate_limiter.h.

Definition at line 544 of file rate_limiter.h.

                                                                            {
        std::unique_lock<std::mutex> lock(mutex_);
        auto it = client_connections_.find(std::string(client_id));
        return (it != client_connections_.end()) ? it->second : 0;
    }

References client_connections_, and mutex_.

◆ release()

void kcenon::network::per_client_connection_limiter::release ( std::string_view client_id )

inline

Release a connection from a client.

Parameters

client_id Client identifier

Examples: /home/runner/work/network_system/network_system/src/internal/utils/rate_limiter.h.

Definition at line 525 of file rate_limiter.h.

                                           {
        std::unique_lock<std::mutex> lock(mutex_);
        std::string key(client_id);
 
        auto it = client_connections_.find(key);
        if (it != client_connections_.end() && it->second > 0) {
            --it->second;
            if (it->second == 0) {
                client_connections_.erase(it);
            }
            total_limiter_.on_disconnect();
        }
    }

References client_connections_, mutex_, kcenon::network::connection_limiter::on_disconnect(), and total_limiter_.

Here is the call graph for this function:

◆ total_connections()

size_t kcenon::network::per_client_connection_limiter::total_connections ( ) const

inlinenodiscardnoexcept

Get total connection count.

Returns: Total number of connections

Examples: /home/runner/work/network_system/network_system/src/internal/utils/rate_limiter.h.

Definition at line 554 of file rate_limiter.h.

                                                            {
        return total_limiter_.current();
    }

References kcenon::network::connection_limiter::current(), and total_limiter_.

Here is the call graph for this function:

◆ try_accept()

bool kcenon::network::per_client_connection_limiter::try_accept ( std::string_view client_id )

inlinenodiscard

Try to accept a connection from a client.

Parameters

client_id Client identifier

Returns: true if accepted

Examples: /home/runner/work/network_system/network_system/src/internal/utils/rate_limiter.h.

Definition at line 499 of file rate_limiter.h.

                                                            {
        // First check total limit
        if (!total_limiter_.can_accept()) {
            return false;
        }
 
        std::unique_lock<std::mutex> lock(mutex_);
        std::string key(client_id);
 
        auto& count = client_connections_[key];
        if (count >= max_per_client_) {
            return false;
        }
 
        if (!total_limiter_.try_accept()) {
            return false;
        }
 
        ++count;
        return true;
    }