thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-19 16:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix data race: delay reset of data hooks until the next sendRequest()

Browse Source
This commit is contained in:
serxa 2024-08-02 13:19:00 +00:00
parent 8724c5ad52
commit 3bff7ddcf8
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/Common/HTTPConnectionPool.cpp
View File

@ -394,8 +394,12 @@ private:
// Reset data hooks for IO scheduling
if (ResourceLink link = CurrentThread::getReadResourceLink())
Session::setReceiveDataHooks(std::make_shared<ResourceGuardSessionDataHooks>(link, ResourceGuard::Metrics::getIORead(), log, request.getMethod(), request.getURI()));
else
Session::setReceiveDataHooks();
if (ResourceLink link = CurrentThread::getWriteResourceLink())
Session::setSendDataHooks(std::make_shared<ResourceGuardSessionDataHooks>(link, ResourceGuard::Metrics::getIOWrite(), log, request.getMethod(), request.getURI()));
else
Session::setSendDataHooks();
std::ostream & result = Session::sendRequest(request);
result.exceptions(std::ios::badbit);
@ -453,8 +457,6 @@ private:
}
}
response_stream = nullptr;
Session::setSendDataHooks();
Session::setReceiveDataHooks();
group->atConnectionDestroy();