ClickHouse/dbms/src/DataStreams/CountingBlockOutputStream.cpp

33 lines
735 B
C++

#include <DataStreams/CountingBlockOutputStream.h>
#include <Common/ProfileEvents.h>
namespace ProfileEvents
{
extern const Event InsertedRows;
extern const Event InsertedBytes;
}
namespace DB
{
void CountingBlockOutputStream::write(const Block & block)
{
stream->write(block);
Progress local_progress(block.rows(), block.bytes(), 0);
progress.incrementPiecewiseAtomically(local_progress);
ProfileEvents::increment(ProfileEvents::InsertedRows, local_progress.rows);
ProfileEvents::increment(ProfileEvents::InsertedBytes, local_progress.bytes);
if (process_elem)
process_elem->updateProgressOut(local_progress);
if (progress_callback)
progress_callback(local_progress);
}
}