Merge pull request #4785 from pwojcikdev/traffic-info

Per traffic type stats

nano/core_test/CMakeLists.txt

@@ -42,6 +42,7 @@ add_executable(
   processor_service.cpp
   random.cpp
   random_pool.cpp
+  rate_limiting.cpp
   rep_crawler.cpp
   receivable.cpp
   peer_history.cpp

nano/core_test/rate_limiting.cpp

@@ -0,0 +1,113 @@
+#include <nano/lib/rate_limiting.hpp>
+#include <nano/lib/utility.hpp>
+
+#include <gtest/gtest.h>
+
+#include <fstream>
+#include <future>
+
+using namespace std::chrono_literals;
+
+TEST (rate, basic)
+{
+	nano::rate::token_bucket bucket (10, 10);
+
+	// Initial burst
+	ASSERT_TRUE (bucket.try_consume (10));
+	ASSERT_FALSE (bucket.try_consume (10));
+
+	// With a fill rate of 10 tokens/sec, await 1/3 sec and get 3 tokens
+	std::this_thread::sleep_for (300ms);
+	ASSERT_TRUE (bucket.try_consume (3));
+	ASSERT_FALSE (bucket.try_consume (10));
+
+	// Allow time for the bucket to completely refill and do a full burst
+	std::this_thread::sleep_for (1s);
+	ASSERT_TRUE (bucket.try_consume (10));
+	ASSERT_EQ (bucket.largest_burst (), 10);
+}
+
+TEST (rate, network)
+{
+	// For the purpose of the test, one token represents 1MB instead of one byte.
+	// Allow for 10 mb/s bursts (max bucket size), 5 mb/s long term rate
+	nano::rate::token_bucket bucket (10, 5);
+
+	// Initial burst of 10 mb/s over two calls
+	ASSERT_TRUE (bucket.try_consume (5));
+	ASSERT_EQ (bucket.largest_burst (), 5);
+	ASSERT_TRUE (bucket.try_consume (5));
+	ASSERT_EQ (bucket.largest_burst (), 10);
+	ASSERT_FALSE (bucket.try_consume (5));
+
+	// After 200 ms, the 5 mb/s fillrate means we have 1 mb available
+	std::this_thread::sleep_for (200ms);
+	ASSERT_TRUE (bucket.try_consume (1));
+	ASSERT_FALSE (bucket.try_consume (1));
+}
+
+TEST (rate, reset)
+{
+	nano::rate::token_bucket bucket (0, 0);
+
+	// consume lots of tokens, buckets should be unlimited
+	ASSERT_TRUE (bucket.try_consume (1000000));
+	ASSERT_TRUE (bucket.try_consume (1000000));
+
+	// set bucket to be limited
+	bucket.reset (1000, 1000);
+	ASSERT_FALSE (bucket.try_consume (1001));
+	ASSERT_TRUE (bucket.try_consume (1000));
+	ASSERT_FALSE (bucket.try_consume (1000));
+	std::this_thread::sleep_for (2ms);
+	ASSERT_TRUE (bucket.try_consume (2));
+
+	// reduce the limit
+	bucket.reset (100, 100 * 1000);
+	ASSERT_FALSE (bucket.try_consume (101));
+	ASSERT_TRUE (bucket.try_consume (100));
+	std::this_thread::sleep_for (1ms);
+	ASSERT_TRUE (bucket.try_consume (100));
+
+	// increase the limit
+	bucket.reset (2000, 1);
+	ASSERT_FALSE (bucket.try_consume (2001));
+	ASSERT_TRUE (bucket.try_consume (2000));
+
+	// back to unlimited
+	bucket.reset (0, 0);
+	ASSERT_TRUE (bucket.try_consume (1000000));
+	ASSERT_TRUE (bucket.try_consume (1000000));
+}
+
+TEST (rate, unlimited)
+{
+	nano::rate::token_bucket bucket (0, 0);
+	ASSERT_TRUE (bucket.try_consume (5));
+	ASSERT_EQ (bucket.largest_burst (), 5);
+	ASSERT_TRUE (bucket.try_consume (static_cast<size_t> (1e9)));
+	ASSERT_EQ (bucket.largest_burst (), static_cast<size_t> (1e9));
+
+	// With unlimited tokens, consuming always succeed
+	ASSERT_TRUE (bucket.try_consume (static_cast<size_t> (1e9)));
+	ASSERT_EQ (bucket.largest_burst (), static_cast<size_t> (1e9));
+}
+
+TEST (rate, busy_spin)
+{
+	// Bucket should refill at a rate of 1 token per second
+	nano::rate::token_bucket bucket (1, 1);
+
+	// Run a very tight loop for 5 seconds + a bit of wiggle room
+	int counter = 0;
+	for (auto start = std::chrono::steady_clock::now (), now = start; now < start + 5500ms; now = std::chrono::steady_clock::now ())
+	{
+		if (bucket.try_consume ())
+		{
+			++counter;
+		}
+	}
+
+	// Bucket starts fully refilled, therefore we see 1 additional request
+	ASSERT_EQ (counter, 6);
+}

nano/core_test/utility.cpp

@@ -15,110 +15,6 @@
 
 using namespace std::chrono_literals;
 
-TEST (rate, basic)
-{
-	nano::rate::token_bucket bucket (10, 10);
-
-	// Initial burst
-	ASSERT_TRUE (bucket.try_consume (10));
-	ASSERT_FALSE (bucket.try_consume (10));
-
-	// With a fill rate of 10 tokens/sec, await 1/3 sec and get 3 tokens
-	std::this_thread::sleep_for (300ms);
-	ASSERT_TRUE (bucket.try_consume (3));
-	ASSERT_FALSE (bucket.try_consume (10));
-
-	// Allow time for the bucket to completely refill and do a full burst
-	std::this_thread::sleep_for (1s);
-	ASSERT_TRUE (bucket.try_consume (10));
-	ASSERT_EQ (bucket.largest_burst (), 10);
-}
-
-TEST (rate, network)
-{
-	// For the purpose of the test, one token represents 1MB instead of one byte.
-	// Allow for 10 mb/s bursts (max bucket size), 5 mb/s long term rate
-	nano::rate::token_bucket bucket (10, 5);
-
-	// Initial burst of 10 mb/s over two calls
-	ASSERT_TRUE (bucket.try_consume (5));
-	ASSERT_EQ (bucket.largest_burst (), 5);
-	ASSERT_TRUE (bucket.try_consume (5));
-	ASSERT_EQ (bucket.largest_burst (), 10);
-	ASSERT_FALSE (bucket.try_consume (5));
-
-	// After 200 ms, the 5 mb/s fillrate means we have 1 mb available
-	std::this_thread::sleep_for (200ms);
-	ASSERT_TRUE (bucket.try_consume (1));
-	ASSERT_FALSE (bucket.try_consume (1));
-}
-
-TEST (rate, reset)
-{
-	nano::rate::token_bucket bucket (0, 0);
-
-	// consume lots of tokens, buckets should be unlimited
-	ASSERT_TRUE (bucket.try_consume (1000000));
-	ASSERT_TRUE (bucket.try_consume (1000000));
-
-	// set bucket to be limited
-	bucket.reset (1000, 1000);
-	ASSERT_FALSE (bucket.try_consume (1001));
-	ASSERT_TRUE (bucket.try_consume (1000));
-	ASSERT_FALSE (bucket.try_consume (1000));
-	std::this_thread::sleep_for (2ms);
-	ASSERT_TRUE (bucket.try_consume (2));
-
-	// reduce the limit
-	bucket.reset (100, 100 * 1000);
-	ASSERT_FALSE (bucket.try_consume (101));
-	ASSERT_TRUE (bucket.try_consume (100));
-	std::this_thread::sleep_for (1ms);
-	ASSERT_TRUE (bucket.try_consume (100));
-
-	// increase the limit
-	bucket.reset (2000, 1);
-	ASSERT_FALSE (bucket.try_consume (2001));
-	ASSERT_TRUE (bucket.try_consume (2000));
-
-	// back to unlimited
-	bucket.reset (0, 0);
-	ASSERT_TRUE (bucket.try_consume (1000000));
-	ASSERT_TRUE (bucket.try_consume (1000000));
-}
-
-TEST (rate, unlimited)
-{
-	nano::rate::token_bucket bucket (0, 0);
-	ASSERT_TRUE (bucket.try_consume (5));
-	ASSERT_EQ (bucket.largest_burst (), 5);
-	ASSERT_TRUE (bucket.try_consume (static_cast<size_t> (1e9)));
-	ASSERT_EQ (bucket.largest_burst (), static_cast<size_t> (1e9));
-
-	// With unlimited tokens, consuming always succeed
-	ASSERT_TRUE (bucket.try_consume (static_cast<size_t> (1e9)));
-	ASSERT_EQ (bucket.largest_burst (), static_cast<size_t> (1e9));
-}
-
-TEST (rate, busy_spin)
-{
-	// Bucket should refill at a rate of 1 token per second
-	nano::rate::token_bucket bucket (1, 1);
-
-	// Run a very tight loop for 5 seconds + a bit of wiggle room
-	int counter = 0;
-	for (auto start = std::chrono::steady_clock::now (), now = start; now < start + std::chrono::milliseconds{ 5500 }; now = std::chrono::steady_clock::now ())
-	{
-		if (bucket.try_consume ())
-		{
-			++counter;
-		}
-	}
-
-	// Bucket starts fully refilled, therefore we see 1 additional request
-	ASSERT_EQ (counter, 6);
-}
-
 TEST (optional_ptr, basic)
 {
 	struct valtype

nano/lib/rate_limiting.cpp

@@ -45,11 +45,6 @@ void nano::rate::token_bucket::refill ()
 	}
 }
 
-std::size_t nano::rate::token_bucket::largest_burst () const
-{
-	return max_token_count - smallest_size;
-}
-
 void nano::rate::token_bucket::reset (std::size_t max_token_count_a, std::size_t refill_rate_a)
 {
 	// A token count of 0 indicates unlimited capacity. We use 1e9 as
@@ -63,6 +58,16 @@ void nano::rate::token_bucket::reset (std::size_t max_token_count_a, std::size_t
 	last_refill = std::chrono::steady_clock::now ();
 }
 
+std::size_t nano::rate::token_bucket::largest_burst () const
+{
+	return max_token_count - smallest_size;
+}
+
+std::size_t nano::rate::token_bucket::size () const
+{
+	return current_size;
+}
+
 /*
  * rate_limiter
  */
@@ -82,4 +87,10 @@ void nano::rate_limiter::reset (std::size_t limit_a, double burst_ratio_a)
 {
 	nano::lock_guard<nano::mutex> guard{ mutex };
 	bucket.reset (static_cast<std::size_t> (limit_a * burst_ratio_a), limit_a);
+}
+
+std::size_t nano::rate_limiter::size () const
+{
+	nano::lock_guard<nano::mutex> guard{ mutex };
+	return bucket.size ();
 }

nano/lib/rate_limiting.hpp

@@ -38,12 +38,13 @@ class token_bucket
 	 */
 	bool try_consume (unsigned tokens_required = 1);
 
-	/** Returns the largest burst observed */
-	std::size_t largest_burst () const;
-
 	/** Update the max_token_count and/or refill_rate_a parameters */
 	void reset (std::size_t max_token_count, std::size_t refill_rate);
 
+	/** Returns the largest burst observed */
+	std::size_t largest_burst () const;
+	std::size_t size () const;
+
 private:
 	void refill ();
 
@@ -71,6 +72,8 @@ class rate_limiter final
 	bool should_pass (std::size_t buffer_size);
 	void reset (std::size_t limit, double burst_ratio = 1.0);
 
+	std::size_t size () const;
+
 private:
 	nano::rate::token_bucket bucket;
 	mutable nano::mutex mutex;

nano/lib/stats_enums.hpp

@@ -13,14 +13,12 @@ enum class type
 	_invalid = 0, // Default value, should not be used
 
 	test,
-	traffic_tcp,
 	error,
 	message,
 	block,
 	ledger,
 	rollback,
 	network,
-	tcp_server,
 	vote,
 	vote_processor,
 	vote_processor_tier,
@@ -31,11 +29,14 @@ enum class type
 	http_callback,
 	ipc,
 	tcp,
+	tcp_server,
 	tcp_channels,
 	tcp_channels_rejected,
 	tcp_channels_purge,
 	tcp_listener,
 	tcp_listener_rejected,
+	traffic_tcp,
+	traffic_tcp_type,
 	channel,
 	socket,
 	confirmation_height,
@@ -294,6 +295,9 @@ enum class detail
 	reachout_live,
 	reachout_cached,
 
+	// traffic
+	generic,
+
 	// tcp
 	tcp_write_drop,
 	tcp_write_no_socket_drop,

nano/node/CMakeLists.txt

@@ -166,6 +166,8 @@ add_library(
   transport/tcp_server.cpp
   transport/tcp_socket.hpp
   transport/tcp_socket.cpp
+  transport/traffic_type.hpp
+  transport/traffic_type.cpp
   transport/transport.hpp
   transport/transport.cpp
   unchecked_map.cpp

nano/node/bandwidth_limiter.cpp

@@ -41,6 +41,14 @@ void nano::bandwidth_limiter::reset (std::size_t limit, double burst_ratio, nano
 	limiter.reset (limit, burst_ratio);
 }
 
+nano::container_info nano::bandwidth_limiter::container_info () const
+{
+	nano::container_info info;
+	info.put ("generic", limiter_generic.size ());
+	info.put ("bootstrap", limiter_bootstrap.size ());
+	return info;
+}
+
 /*
  * bandwidth_limiter_config
  */

nano/node/bandwidth_limiter.hpp

@@ -37,6 +37,8 @@ class bandwidth_limiter final
 	 */
 	void reset (std::size_t limit, double burst_ratio, nano::transport::traffic_type type = nano::transport::traffic_type::generic);
 
+	nano::container_info container_info () const;
+
 private:
 	/**
 	 * Returns reference to limiter corresponding to the limit type

nano/node/node.cpp

@@ -1217,6 +1217,7 @@ nano::container_info nano::node::container_info () const
 	info.add ("local_block_broadcaster", local_block_broadcaster.container_info ());
 	info.add ("rep_tiers", rep_tiers.container_info ());
 	info.add ("message_processor", message_processor.container_info ());
+	info.add ("bandwidth", outbound_limiter.container_info ());
 	return info;
 }