2016-12-06 20:55:13 +00:00
|
|
|
#pragma once
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <DataStreams/IBlockOutputStream.h>
|
2019-01-23 14:48:50 +00:00
|
|
|
#include <DataStreams/IBlockInputStream.h>
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Interpreters/ProcessList.h>
|
2016-12-06 20:55:13 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
|
|
/// Proxy class which counts number of written block, rows, bytes
|
|
|
|
class CountingBlockOutputStream : public IBlockOutputStream
|
|
|
|
{
|
|
|
|
public:
|
2017-04-01 07:20:54 +00:00
|
|
|
CountingBlockOutputStream(const BlockOutputStreamPtr & stream_)
|
|
|
|
: stream(stream_) {}
|
2016-12-06 20:55:13 +00:00
|
|
|
|
2017-09-08 03:47:27 +00:00
|
|
|
void setProgressCallback(const ProgressCallback & callback)
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
|
|
|
progress_callback = callback;
|
|
|
|
}
|
2016-12-06 20:55:13 +00:00
|
|
|
|
2018-02-01 17:55:08 +00:00
|
|
|
void setProcessListElement(QueryStatus * elem)
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
|
|
|
process_elem = elem;
|
|
|
|
}
|
2016-12-06 20:55:13 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
const Progress & getProgress() const
|
|
|
|
{
|
|
|
|
return progress;
|
|
|
|
}
|
2016-12-06 20:55:13 +00:00
|
|
|
|
2018-02-19 00:45:32 +00:00
|
|
|
Block getHeader() const override { return stream->getHeader(); }
|
2017-04-01 07:20:54 +00:00
|
|
|
void write(const Block & block) override;
|
2016-12-06 20:55:13 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
void writePrefix() override { stream->writePrefix(); }
|
|
|
|
void writeSuffix() override { stream->writeSuffix(); }
|
2017-09-08 03:47:27 +00:00
|
|
|
void flush() override { stream->flush(); }
|
2018-06-03 17:43:56 +00:00
|
|
|
void onProgress(const Progress & current_progress) override { stream->onProgress(current_progress); }
|
2017-09-08 03:47:27 +00:00
|
|
|
String getContentType() const override { return stream->getContentType(); }
|
2016-12-06 20:55:13 +00:00
|
|
|
|
|
|
|
protected:
|
2017-04-01 07:20:54 +00:00
|
|
|
BlockOutputStreamPtr stream;
|
|
|
|
Progress progress;
|
|
|
|
ProgressCallback progress_callback;
|
2018-02-01 17:55:08 +00:00
|
|
|
QueryStatus * process_elem = nullptr;
|
2016-12-06 20:55:13 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|