wanderer/ fastnetmon-rewritten
1
0
Fork 0
mirror of https://github.com/pavel-odintsov/fastnetmon synced 2024-05-22 16:26:25 +02:00
Code Issues Releases Activity

Added template option to configure type of data structure

This commit is contained in:
Pavel Odintsov 2023-07-11 13:36:35 +01:00
parent f5b7cd8e31
commit 3738ef9f8f
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/abstract_subnet_counters.hpp
View File

@ -1,7 +1,7 @@
#pragma once #pragma once
#include <functional>
#include <mutex> #include <mutex>
#include <functional>
#include "speed_counters.hpp" #include "speed_counters.hpp"
@ -19,14 +19,14 @@
#include <boost/serialization/unordered_map.hpp> #include <boost/serialization/unordered_map.hpp>
// Class for abstract per key counters // Class for abstract per key counters
template <typename T, typename Counter> class abstract_subnet_counters_t { template <typename T, typename Counter, typename UM = std::unordered_map<T, Counter>> class abstract_subnet_counters_t {
public: public:
std::unordered_map<T, Counter> counter_map; UM counter_map;
std::mutex counter_map_mutex; std::mutex counter_map_mutex;
std::unordered_map<T, Counter> average_speed_map; UM average_speed_map;
// By using single map for speed and data we can accomplish imprevement from 3-4 seconds for 14m hosts to 2-3 seconds // By using single map for speed and data we can accomplish improvement from 3-4 seconds for 14m hosts to 2-3 seconds
template <class Archive> void serialize(Archive& ar, [[maybe_unused]] const unsigned int version) { template <class Archive> void serialize(Archive& ar, [[maybe_unused]] const unsigned int version) {
ar& BOOST_SERIALIZATION_NVP(counter_map); ar& BOOST_SERIALIZATION_NVP(counter_map);
@ -102,7 +102,7 @@ template <typename T, typename Counter> class abstract_subnet_counters_t {
} }
// Retrieves all elements // Retrieves all elements
void get_all_average_speed_elements(std::unordered_map<T, Counter>& copy_of_average_speed_map) { void get_all_average_speed_elements(UM& copy_of_average_speed_map) {
std::lock_guard<std::mutex> lock_guard(counter_map_mutex); std::lock_guard<std::mutex> lock_guard(counter_map_mutex);
copy_of_average_speed_map = this->average_speed_map; copy_of_average_speed_map = this->average_speed_map;