66 lines
1.6 KiB
C++
66 lines
1.6 KiB
C++
|
/***
|
||
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
||
|
|
||
|
File: TLSVariable.hpp
|
||
|
Date: 2021-6-11
|
||
|
Author: Reece
|
||
|
***/
|
||
|
#pragma once
|
||
|
|
||
|
namespace Aurora::Threading::Threads
|
||
|
{
|
||
|
template<typename T>
|
||
|
class TLSVariable
|
||
|
{
|
||
|
private:
|
||
|
int _;
|
||
|
|
||
|
public:
|
||
|
TLSVariable() {}
|
||
|
~TLSVariable()
|
||
|
{
|
||
|
GetThread()->GetTlsView()->Remove(GetHandle());
|
||
|
}
|
||
|
|
||
|
AuUInt64 GetHandle()
|
||
|
{
|
||
|
return (AuUInt64(reinterpret_cast<AuUInt>(&_)) & (~kTlsKeyMask)) | kTlsKeyFollowsConvention | kTlsKeyResettablePointerHandle;
|
||
|
}
|
||
|
|
||
|
T &Get()
|
||
|
{
|
||
|
auto view = GetThread()->GetTlsView();
|
||
|
auto ptr = view->GetOrSetup(GetHandle(),
|
||
|
[](void *buffer) -> void
|
||
|
{
|
||
|
if constexpr (std::is_class_v<T>)
|
||
|
{
|
||
|
new (buffer) T();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//std::memset(buffer, 0, sizeof(T));
|
||
|
}
|
||
|
},
|
||
|
[](void *buffer) -> void
|
||
|
{
|
||
|
if constexpr (std::is_class_v<T>)
|
||
|
{
|
||
|
reinterpret_cast<T *>(buffer)->~T();
|
||
|
}
|
||
|
});
|
||
|
return *reinterpret_cast<T *>(ptr);
|
||
|
}
|
||
|
|
||
|
T& operator ->()
|
||
|
{
|
||
|
return Get();
|
||
|
}
|
||
|
|
||
|
TLSVariable& operator =(const T & val)
|
||
|
{
|
||
|
Get() = val;
|
||
|
return *this;
|
||
|
}
|
||
|
};
|
||
|
}
|