SharedMap.hpp

Go to the documentation of this file.

#ifndef _IDENTT_UTILS_SHARED_MAP_HPP_
#define _IDENTT_UTILS_SHARED_MAP_HPP_

#include <unordered_map>
#include <utility>
#include <functional>
#include <boost/thread/locks.hpp>
#include <boost/thread/shared_mutex.hpp>

namespace identt {
namespace utils {
template<class KeyT,class ValueT>
class SharedMap {
public:
    using MapT = std::unordered_map<KeyT,ValueT>;
    using LockT = boost::shared_mutex;
    using WriteLockT = boost::unique_lock< LockT >;
    using ReadLockT = boost::shared_lock< LockT >;

    SharedMap(const SharedMap&) = delete;
    SharedMap& operator=(const SharedMap&) = delete;

    SharedMap() {}

    virtual ~SharedMap () {}

    void AddOne(KeyT key, ValueT value)
    {
        WriteLockT writelock(mutex_);
        t_[key]=value;
    }

    void AddMany(const MapT& m)
    {
        WriteLockT writelock(mutex_);
        for (auto& p : m) {
            t_[p.first]=p.second;
        }
    }

    void AddMany(MapT&& m)
    {
        WriteLockT writelock(mutex_);
        for (auto& p : m) {
            t_[p.first]=p.second;
        }
    }

    bool GetValues(KeyT& key, ValueT& value)
    {
        ReadLockT readlock(mutex_);
        typename MapT::const_iterator it = t_.find(key);
        bool stat = (it!=t_.end());
        value = stat ? it.second : ValueT();
        return stat;
    }

    bool GetValues(KeyT&& key, ValueT& value)
    {
        ReadLockT readlock(mutex_);
        typename MapT::const_iterator it = t_.find(key);
        bool stat = (it!=t_.end());
        value = stat ? it.second : ValueT();
        return stat;
    }

    MapT GetCopy()
    {
        ReadLockT readlock(mutex_);
        return t_;
    }

    size_t GetSize()
    {
        ReadLockT readlock(mutex_);
        return t_.size();
    }

    void Reset()
    {
        WriteLockT writelock(mutex_);
        t_.clear();
    }

    void Swap(MapT t)
    {
        WriteLockT writelock(mutex_);
        t_.swap(t);
    }


protected:
    MapT t_;
    LockT mutex_;

};
} // namespace utils
} // namespace identt
#endif /* _IDENTT_UTILS_SHARED_MAP_HPP_ */