library/cpp/yt/misc/thread_name.cpp


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

#include "thread_name.h"

#include <library/cpp/yt/misc/tls.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 YT_THREAD_LOCAL(TThreadName) ThreadName;
    auto& threadName = GetTlsRef(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