forked from greg7mdp/parallel-hashmap
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathlazy_emplace_l.cc
45 lines (39 loc) · 1.47 KB
/
lazy_emplace_l.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
// ------------------------
// Windows specific example
// curtesy of @kanonka
// ------------------------
#include <windows.h>
#include "parallel_hashmap/phmap.h"
#include <cstring>
#include <vector>
#include <ppl.h>
using Map = phmap::parallel_flat_hash_map<std::string, int, phmap::priv::hash_default_hash<std::string>,
phmap::priv::hash_default_eq<std::string>,
std::allocator<std::pair<const std::string, int>>, 8, phmap::srwlock>;
class Dict
{
Map m_stringsMap;
public:
int addParallel(std::string&& str, volatile long* curIdx)
{
int newIndex = -1;
m_stringsMap.lazy_emplace_l(std::move(str),
[&](Map::value_type& p) { newIndex = p.second; }, // called only when key was already present
[&](const Map::constructor& ctor) // construct value_type in place when key not present
{ newIndex = InterlockedIncrement(curIdx); ctor(std::move(str), newIndex); });
return newIndex;
}
};
int main()
{
size_t totalSize = 6000000;
std::vector<int> values(totalSize);
Dict dict;
volatile long index = 0;
concurrency::parallel_for(size_t(0), size_t(totalSize),
[&](size_t i) {
std::string s = "ab_uu_" + std::to_string(i % 1000000);
values[i] = dict.addParallel(std::move(s), &index);
});
return 0;
}