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

Try fix tests.

Browse Source
This commit is contained in:
Nikolai Kochetov 2021-04-19 20:55:18 +03:00
parent 93ae00bd0d
commit f6bdd35943
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/Server/TCPHandler.cpp
View File

@ -300,6 +300,8 @@ void TCPHandler::runImpl()
/// Processing Query
state.io = executeQuery(state.query, query_context, false, state.stage, may_have_embedded_data);
unknown_packet_in_send_data = query_context->getSettingsRef().unknown_packet_in_send_data;
after_check_cancelled.restart();
after_send_progress.restart();
@ -1464,11 +1466,10 @@ void TCPHandler::sendData(const Block & block)
try
{
/// For testing hedged requests
const Settings & settings = query_context->getSettingsRef();
if (settings.unknown_packet_in_send_data)
if (unknown_packet_in_send_data)
{
--const_cast<SettingFieldUInt64 &>(settings.unknown_packet_in_send_data).value;
if (settings.unknown_packet_in_send_data == 0)
--unknown_packet_in_send_data;
if (unknown_packet_in_send_data == 0)
writeVarUInt(UInt64(-1), *out);
}

2
src/Server/TCPHandler.h
View File

@ -135,6 +135,8 @@ private:
ContextPtr connection_context;
ContextPtr query_context;
size_t unknown_packet_in_send_data = 0;
/// Streams for reading/writing from/to client connection socket.
std::shared_ptr<ReadBuffer> in;
std::shared_ptr<WriteBuffer> out;