blob: ca3a58483c9e2a3e318fb506e3a5472da02e1a24 (
plain) (
blame)
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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "threadpool.h"
namespace vespamalloc {
#ifdef __PIC__
#define TLS_LINKAGE __attribute__((visibility("hidden"), tls_model("initial-exec")))
#else
#define TLS_LINKAGE __attribute__((visibility("hidden"), tls_model("local-exec")))
#endif
template <typename MemBlockPtrT, typename ThreadStatT>
class ThreadListT
{
public:
using ThreadPool = ThreadPoolT<MemBlockPtrT, ThreadStatT >;
using AllocPool = AllocPoolT<MemBlockPtrT>;
ThreadListT(AllocPool & allocPool, MMapPool & mmapPool);
ThreadListT(const ThreadListT & tl) = delete;
ThreadListT & operator = (const ThreadListT & tl) = delete;
~ThreadListT();
void setParams(size_t threadCacheLimit) {
ThreadPool::setParams(threadCacheLimit);
}
bool quitThisThread();
bool initThisThread();
ThreadPool & getCurrent() { return *_myPool; }
size_t getThreadId() const { return (_myPool - _threadVector); }
void enableThreadSupport() {
_isThreaded.test_and_set();
}
void info(FILE * os, size_t level=0);
size_t getMaxNumThreads() const { return NELEMS(_threadVector); }
private:
std::atomic_flag _isThreaded;
std::atomic<uint32_t> _threadCount;
std::atomic<uint32_t> _threadCountAccum;
ThreadPool _threadVector[NUM_THREADS];
AllocPoolT<MemBlockPtrT> & _allocPool;
MMapPool & _mmapPool;
static thread_local ThreadPool * _myPool TLS_LINKAGE;
};
template <typename MemBlockPtrT, typename ThreadStatT>
thread_local ThreadPoolT<MemBlockPtrT, ThreadStatT> * ThreadListT<MemBlockPtrT, ThreadStatT>::_myPool TLS_LINKAGE = nullptr;
}
|
