thread_safe_map.hpp

Go to the documentation of this file.

#ifndef _THREAD_SAFE_MAP_HPP_
#define _THREAD_SAFE_MAP_HPP_

#include <boost/smart_ptr.hpp>
#include <boost/thread.hpp>

#include <map>
#include <vector>
#include <iostream>

namespace threadsafe
{
  template <class T>
  class Map
  {
  public:
    Map()
    {
      mutex_ = boost::shared_ptr<boost::shared_mutex>(new boost::shared_mutex());
      map_ = boost::shared_ptr<InternalMap>(new InternalMap());
    };
    ~Map() {};

    T find(std::string first)
    {
      boost::shared_lock< boost::shared_mutex > lock( *(mutex_.get()) );

      return map_->find(first)->second;
    }

    bool insert(const std::string& first, const T& value)
    {
      if(!mutex_->timed_lock(boost::posix_time::microseconds(lock_wait_time)))
        return false;

      keys_.push_back(first);
      map_->insert(std::pair<std::string, T>(first, value));
      mutex_->unlock();
      return true;
    }

    bool update(const std::string& first, const T& value)
    {
      if(!mutex_->timed_lock(boost::posix_time::microseconds(lock_wait_time)))
        return false;

      (*map_)[first] = value;
      mutex_->unlock();
      return true;

    }

    std::vector<std::string> keys()
    {
      return keys_;
    }

  private:
    static const int lock_wait_time = 100;

    typedef std::map<std::string , T> InternalMap;

    boost::shared_ptr<InternalMap> map_;

    boost::shared_ptr<boost::shared_mutex> mutex_;
    std::vector<std::string> keys_;
  };
}

/* For the emacs weenies in the crowd.
Local Variables:
   c-basic-offset: 2
End:
*/

#endif