blob: c485b91acb47a0216451b113f16238ac747ae8df [file] [log] [blame]
//===-- ThreadSafeDenseMap.h ------------------------------------------*- C++
//-*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef liblldb_ThreadSafeDenseMap_h_
#define liblldb_ThreadSafeDenseMap_h_
#include <mutex>
#include "llvm/ADT/DenseMap.h"
namespace lldb_private {
template <typename _KeyType, typename _ValueType,
typename _MutexType = std::mutex>
class ThreadSafeDenseMap {
public:
typedef llvm::DenseMap<_KeyType, _ValueType> LLVMMapType;
ThreadSafeDenseMap(unsigned map_initial_capacity = 0)
: m_map(map_initial_capacity), m_mutex() {}
void Insert(_KeyType k, _ValueType v) {
std::lock_guard<_MutexType> guard(m_mutex);
m_map.insert(std::make_pair(k, v));
}
void Erase(_KeyType k) {
std::lock_guard<_MutexType> guard(m_mutex);
m_map.erase(k);
}
_ValueType Lookup(_KeyType k) {
std::lock_guard<_MutexType> guard(m_mutex);
return m_map.lookup(k);
}
bool Lookup(_KeyType k, _ValueType &v) {
std::lock_guard<_MutexType> guard(m_mutex);
auto iter = m_map.find(k), end = m_map.end();
if (iter == end)
return false;
v = iter->second;
return true;
}
void Clear() {
std::lock_guard<_MutexType> guard(m_mutex);
m_map.clear();
}
protected:
LLVMMapType m_map;
_MutexType m_mutex;
};
} // namespace lldb_private
#endif // liblldb_ThreadSafeSTLMap_h_