thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-29 02:52:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initialize thread tracing context for TCPServer

Browse Source
This commit is contained in:
Frank Chen 2022-07-07 17:42:22 +08:00
parent 8c04456148
commit 19f3f3e6ca
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Server/TCPHandler.cpp
View File

@ -197,6 +197,7 @@ void TCPHandler::runImpl()
/// Initialized later.
std::optional<CurrentThread::QueryScope> query_scope;
OpenTelemetryThreadTraceContextScopePtr thread_trace_context;
/** An exception during the execution of request (it must be sent over the network to the client).
* The client will be able to accept it, if it did not happen while sending another packet and the client has not disconnected yet.
@ -222,6 +223,8 @@ void TCPHandler::runImpl()
if (state.empty() && state.part_uuids_to_ignore && !receivePacket())
continue;
thread_trace_context = query_context->startTracing("TCPHandler");
query_scope.emplace(query_context);
/// If query received, then settings in query_context has been updated.
@ -396,6 +399,7 @@ void TCPHandler::runImpl()
/// (i.e. deallocations from the Aggregator with two-level aggregation)
state.reset();
query_scope.reset();
thread_trace_context.reset();
}
catch (const Exception & e)
{
@ -458,6 +462,9 @@ void TCPHandler::runImpl()
{
if (exception)
{
if (thread_trace_context)
thread_trace_context->root_span.addAttribute(exception.value());
try
{
/// Try to send logs to client, but it could be risky too
@ -501,6 +508,7 @@ void TCPHandler::runImpl()
/// (i.e. deallocations from the Aggregator with two-level aggregation)
state.reset();
query_scope.reset();
thread_trace_context.reset();
}
catch (...)
{