From da6d43ea8ac2fa4ecfbe71b932304ba5b6fa0cff Mon Sep 17 00:00:00 2001
From: Nikolai Kochetov <nik-kochetov@yandex-team.ru>
Date: Fri, 14 Feb 2020 17:00:39 +0300
Subject: [PATCH] Fix TreeExecutor.

---
 .../Processors/Executors/TreeExecutorBlockInputStream.cpp    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/dbms/src/Processors/Executors/TreeExecutorBlockInputStream.cpp b/dbms/src/Processors/Executors/TreeExecutorBlockInputStream.cpp
index 44967440957..00df9d24780 100644
--- a/dbms/src/Processors/Executors/TreeExecutorBlockInputStream.cpp
+++ b/dbms/src/Processors/Executors/TreeExecutorBlockInputStream.cpp
@@ -2,12 +2,17 @@
 #include <Processors/Sources/SourceWithProgress.h>
 #include <Interpreters/ProcessList.h>
 #include <stack>
+#include <Processors/Sources/SourceFromInputStream.h>
 
 namespace DB
 {
 
 static void checkProcessorHasSingleOutput(IProcessor * processor)
 {
+    /// SourceFromInputStream may have totals port. Skip this check.
+    if (typeid_cast<const SourceFromInputStream *>(processor))
+        return;
+
     size_t num_outputs = processor->getOutputs().size();
     if (num_outputs != 1)
         throw Exception("All processors in TreeExecutorBlockInputStream must have single output, "