1 //===-- ThreadSafeDenseSet.h ------------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef liblldb_ThreadSafeDenseSet_h_
11 #define liblldb_ThreadSafeDenseSet_h_
12 
13 // C Includes
14 // C++ Includes
15 #include <mutex>
16 
17 // Other libraries and framework includes
18 #include "llvm/ADT/DenseSet.h"
19 
20 // Project includes
21 
22 namespace lldb_private {
23 
24 template <typename _ElementType, typename _MutexType = std::mutex>
25 class ThreadSafeDenseSet
26 {
27 public:
28     typedef llvm::DenseSet<_ElementType> LLVMSetType;
29 
30     ThreadSafeDenseSet(unsigned set_initial_capacity = 0) : m_set(set_initial_capacity), m_mutex() {}
31 
32     void
33     Insert(_ElementType e)
34     {
35         std::lock_guard<_MutexType> guard(m_mutex);
36         m_set.insert(e);
37     }
38 
39     void
40     Erase(_ElementType e)
41     {
42         std::lock_guard<_MutexType> guard(m_mutex);
43         m_set.erase(e);
44     }
45 
46     bool
47     Lookup(_ElementType e)
48     {
49         std::lock_guard<_MutexType> guard(m_mutex);
50         return (m_set.count(e) > 0);
51     }
52 
53     void
54     Clear()
55     {
56         stds::lock_guard<_MutexType> guard(m_mutex);
57         m_set.clear();
58     }
59 
60 protected:
61     LLVMSetType m_set;
62     _MutexType m_mutex;
63 };
64 
65 } // namespace lldb_private
66 
67 #endif  // liblldb_ThreadSafeDenseSet_h_
68