thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-04 05:22:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug with buffered read

Browse Source
This commit is contained in:
alesapin 2020-11-26 15:26:41 +03:00
parent 0b079cdeb8
commit fc0bca1ff2
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/Server/TestKeeperTCPHandler.cpp
View File

@ -89,6 +89,7 @@ struct SocketInterruptablePollWrapper
#if defined(POCO_HAVE_FD_EPOLL)
int rc;
epoll_event evout[2];
memset(evout, 0, sizeof(evout));
do
{
Poco::Timestamp start;
@ -284,16 +285,20 @@ void TestKeeperTCPHandler::runImpl()
auto state = poll_wrapper->poll(session_timeout);
if (state & SocketInterruptablePollWrapper::HAS_REQUEST)
{
auto received_op = receiveRequest();
if (received_op == Coordination::OpNum::Close)
do
{
LOG_DEBUG(log, "Received close request for session #{}", session_id);
break;
}
else if (received_op == Coordination::OpNum::Heartbeat)
{
session_stopwatch.restart();
}
auto received_op = receiveRequest();
if (received_op == Coordination::OpNum::Close)
{
LOG_DEBUG(log, "Received close request for session #{}", session_id);
break;
}
else if (received_op == Coordination::OpNum::Heartbeat)
{
LOG_TRACE(log, "Received heartbeat for session #{}", session_id);
session_stopwatch.restart();
}
} while (in->available());
}
if (state & SocketInterruptablePollWrapper::HAS_RESPONSE)
@ -360,7 +365,7 @@ Coordination::OpNum TestKeeperTCPHandler::receiveRequest()
Coordination::ZooKeeperRequestPtr request = Coordination::ZooKeeperRequestFactory::instance().get(opnum);
request->xid = xid;
request->readImpl(*in);
int response_fd = poll_wrapper->getResponseFD();
int response_fd = poll_wrapper->getResponseFD();
if (opnum != Coordination::OpNum::Close)
{
auto promise = std::make_shared<std::promise<Coordination::ZooKeeperResponsePtr>>();