blob: 63ba1ee227fc5a4f2c4e5e42f95e5586bf697c5c (
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
|
#include "thread_name.h"
#include <util/generic/string.h>
#include <util/system/thread.h>
#include <algorithm>
#include <cstring>
namespace NYT {
////////////////////////////////////////////////////////////////////////////////
TStringBuf TThreadName::ToStringBuf() const
{
// Buffer is zero terminated.
return TStringBuf(Buffer.data(), Length);
}
////////////////////////////////////////////////////////////////////////////////
TThreadName::TThreadName(const TString& name)
{
Length = std::min<int>(TThreadName::BufferCapacity - 1, name.length());
::memcpy(Buffer.data(), name.data(), Length);
}
////////////////////////////////////////////////////////////////////////////////
// This function uses cached TThread::CurrentThreadName() result
TThreadName GetCurrentThreadName()
{
static thread_local TThreadName threadName;
if (threadName.Length == 0) {
if (auto name = TThread::CurrentThreadName()) {
auto length = std::min<int>(TThreadName::BufferCapacity - 1, name.length());
threadName.Length = length;
::memcpy(threadName.Buffer.data(), name.data(), length);
}
}
return threadName;
}
////////////////////////////////////////////////////////////////////////////////
} // namespace NYT
|
