mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 01:12:12 +00:00
33 lines
745 B
C++
33 lines
745 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.read_rows);
|
|
ProfileEvents::increment(ProfileEvents::InsertedBytes, local_progress.read_bytes);
|
|
|
|
if (process_elem)
|
|
process_elem->updateProgressOut(local_progress);
|
|
|
|
if (progress_callback)
|
|
progress_callback(local_progress);
|
|
}
|
|
|
|
}
|