2016-12-06 20:55:13 +00:00
|
|
|
#pragma once
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <DataStreams/IBlockOutputStream.h>
|
|
|
|
#include <DataStreams/IProfilingBlockInputStream.h>
|
|
|
|
#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
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
void setProcessListElement(ProcessListElement * elem)
|
|
|
|
{
|
|
|
|
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(); }
|
2017-04-01 07:20:54 +00:00
|
|
|
void onProgress(const Progress & progress) override { stream->onProgress(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;
|
|
|
|
ProcessListElement * process_elem = nullptr;
|
2016-12-06 20:55:13 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|