mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 17:32:32 +00:00
81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
#include "executeQuery.h"
|
|
#include <IO/Progress.h>
|
|
#include <DataStreams/RemoteBlockInputStream.h>
|
|
#include <Core/Block.h>
|
|
#include <Poco/UUIDGenerator.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace
|
|
{
|
|
|
|
void checkFulfilledConditionsAndUpdate(
|
|
const Progress & progress, RemoteBlockInputStream & stream,
|
|
TestStats & statistics, TestStopConditions & stop_conditions,
|
|
InterruptListener & interrupt_listener)
|
|
{
|
|
statistics.add(progress.read_rows, progress.read_bytes);
|
|
|
|
stop_conditions.reportRowsRead(statistics.total_rows_read);
|
|
stop_conditions.reportBytesReadUncompressed(statistics.total_bytes_read);
|
|
stop_conditions.reportTotalTime(statistics.watch.elapsed() / (1000 * 1000));
|
|
stop_conditions.reportMinTimeNotChangingFor(statistics.min_time_watch.elapsed() / (1000 * 1000));
|
|
|
|
if (stop_conditions.areFulfilled())
|
|
{
|
|
statistics.last_query_was_cancelled = true;
|
|
stream.cancel(false);
|
|
}
|
|
|
|
if (interrupt_listener.check())
|
|
{
|
|
statistics.got_SIGINT = true;
|
|
statistics.last_query_was_cancelled = true;
|
|
stream.cancel(false);
|
|
}
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
void executeQuery(
|
|
Connection & connection,
|
|
const std::string & query,
|
|
TestStats & statistics,
|
|
TestStopConditions & stop_conditions,
|
|
InterruptListener & interrupt_listener,
|
|
Context & context,
|
|
const Settings & settings)
|
|
{
|
|
static const std::string query_id_prefix
|
|
= Poco::UUIDGenerator::defaultGenerator().create().toString() + "-";
|
|
static int next_query_id = 1;
|
|
|
|
statistics.watch_per_query.restart();
|
|
statistics.last_query_was_cancelled = false;
|
|
statistics.last_query_rows_read = 0;
|
|
statistics.last_query_bytes_read = 0;
|
|
statistics.query_id = query_id_prefix + std::to_string(next_query_id++);
|
|
|
|
RemoteBlockInputStream stream(connection, query, {}, context, &settings);
|
|
stream.setQueryId(statistics.query_id);
|
|
|
|
stream.setProgressCallback(
|
|
[&](const Progress & value)
|
|
{
|
|
checkFulfilledConditionsAndUpdate(
|
|
value, stream, statistics,
|
|
stop_conditions, interrupt_listener);
|
|
});
|
|
stream.readPrefix();
|
|
while (Block block = stream.read());
|
|
stream.readSuffix();
|
|
|
|
if (!statistics.last_query_was_cancelled)
|
|
statistics.updateQueryInfo();
|
|
|
|
statistics.setTotalTime();
|
|
}
|
|
|
|
}
|