blob: 001f2d09f36d88c2ef0c2e6f81ed9c915334fa84 (
plain)
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#ifndef ENV_ENVFS_H
#define ENV_ENVFS_H
#include <thread>
namespace env
{
struct File
{
std::wstring name;
std::wstring lcname;
FILETIME lastModified;
File(std::wstring_view name, FILETIME ft);
};
struct Directory
{
std::wstring name;
std::wstring lcname;
std::vector<Directory> dirs;
std::vector<File> files;
Directory();
Directory(std::wstring_view name);
};
template <class T>
class ThreadPool
{
public:
ThreadPool(std::size_t max=1)
: m_threads(max)
{
}
~ThreadPool()
{
join();
}
void setMax(std::size_t n)
{
m_threads.resize(n);
}
void join()
{
for (auto& ti : m_threads) {
if (ti.thread.joinable()) {
ti.thread.join();
}
}
}
T& request()
{
if (m_threads.empty()) {
std::terminate();
}
for (;;) {
for (auto& ti : m_threads) {
bool expected = false;
if (ti.busy.compare_exchange_strong(expected, true)) {
if (ti.thread.joinable()) {
ti.thread.join();
}
ti.thread = std::thread([&]{
ti.o.run();
ti.busy = false;
});
return ti.o;
}
}
std::this_thread::sleep_for(std::chrono::milliseconds(1));
}
}
template <class F>
void forEach(F&& f)
{
for (auto& ti : m_threads) {
f(ti.o);
}
}
private:
struct ThreadInfo
{
std::thread thread;
std::atomic<bool> busy;
T o;
};
std::list<ThreadInfo> m_threads;
};
using DirStartF = void (void*, std::wstring_view);
using DirEndF = void (void*, std::wstring_view);
using FileF = void (void*, std::wstring_view, FILETIME);
void setHandleCloserThreadCount(std::size_t n);
void shrinkFs();
void forEachEntry(
const std::wstring& path, void* cx,
DirStartF* dirStartF, DirEndF* dirEndF, FileF* fileF);
Directory getFilesAndDirs(const std::wstring& path);
Directory getFilesAndDirsWithFind(const std::wstring& path);
} // namespace
#endif // ENV_ENVFS_H
|
