Fix other tests.

src/DataStreams/CountingBlockOutputStream.cpp

@@ -1,5 +1,6 @@
 #include <DataStreams/CountingBlockOutputStream.h>
 #include <Common/ProfileEvents.h>
+#include <iostream>
 
 
 namespace ProfileEvents
@@ -17,6 +18,8 @@ void CountingTransform::transform(Chunk & chunk)
     Progress local_progress(chunk.getNumRows(), chunk.bytes(), 0);
     progress.incrementPiecewiseAtomically(local_progress);
 
+    std::cerr << "============ counting adding progress for " << static_cast<const void *>(thread_status) << ' ' << chunk.getNumRows() << " rows\n";
+
     if (thread_status)
     {
         thread_status->performance_counters.increment(ProfileEvents::InsertedRows, local_progress.read_rows);

src/DataStreams/PushingToViewsBlockOutputStream.cpp

@@ -173,6 +173,7 @@ public:
             if (input.hasData())
             {
                 auto data = input.pullData();
+                //std::cerr << "********** FinalizingViewsTransform got input " << i << " has exc " << bool(data.exception) << std::endl;
                 if (data.exception)
                 {
                     if (views_data->has_exception && views_data->first_exception == data.exception)
@@ -197,6 +198,9 @@ public:
             if (!statuses.empty())
                 return Status::Ready;
 
+            if (any_exception)
+                output.pushException(std::move(any_exception));
+
             output.finish();
             return Status::Finished;
         }
@@ -217,6 +221,7 @@ public:
                 if (!any_exception)
                     any_exception = status.exception;
 
+                //std::cerr << "=== Setting exception for " << view.table_id.getFullNameNotQuoted() << std::endl;
                 view.setException(std::move(status.exception));
             }
             else
@@ -265,10 +270,10 @@ Chain buildPushingToViewsDrain(
     ContextPtr context,
     const ASTPtr & query_ptr,
     bool no_destination,
-    std::vector<TableLockHolder> & locks,
     ExceptionKeepingTransformRuntimeDataPtr runtime_data)
 {
     checkStackSize();
+    Chain result_chain;
 
     /// If we don't write directly to the destination
     /// then expect that we're inserting with precalculated virtual columns
@@ -279,7 +284,7 @@ Chain buildPushingToViewsDrain(
       * Although now any insertion into the table is done via PushingToViewsBlockOutputStream,
       *  but it's clear that here is not the best place for this functionality.
       */
-    locks.emplace_back(storage->lockForShare(context->getInitialQueryId(), context->getSettingsRef().lock_acquire_timeout));
+    result_chain.addTableLock(storage->lockForShare(context->getInitialQueryId(), context->getSettingsRef().lock_acquire_timeout));
 
     /// If the "root" table deduplicates blocks, there are no need to make deduplication for children
     /// Moreover, deduplication for AggregatingMergeTree children could produce false positives due to low size of inserting blocks
@@ -355,7 +360,7 @@ Chain buildPushingToViewsDrain(
         if (auto * materialized_view = dynamic_cast<StorageMaterializedView *>(dependent_table.get()))
         {
             type = QueryViewsLogElement::ViewType::MATERIALIZED;
-            locks.emplace_back(materialized_view->lockForShare(context->getInitialQueryId(), context->getSettingsRef().lock_acquire_timeout));
+            result_chain.addTableLock(materialized_view->lockForShare(context->getInitialQueryId(), context->getSettingsRef().lock_acquire_timeout));
 
             StoragePtr inner_table = materialized_view->getTargetTable();
             auto inner_table_id = inner_table->getStorageID();
@@ -383,7 +388,7 @@ Chain buildPushingToViewsDrain(
             insert->columns = std::move(list);
 
             ASTPtr insert_query_ptr(insert.release());
-            InterpreterInsertQuery interpreter(insert_query_ptr, insert_context, false, false, false, runtime_data);
+            InterpreterInsertQuery interpreter(insert_query_ptr, insert_context, false, false, false, view_runtime_data);
             BlockIO io = interpreter.execute();
             out = std::move(io.out);
         }
@@ -392,11 +397,11 @@ Chain buildPushingToViewsDrain(
             type = QueryViewsLogElement::ViewType::LIVE;
             query = live_view->getInnerQuery(); // Used only to log in system.query_views_log
             out = buildPushingToViewsDrain(
-                dependent_table, dependent_metadata_snapshot, insert_context, ASTPtr(), true, locks, view_runtime_data);
+                dependent_table, dependent_metadata_snapshot, insert_context, ASTPtr(), true, view_runtime_data);
         }
         else
             out = buildPushingToViewsDrain(
-                dependent_table, dependent_metadata_snapshot, insert_context, ASTPtr(), false, locks, view_runtime_data);
+                dependent_table, dependent_metadata_snapshot, insert_context, ASTPtr(), false, view_runtime_data);
 
         QueryViewsLogElement::ViewRuntimeStats runtime_stats{
             target_name,
@@ -435,8 +440,6 @@ Chain buildPushingToViewsDrain(
         }
     }
 
-    Chain result_chain;
-
     size_t num_views = views_data->views.size();
     if (num_views != 0)
     {
@@ -454,6 +457,7 @@ Chain buildPushingToViewsDrain(
 
         for (auto & chain : chains)
         {
+            result_chain.attachResourcesFrom(chain);
             connect(*out, chain.getInputPort());
             connect(chain.getOutputPort(), *in);
             ++in;
@@ -581,6 +585,8 @@ static void logQueryViews(std::list<ViewRuntimeData> & views, ContextPtr context
 
         try
         {
+            //std::cerr << "============ Logging for " << static_cast<const void *>(view.runtime_stats.thread_status.get()) << ' ' << view.table_id.getNameForLogs() << "\n";
+
             if (view.runtime_stats.thread_status)
                 view.runtime_stats.thread_status->logToQueryViewsLog(view);
         }

src/DataStreams/PushingToViewsBlockOutputStream.h

@@ -51,7 +51,6 @@ Chain buildPushingToViewsDrain(
     ContextPtr context,
     const ASTPtr & query_ptr,
     bool no_destination,
-    std::vector<TableLockHolder> & locks,
     ExceptionKeepingTransformRuntimeDataPtr runtime_data);
 
 

src/Interpreters/ActionsVisitor.cpp

@@ -26,6 +26,8 @@
 #include <Parsers/ASTSubquery.h>
 #include <Parsers/ASTTablesInSelectQuery.h>
 
+#include <Processors/QueryPlan/QueryPlan.h>
+
 #include <Interpreters/Context.h>
 #include <Interpreters/ExpressionActions.h>
 #include <Interpreters/misc.h>

src/Interpreters/ExpressionAnalyzer.cpp

@@ -56,6 +56,7 @@
 #include <IO/WriteBufferFromString.h>
 
 #include <Processors/Executors/PullingAsyncPipelineExecutor.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Parsers/formatAST.h>
 
 namespace DB

src/Interpreters/InterpreterInsertQuery.cpp

@@ -51,7 +51,7 @@ InterpreterInsertQuery::InterpreterInsertQuery(
     , allow_materialized(allow_materialized_)
     , no_squash(no_squash_)
     , no_destination(no_destination_)
-    , runtime_data(runtime_data_)
+    , runtime_data(std::move(runtime_data_))
 {
     checkStackSize();
 }
@@ -281,10 +281,7 @@ BlockIO InterpreterInsertQuery::execute()
             }
             else
             {
-                std::vector<TableLockHolder> locks;
-                out = buildPushingToViewsDrain(table, metadata_snapshot, getContext(), query_ptr, no_destination, locks, runtime_data);
-                for (auto & lock : locks)
-                    res.pipeline.addTableLock(std::move(lock));
+                out = buildPushingToViewsDrain(table, metadata_snapshot, getContext(), query_ptr, no_destination, runtime_data);
             }
 
             /// Note that we wrap transforms one on top of another, so we write them in reverse of data processing order.

src/Processors/Chain.h

@@ -1,6 +1,7 @@
 #pragma once
 
 #include <Processors/IProcessor.h>
+#include <Processors/PipelineResourcesHolder.h>
 
 namespace DB
 {
@@ -35,11 +36,16 @@ public:
     const std::list<ProcessorPtr> & getProcessors() const { return processors; }
     static std::list<ProcessorPtr> getProcessors(Chain chain) { return std::move(chain.processors); }
 
+    void addTableLock(TableLockHolder lock) { holder.table_locks.emplace_back(std::move(lock)); }
+    void attachResourcesFrom(Chain & other) { holder = std::move(other.holder); }
+    PipelineResourcesHolder detachResources() { return std::move(holder); }
+
 private:
     /// -> source -> transform -> ... -> transform -> sink ->
     ///  ^        ->           ->     ->           ->       ^
     ///  input port                               output port
     std::list<ProcessorPtr> processors;
+    PipelineResourcesHolder holder;
 };
 
 }

src/Processors/Pipe.cpp

@@ -9,6 +9,7 @@
 #include <Processors/Transforms/ExtremesTransform.h>
 #include <Processors/Formats/IOutputFormat.h>
 #include <Processors/Sources/NullSource.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Columns/ColumnConst.h>
 
 namespace DB
@@ -98,18 +99,9 @@ static OutputPort * uniteTotals(const OutputPortRawPtrs & ports, const Block & h
     return totals_port;
 }
 
-Pipe::Holder & Pipe::Holder::operator=(Holder && rhs)
+void Pipe::addQueryPlan(std::unique_ptr<QueryPlan> plan)
 {
-    table_locks.insert(table_locks.end(), rhs.table_locks.begin(), rhs.table_locks.end());
-    storage_holders.insert(storage_holders.end(), rhs.storage_holders.begin(), rhs.storage_holders.end());
-    interpreter_context.insert(interpreter_context.end(),
-                               rhs.interpreter_context.begin(), rhs.interpreter_context.end());
-    for (auto & plan : rhs.query_plans)
-        query_plans.emplace_back(std::move(plan));
-
-    query_id_holder = std::move(rhs.query_id_holder);
-
-    return *this;
+    holder.query_plans.emplace_back(std::move(plan));
 }
 
 Pipe::Pipe(ProcessorPtr source, OutputPort * output, OutputPort * totals, OutputPort * extremes)
@@ -689,6 +681,7 @@ void Pipe::addChains(std::vector<Chain> chains)
         connect(*output_ports[i], chains[i].getInputPort());
         output_ports[i] = &chains[i].getOutputPort();
 
+        holder = chains[i].detachResources();
         auto added_processors = Chain::getProcessors(std::move(chains[i]));
         for (auto & transform : added_processors)
         {

src/Processors/Pipe.h

@@ -1,12 +1,10 @@
 #pragma once
 
 #include <Processors/IProcessor.h>
-#include <Processors/QueryPlan/QueryIdHolder.h>
-#include <Processors/QueryPlan/QueryPlan.h>
+#include <Processors/PipelineResourcesHolder.h>
 #include <Processors/Chain.h>
 #include <Access/EnabledQuota.h>
 #include <DataStreams/SizeLimits.h>
-#include <Storages/TableLockHolder.h>
 
 namespace DB
 {
@@ -18,9 +16,6 @@ using Pipes = std::vector<Pipe>;
 
 class QueryPipeline;
 
-class IStorage;
-using StoragePtr = std::shared_ptr<IStorage>;
-
 using OutputPortRawPtrs = std::vector<OutputPort *>;
 
 /// Pipe is a set of processors which represents the part of pipeline.
@@ -118,29 +113,11 @@ public:
     void addStorageHolder(StoragePtr storage) { holder.storage_holders.emplace_back(std::move(storage)); }
     void addQueryIdHolder(std::shared_ptr<QueryIdHolder> query_id_holder) { holder.query_id_holder = std::move(query_id_holder); }
     /// For queries with nested interpreters (i.e. StorageDistributed)
-    void addQueryPlan(std::unique_ptr<QueryPlan> plan) { holder.query_plans.emplace_back(std::move(plan)); }
+    void addQueryPlan(std::unique_ptr<QueryPlan> plan);
 
 private:
     /// Destruction order: processors, header, locks, temporary storages, local contexts
-
-    struct Holder
-    {
-        Holder() = default;
-        Holder(Holder &&) = default;
-        /// Custom mode assignment does not destroy data from lhs. It appends data from rhs to lhs.
-        Holder& operator=(Holder &&);
-
-        /// Some processors may implicitly use Context or temporary Storage created by Interpreter.
-        /// But lifetime of Streams is not nested in lifetime of Interpreters, so we have to store it here,
-        /// because QueryPipeline is alive until query is finished.
-        std::vector<std::shared_ptr<const Context>> interpreter_context;
-        std::vector<StoragePtr> storage_holders;
-        std::vector<TableLockHolder> table_locks;
-        std::vector<std::unique_ptr<QueryPlan>> query_plans;
-        std::shared_ptr<QueryIdHolder> query_id_holder;
-    };
-
-    Holder holder;
+    PipelineResourcesHolder holder;
 
     /// Header is common for all output below.
     Block header;

src/Processors/PipelineResourcesHolder.cpp

@@ -0,0 +1,25 @@
+#include <Processors/PipelineResourcesHolder.h>
+#include <Processors/QueryPlan/QueryPlan.h>
+
+namespace DB
+{
+
+PipelineResourcesHolder::PipelineResourcesHolder() = default;
+PipelineResourcesHolder::PipelineResourcesHolder(PipelineResourcesHolder &&) = default;
+PipelineResourcesHolder::~PipelineResourcesHolder() = default;
+
+PipelineResourcesHolder & PipelineResourcesHolder::operator=(PipelineResourcesHolder && rhs)
+{
+    table_locks.insert(table_locks.end(), rhs.table_locks.begin(), rhs.table_locks.end());
+    storage_holders.insert(storage_holders.end(), rhs.storage_holders.begin(), rhs.storage_holders.end());
+    interpreter_context.insert(interpreter_context.end(),
+                               rhs.interpreter_context.begin(), rhs.interpreter_context.end());
+    for (auto & plan : rhs.query_plans)
+        query_plans.emplace_back(std::move(plan));
+
+    query_id_holder = std::move(rhs.query_id_holder);
+
+    return *this;
+}
+
+}

src/Processors/PipelineResourcesHolder.h

@@ -0,0 +1,34 @@
+#pragma once
+#include <Storages/TableLockHolder.h>
+#include <Processors/QueryPlan/QueryIdHolder.h>
+
+namespace DB
+{
+
+class QueryPipeline;
+
+class IStorage;
+using StoragePtr = std::shared_ptr<IStorage>;
+
+class QueryPlan;
+class Context;
+
+struct PipelineResourcesHolder
+{
+    PipelineResourcesHolder();
+    PipelineResourcesHolder(PipelineResourcesHolder &&);
+    ~PipelineResourcesHolder();
+    /// Custom mode assignment does not destroy data from lhs. It appends data from rhs to lhs.
+    PipelineResourcesHolder& operator=(PipelineResourcesHolder &&);
+
+    /// Some processors may implicitly use Context or temporary Storage created by Interpreter.
+    /// But lifetime of Streams is not nested in lifetime of Interpreters, so we have to store it here,
+    /// because QueryPipeline is alive until query is finished.
+    std::vector<std::shared_ptr<const Context>> interpreter_context;
+    std::vector<StoragePtr> storage_holders;
+    std::vector<TableLockHolder> table_locks;
+    std::vector<std::unique_ptr<QueryPlan>> query_plans;
+    std::shared_ptr<QueryIdHolder> query_id_holder;
+};
+
+}

src/Processors/QueryPipeline.cpp

@@ -21,6 +21,7 @@
 #include <Processors/DelayedPortsProcessor.h>
 #include <Processors/RowsBeforeLimitCounter.h>
 #include <Processors/Sources/RemoteSource.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 
 namespace DB
 {
@@ -29,6 +30,11 @@ namespace ErrorCodes
     extern const int LOGICAL_ERROR;
 }
 
+void QueryPipeline::addQueryPlan(std::unique_ptr<QueryPlan> plan)
+{
+    pipe.addQueryPlan(std::move(plan));
+}
+
 void QueryPipeline::checkInitialized()
 {
     if (!initialized())

src/Processors/QueryPipeline.h

@@ -123,7 +123,7 @@ public:
     void addTableLock(TableLockHolder lock) { pipe.addTableLock(std::move(lock)); }
     void addInterpreterContext(std::shared_ptr<const Context> context) { pipe.addInterpreterContext(std::move(context)); }
     void addStorageHolder(StoragePtr storage) { pipe.addStorageHolder(std::move(storage)); }
-    void addQueryPlan(std::unique_ptr<QueryPlan> plan) { pipe.addQueryPlan(std::move(plan)); }
+    void addQueryPlan(std::unique_ptr<QueryPlan> plan);
     void setLimits(const StreamLocalLimits & limits) { pipe.setLimits(limits); }
     void setLeafLimits(const SizeLimits & limits) { pipe.setLeafLimits(limits); }
     void setQuota(const std::shared_ptr<const EnabledQuota> & quota) { pipe.setQuota(quota); }

src/Processors/QueryPlan/CreatingSetsStep.cpp

@@ -1,4 +1,5 @@
 #include <Processors/QueryPlan/CreatingSetsStep.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPipeline.h>
 #include <Processors/Transforms/CreatingSetsTransform.h>
 #include <IO/Operators.h>

src/Processors/QueryPlan/ReadFromRemote.cpp

@@ -1,4 +1,5 @@
 #include <Processors/QueryPlan/ReadFromRemote.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/ExpressionStep.h>
 #include <Processors/QueryPlan/Optimizations/QueryPlanOptimizationSettings.h>
 #include <DataStreams/RemoteQueryExecutor.h>

src/Processors/Sinks/SinkToStorage.cpp

@@ -98,6 +98,7 @@ static std::exception_ptr runStep(std::function<void()> step, ExceptionKeepingTr
     }
     catch (Exception & exception)
     {
+        // std::cerr << "===== got exception " << getExceptionMessage(exception, false);
         if (runtime_data && !runtime_data->additional_exception_message.empty())
             exception.addMessage(runtime_data->additional_exception_message);
 
@@ -105,6 +106,7 @@ static std::exception_ptr runStep(std::function<void()> step, ExceptionKeepingTr
     }
     catch (...)
     {
+        // std::cerr << "===== got exception " << getExceptionMessage(std::current_exception(), false);
         res = std::current_exception();
     }
 
@@ -118,9 +120,10 @@ static std::exception_ptr runStep(std::function<void()> step, ExceptionKeepingTr
 
     return res;
 }
-
+\
 void ExceptionKeepingTransform::work()
 {
+    // std::cerr << "============ Executing " << getName() << std::endl;
     if (!was_on_start_called)
     {
         was_on_start_called = true;
@@ -138,6 +141,8 @@ void ExceptionKeepingTransform::work()
 
         if (auto exception = runStep([this] { transform(data.chunk); }, runtime_data.get()))
         {
+            // std::cerr << "===== got exception in " << getName() << std::endl;
+            // std::cerr << getExceptionMessage(exception, true) << std::endl;
             has_exception = true;
             data.chunk.clear();
             data.exception = std::move(exception);

src/Processors/Sinks/SinkToStorage.h

@@ -37,7 +37,7 @@ using ExceptionKeepingTransformRuntimeDataPtr = std::shared_ptr<ExceptionKeeping
 /// In case of exception, it is additionally pushed into pipeline.
 class ExceptionKeepingTransform : public IProcessor
 {
-private:
+protected:
     InputPort & input;
     OutputPort & output;
     Port::Data data;
@@ -48,7 +48,7 @@ private:
     bool was_on_start_called = false;
     bool was_on_finish_called = false;
 
-protected:
+//protected:
     virtual void transform(Chunk & chunk) = 0;
     virtual void onStart() {}
     virtual void onFinish() {}

src/Processors/Transforms/SquashingChunksTransform.cpp

@@ -6,23 +6,33 @@ namespace DB
 
 SquashingChunksTransform::SquashingChunksTransform(
     const Block & header, size_t min_block_size_rows, size_t min_block_size_bytes, bool reserve_memory)
-    : IAccumulatingTransform(header, header)
+    : ExceptionKeepingTransform(header, header)
     , squashing(min_block_size_rows, min_block_size_bytes, reserve_memory)
 {
 }
 
-void SquashingChunksTransform::consume(Chunk chunk)
+void SquashingChunksTransform::transform(Chunk & chunk)
 {
     if (auto block = squashing.add(getInputPort().getHeader().cloneWithColumns(chunk.detachColumns())))
     {
-        setReadyChunk(Chunk(block.getColumns(), block.rows()));
+        chunk.setColumns(block.getColumns(), block.rows());
     }
 }
 
-Chunk SquashingChunksTransform::generate()
+void SquashingChunksTransform::onFinish()
 {
     auto block = squashing.add({});
-    return Chunk(block.getColumns(), block.rows());
+    finish_chunk.setColumns(block.getColumns(), block.rows());
+}
+
+void SquashingChunksTransform::work()
+{
+    ExceptionKeepingTransform::work();
+    if (finish_chunk)
+    {
+        data.chunk = std::move(finish_chunk);
+        ready_output = true;
+    }
 }
 
 }

src/Processors/Transforms/SquashingChunksTransform.h

@@ -1,11 +1,11 @@
 #pragma once
-#include <Processors/IAccumulatingTransform.h>
+#include <Processors/Sinks/SinkToStorage.h>
 #include <DataStreams/SquashingTransform.h>
 
 namespace DB
 {
 
-class SquashingChunksTransform : public IAccumulatingTransform
+class SquashingChunksTransform : public ExceptionKeepingTransform
 {
 public:
     explicit SquashingChunksTransform(
@@ -13,12 +13,16 @@ public:
 
     String getName() const override { return "SquashingTransform"; }
 
+    void work() override;
+
 protected:
-    void consume(Chunk chunk) override;
-    Chunk generate() override;
+    void transform(Chunk & chunk) override;
+    void onFinish() override;
+
 
 private:
     SquashingTransform squashing;
+    Chunk finish_chunk;
 };
 
 }

src/Storages/IStorage.cpp

@@ -11,6 +11,7 @@
 #include <Parsers/ASTSetQuery.h>
 #include <Processors/Pipe.h>
 #include <Processors/QueryPlan/ReadFromPreparedSource.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Storages/AlterCommands.h>
 
 

src/Storages/StorageBuffer.cpp

@@ -26,6 +26,7 @@
 #include <Processors/Transforms/ExpressionTransform.h>
 #include <Processors/Sources/SourceFromInputStream.h>
 #include <Processors/Sinks/SinkToStorage.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/SettingQuotaAndLimitsStep.h>
 #include <Processors/QueryPlan/ReadFromPreparedSource.h>
 #include <Processors/QueryPlan/UnionStep.h>

src/Storages/StorageDistributed.cpp

@@ -53,6 +53,7 @@
 #include <Interpreters/getTableExpressions.h>
 #include <Functions/IFunction.h>
 
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/BuildQueryPipelineSettings.h>
 #include <Processors/QueryPlan/Optimizations/QueryPlanOptimizationSettings.h>
 #include <Processors/QueryPlan/ReadFromPreparedSource.h>

src/Storages/StorageMaterializedView.cpp

@@ -22,6 +22,7 @@
 #include <Common/typeid_cast.h>
 #include <Common/checkStackSize.h>
 #include <Processors/Sources/SourceFromInputStream.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/SettingQuotaAndLimitsStep.h>
 #include <Processors/QueryPlan/ExpressionStep.h>
 #include <Processors/QueryPlan/BuildQueryPipelineSettings.h>

src/Storages/StorageMergeTree.cpp

@@ -24,6 +24,7 @@
 #include <Storages/MergeTree/MergeList.h>
 #include <Storages/MergeTree/checkDataPart.h>
 #include <Processors/Pipe.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/BuildQueryPipelineSettings.h>
 #include <Processors/QueryPlan/Optimizations/QueryPlanOptimizationSettings.h>
 #include <optional>

src/Storages/StorageReplicatedMergeTree.cpp

@@ -42,6 +42,7 @@
 #include <Parsers/ASTCheckQuery.h>
 #include <Parsers/ASTSetQuery.h>
 
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/BuildQueryPipelineSettings.h>
 #include <Processors/QueryPlan/Optimizations/QueryPlanOptimizationSettings.h>
 

src/Storages/StorageView.cpp

@@ -16,6 +16,7 @@
 
 #include <Processors/Pipe.h>
 #include <Processors/Transforms/MaterializingTransform.h>
+#include <Processors/QueryPlan/QueryPlan.h>
 #include <Processors/QueryPlan/ExpressionStep.h>
 #include <Processors/QueryPlan/SettingQuotaAndLimitsStep.h>
 #include <Processors/QueryPlan/BuildQueryPipelineSettings.h>