ClickHouse/dbms/src/DataStreams/CountingBlockOutputStream.cpp

#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);
}

}