collection_set.hpp

Go to the documentation of this file.

#ifndef IROHA_COLLECTION_SET_HPP
#define IROHA_COLLECTION_SET_HPP

#include <shared_mutex>
#include <unordered_set>

namespace iroha {
  namespace set {

    template <typename Key,
              typename Hash = std::hash<Key>,
              typename KeyEqual = std::equal_to<Key>>
    class CollectionSet {
     public:
      CollectionSet() = default;

      using value_type = Key;

      template <typename Collection>
      void insertValues(Collection &&collection) {
        std::lock_guard<std::shared_timed_mutex> lock(mutex_);
        set_.insert(collection.begin(), collection.end());
      }

      template <typename Collection>
      void removeValues(Collection &&collection) {
        std::lock_guard<std::shared_timed_mutex> lock(mutex_);
        for (auto &&val : collection) {
          set_.erase(val);
        }
      }

      template <typename Callable>
      void forEach(Callable &&callable) const {
        std::shared_lock<std::shared_timed_mutex> lock(mutex_);
        std::for_each(set_.begin(), set_.end(), callable);
      }

     private:
      std::unordered_set<Key, Hash, KeyEqual> set_;
      mutable std::shared_timed_mutex mutex_;
    };
  }  // namespace set
}  // namespace iroha

#endif  // IROHA_COLLECTION_SET_HPP