From 5ea90e6961065f64f513d69a6d62a888f7c57643 Mon Sep 17 00:00:00 2001
From: Ivan Lezhankin <ilezhankin@yandex-team.ru>
Date: Mon, 28 Jun 2021 23:04:14 +0300
Subject: [PATCH] Restore missing file

---
 .../NullAndDoCopyBlockInputStream.h           | 56 +++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 src/DataStreams/NullAndDoCopyBlockInputStream.h

diff --git a/src/DataStreams/NullAndDoCopyBlockInputStream.h b/src/DataStreams/NullAndDoCopyBlockInputStream.h
new file mode 100644
index 00000000000..8bfb3538f3a
--- /dev/null
+++ b/src/DataStreams/NullAndDoCopyBlockInputStream.h
@@ -0,0 +1,56 @@
+#pragma once
+
+#include <DataStreams/IBlockInputStream.h>
+#include <DataStreams/copyData.h>
+
+
+namespace DB
+{
+
+class IBlockOutputStream;
+using BlockOutputStreamPtr = std::shared_ptr<IBlockOutputStream>;
+
+
+/** An empty stream of blocks.
+  * But at the first read attempt, copies the data from the passed `input` to the `output`.
+  * This is necessary to execute the query INSERT SELECT - the query copies data, but returns nothing.
+  * The query could be executed without wrapping it in an empty BlockInputStream,
+  *  but the progress of query execution and the ability to cancel the query would not work.
+  */
+class NullAndDoCopyBlockInputStream : public IBlockInputStream
+{
+public:
+    NullAndDoCopyBlockInputStream(const BlockInputStreamPtr & input_, BlockOutputStreamPtr output_)
+        : input(std::move(input_))
+        , output(std::move(output_))
+    {
+        children.push_back(input);
+    }
+
+    /// Suppress readPrefix and readSuffix, because they are called by copyData.
+    void readPrefix() override {}
+    void readSuffix() override {}
+
+    String getName() const override { return "NullAndDoCopy"; }
+
+    Block getHeader() const override { return {}; }
+    Block getTotals() override { return {}; }
+    Block getExtremes() override { return {}; }
+
+protected:
+    Block readImpl() override
+    {
+        /// We do not use cancel flag here.
+        /// If query was cancelled, it will be processed by child streams.
+        /// Part of the data will be processed.
+
+        copyData(*input, *output);
+        return Block();
+    }
+
+private:
+    BlockInputStreamPtr input;
+    BlockOutputStreamPtr output;
+};
+
+}