diff --git a/dbms/include/DB/DataStreams/RemoteBlockInputStream.h b/dbms/include/DB/DataStreams/RemoteBlockInputStream.h index b160cbb1190..e378b4505fc 100644 --- a/dbms/include/DB/DataStreams/RemoteBlockInputStream.h +++ b/dbms/include/DB/DataStreams/RemoteBlockInputStream.h @@ -1,5 +1,7 @@ #pragma once +#include + #include #include @@ -14,7 +16,8 @@ class RemoteBlockInputStream : public IProfilingBlockInputStream { public: RemoteBlockInputStream(Connection & connection_, const String & query_, QueryProcessingStage::Enum stage_ = QueryProcessingStage::Complete) - : connection(connection_), query(query_), stage(stage_), sent_query(false), finished(false), cancelled(false) + : connection(connection_), query(query_), stage(stage_), sent_query(false), finished(false), cancelled(false), + log(&Logger::get("RemoteBlockInputStream (" + connection.getServerAddress() + ")")) { } @@ -33,6 +36,8 @@ public: { if (is_cancelled_callback && is_cancelled_callback()) { + LOG_TRACE(log, "Cancelling query"); + /// Если да - запросим удалённый сервер тоже прервать запрос. cancelled = true; connection.sendCancel(); @@ -83,6 +88,8 @@ public: { if (!cancelled) { + LOG_TRACE(log, "Cancelling query because enough data has been read"); + cancelled = true; connection.sendCancel(); } @@ -122,6 +129,8 @@ private: bool sent_query; bool finished; bool cancelled; + + Logger * log; }; }