mirror of https://github.com/gnss-sdr/gnss-sdr
97 lines
2.3 KiB
C++
97 lines
2.3 KiB
C++
/*!
|
|
* \file concurrent_map.h
|
|
* \brief Interface of a thread-safe std::map
|
|
* \author Javier Arribas, 2011. jarribas(at)cttc.es
|
|
*
|
|
* -----------------------------------------------------------------------------
|
|
*
|
|
* GNSS-SDR is a Global Navigation Satellite System software-defined receiver.
|
|
* This file is part of GNSS-SDR.
|
|
*
|
|
* Copyright (C) 2010-2020 (see AUTHORS file for a list of contributors)
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
|
*
|
|
* -----------------------------------------------------------------------------
|
|
*/
|
|
|
|
#ifndef GNSS_SDR_CONCURRENT_MAP_H
|
|
#define GNSS_SDR_CONCURRENT_MAP_H
|
|
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <utility>
|
|
|
|
/** \addtogroup Core
|
|
* \{ */
|
|
/** \addtogroup Core_Receiver core_receiver
|
|
* \{ */
|
|
|
|
|
|
template <typename Data>
|
|
|
|
|
|
/*!
|
|
* \brief This class implements a thread-safe std::map
|
|
*
|
|
*/
|
|
class Concurrent_Map
|
|
{
|
|
typedef typename std::map<int, Data>::iterator Data_iterator; // iterator is scope dependent
|
|
public:
|
|
void write(int key, Data const& data)
|
|
{
|
|
std::unique_lock<std::mutex> lock(the_mutex);
|
|
Data_iterator data_iter;
|
|
data_iter = the_map.find(key);
|
|
if (data_iter != the_map.end())
|
|
{
|
|
data_iter->second = data; // update
|
|
}
|
|
else
|
|
{
|
|
the_map.insert(std::pair<int, Data>(key, data)); // insert SILENTLY fails if the item already exists in the map!
|
|
}
|
|
lock.unlock();
|
|
}
|
|
|
|
std::map<int, Data> get_map_copy()
|
|
{
|
|
std::unique_lock<std::mutex> lock(the_mutex);
|
|
std::map<int, Data> map_aux = the_map;
|
|
lock.unlock();
|
|
return map_aux;
|
|
}
|
|
|
|
size_t size()
|
|
{
|
|
std::unique_lock<std::mutex> lock(the_mutex);
|
|
size_t size_ = the_map.size();
|
|
lock.unlock();
|
|
return size_;
|
|
}
|
|
|
|
bool read(int key, Data& p_data)
|
|
{
|
|
std::unique_lock<std::mutex> lock(the_mutex);
|
|
Data_iterator data_iter;
|
|
data_iter = the_map.find(key);
|
|
if (data_iter != the_map.end())
|
|
{
|
|
p_data = data_iter->second;
|
|
lock.unlock();
|
|
return true;
|
|
}
|
|
lock.unlock();
|
|
return false;
|
|
}
|
|
|
|
private:
|
|
std::map<int, Data> the_map;
|
|
mutable std::mutex the_mutex;
|
|
};
|
|
|
|
|
|
/** \} */
|
|
/** \} */
|
|
#endif // GNSS_SDR_CONCURRENT_MAP_H
|