blob: 2ec65f8b504cd73f50897a7926cdbb32cf9b7a65 (
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
|
#pragma once
#include <Main.h>
namespace Balau {
class TLSManager {
public:
TLSManager() { }
virtual void * getTLS();
virtual void * setTLS(void * val);
private:
TLSManager(const TLSManager &) = delete;
TLSManager & operator=(const TLSManager &) = delete;
};
class PThreadsTLSManager : public TLSManager {
public:
virtual void * getTLS();
virtual void * setTLS(void * val);
void init();
private:
pthread_key_t m_key;
};
template <class TLS>
class PThreadsTLSFactory : private PThreadsTLSManager {
public:
PThreadsTLSFactory() : m_constructor([]() -> TLS * { return new TLS(); }) { init(); }
~PThreadsTLSFactory() { destroyAll(); }
void setConstructor(const std::function<TLS *()> & constructor) { m_constructor = constructor; }
TLS * get() {
TLS * tls = (TLS *) getTLS();
if (!tls) {
tls = m_constructor();
setTLS(tls);
m_TLSes.push(tls);
++m_numTLSes;
}
return tls;
}
private:
void destroyAll() {
while (m_numTLSes--) {
TLS * tls = m_TLSes.pop();
delete tls;
}
}
Queue<TLS> m_TLSes;
std::atomic<unsigned> m_numTLSes;
std::function<TLS *()> m_constructor;
};
extern TLSManager * g_tlsManager;
class Local : public AtStart {
public:
static int getSize() { return s_size; }
static void * createTLS(void * c = NULL) {
void * r = calloc(s_size * sizeof(void *), 1);
if (c)
memcpy(r, c, s_size * sizeof(void *));
return r;
}
protected:
Local() : AtStart(0) { }
void * getGlobal() { return m_globals[m_idx]; }
void * getLocal() { return reinterpret_cast<void **>(getTLS())[m_idx]; }
void * get() { if (getTLS()) { void * l = getLocal(); return l ? l : getGlobal(); } else return getGlobal(); }
void setGlobal(void * obj) { m_globals[m_idx] = obj; }
void setLocal(void * obj) { void * r = getTLS(); reinterpret_cast<void **>(r)[m_idx] = obj; }
void set(void * obj) { void * r = getTLS(); if (r) setLocal(obj); else setGlobal(obj); }
int getIndex() { return m_idx; }
private:
static void * getTLS() { return g_tlsManager->getTLS(); }
static void * setTLS(void * val) { return g_tlsManager->setTLS(val); }
virtual void doStart();
int m_idx;
static int s_size;
static void ** m_globals;
friend class TLSManager;
};
template<class T>
class DefaultTmpl : public AtStart {
public:
DefaultTmpl(int pri) : AtStart(pri) { }
protected:
virtual void doStart() { new T; }
};
template<class T>
class LocalTmpl : public Local {
public:
LocalTmpl() { }
T * getGlobal() { return reinterpret_cast<T *>(Local::getGlobal()); }
T * get() { return reinterpret_cast<T *>(Local::get()); }
void setGlobal(T * obj) { Local::setGlobal(obj); }
void set(T * obj) { Local::set(obj); }
};
};
|
