mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-21 15:12:02 +00:00
63 lines
1.6 KiB
C++
63 lines
1.6 KiB
C++
#include <pthread.h>
|
|
|
|
#if defined(__APPLE__)
|
|
#elif defined(__FreeBSD__)
|
|
#include <pthread_np.h>
|
|
#else
|
|
#include <sys/prctl.h>
|
|
#endif
|
|
|
|
#include <cstring>
|
|
|
|
#include <Common/Exception.h>
|
|
#include <Common/setThreadName.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int PTHREAD_ERROR;
|
|
}
|
|
}
|
|
|
|
|
|
void setThreadName(const char * name)
|
|
{
|
|
#ifndef NDEBUG
|
|
if (strlen(name) > 15)
|
|
throw DB::Exception("Thread name cannot be longer than 15 bytes", DB::ErrorCodes::PTHREAD_ERROR);
|
|
#endif
|
|
|
|
#if defined(__FreeBSD__)
|
|
pthread_set_name_np(pthread_self(), name);
|
|
return;
|
|
|
|
#elif defined(__APPLE__)
|
|
if (0 != pthread_setname_np(name))
|
|
#else
|
|
if (0 != prctl(PR_SET_NAME, name, 0, 0, 0))
|
|
#endif
|
|
DB::throwFromErrno("Cannot set thread name with prctl(PR_SET_NAME, ...)", DB::ErrorCodes::PTHREAD_ERROR);
|
|
}
|
|
|
|
std::string getThreadName()
|
|
{
|
|
std::string name(16, '\0');
|
|
|
|
#if defined(__APPLE__)
|
|
if (pthread_getname_np(pthread_self(), name.data(), name.size()))
|
|
throw DB::Exception("Cannot get thread name with pthread_getname_np()", DB::ErrorCodes::PTHREAD_ERROR);
|
|
#elif defined(__FreeBSD__)
|
|
// TODO: make test. freebsd will have this function soon https://freshbsd.org/commit/freebsd/r337983
|
|
// if (pthread_get_name_np(pthread_self(), name.data(), name.size()))
|
|
// throw DB::Exception("Cannot get thread name with pthread_get_name_np()", DB::ErrorCodes::PTHREAD_ERROR);
|
|
#else
|
|
if (0 != prctl(PR_GET_NAME, name.data(), 0, 0, 0))
|
|
DB::throwFromErrno("Cannot get thread name with prctl(PR_GET_NAME)", DB::ErrorCodes::PTHREAD_ERROR);
|
|
#endif
|
|
|
|
name.resize(std::strlen(name.data()));
|
|
return name;
|
|
}
|