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

ISSUES-16605 try fix review comment

Browse Source
This commit is contained in:
zhang2014 2020-11-22 20:42:55 +08:00
parent 0985029f49
commit ade04b5dc4
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/Server/MySQLHandler.cpp
View File

@ -328,8 +328,15 @@ void MySQLHandler::comQuery(ReadBuffer & payload)
Context query_context = connection_context;
size_t affected_rows = 0;
query_context.setProgressCallback([&](const Progress & progress) { affected_rows += progress.written_rows; });
std::atomic<size_t> affected_rows {0};
auto prev = query_context.getProgressCallback();
query_context.setProgressCallback([&, prev = prev](const Progress & progress)
{
if (prev)
prev(progress);
affected_rows += progress.written_rows;
});
executeQuery(should_replace ? replacement : payload, *out, true, query_context,
[&with_output](const String &, const String &, const String &, const String &)