mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
78fc36ca49
This makes the target location consistent with other auto-generated files like config_formats.h, config_core.h, and config_functions.h and simplifies the build of clickhouse_common.
70 lines
1.6 KiB
C++
70 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include <optional>
|
|
#include <base/types.h>
|
|
#include <signal.h>
|
|
#include <time.h>
|
|
|
|
#include "config.h"
|
|
|
|
|
|
namespace Poco
|
|
{
|
|
class Logger;
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/**
|
|
* Query profiler implementation for selected thread.
|
|
*
|
|
* This class installs timer and signal handler on creation to:
|
|
* 1. periodically pause given thread
|
|
* 2. collect thread's current stack trace
|
|
* 3. write collected stack trace to trace_pipe for TraceCollector
|
|
*
|
|
* Destructor tries to unset timer and restore previous signal handler.
|
|
* Note that signal handler implementation is defined by template parameter. See QueryProfilerReal and QueryProfilerCPU.
|
|
*/
|
|
template <typename ProfilerImpl>
|
|
class QueryProfilerBase
|
|
{
|
|
public:
|
|
QueryProfilerBase(UInt64 thread_id, int clock_type, UInt32 period, int pause_signal_);
|
|
~QueryProfilerBase();
|
|
|
|
private:
|
|
void tryCleanup();
|
|
|
|
Poco::Logger * log;
|
|
|
|
#if USE_UNWIND
|
|
/// Timer id from timer_create(2)
|
|
std::optional<timer_t> timer_id;
|
|
#endif
|
|
|
|
/// Pause signal to interrupt threads to get traces
|
|
int pause_signal;
|
|
};
|
|
|
|
/// Query profiler with timer based on real clock
|
|
class QueryProfilerReal : public QueryProfilerBase<QueryProfilerReal>
|
|
{
|
|
public:
|
|
QueryProfilerReal(UInt64 thread_id, UInt32 period); /// NOLINT
|
|
|
|
static void signalHandler(int sig, siginfo_t * info, void * context);
|
|
};
|
|
|
|
/// Query profiler with timer based on CPU clock
|
|
class QueryProfilerCPU : public QueryProfilerBase<QueryProfilerCPU>
|
|
{
|
|
public:
|
|
QueryProfilerCPU(UInt64 thread_id, UInt32 period); /// NOLINT
|
|
|
|
static void signalHandler(int sig, siginfo_t * info, void * context);
|
|
};
|
|
|
|
}
|