2017-11-24 13:55:31 +00:00
|
|
|
#include "Progress.h"
|
2017-01-22 15:03:55 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <IO/ReadBuffer.h>
|
|
|
|
#include <IO/WriteBuffer.h>
|
|
|
|
#include <IO/ReadHelpers.h>
|
|
|
|
#include <IO/WriteHelpers.h>
|
2017-01-22 15:03:55 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2019-05-13 04:48:09 +00:00
|
|
|
void ProgressValues::read(ReadBuffer & in, UInt64 server_revision)
|
2017-01-22 15:03:55 +00:00
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
size_t new_rows = 0;
|
|
|
|
size_t new_bytes = 0;
|
|
|
|
size_t new_total_rows = 0;
|
2019-04-25 11:25:49 +00:00
|
|
|
size_t new_write_rows = 0;
|
|
|
|
size_t new_write_bytes = 0;
|
2019-05-06 06:57:48 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
readVarUInt(new_rows, in);
|
|
|
|
readVarUInt(new_bytes, in);
|
2017-08-16 20:27:35 +00:00
|
|
|
readVarUInt(new_total_rows, in);
|
2019-05-13 04:48:09 +00:00
|
|
|
if (DBMS_MIN_REVISION_WITH_CLIENT_WRITE_INFO >= server_revision)
|
|
|
|
{
|
|
|
|
readVarUInt(new_write_rows, in);
|
|
|
|
readVarUInt(new_write_bytes, in);
|
|
|
|
}
|
|
|
|
|
|
|
|
this->rows = new_rows;
|
|
|
|
this->bytes = new_bytes;
|
|
|
|
this->total_rows = new_total_rows;
|
|
|
|
this->write_rows = new_write_rows;
|
|
|
|
this->write_bytes = new_write_bytes;
|
2019-05-06 06:57:48 +00:00
|
|
|
}
|
2017-01-22 15:03:55 +00:00
|
|
|
|
|
|
|
|
2019-05-13 04:48:09 +00:00
|
|
|
void ProgressValues::write(WriteBuffer & out, UInt64 client_revision) const
|
2017-01-22 15:03:55 +00:00
|
|
|
{
|
2019-05-13 04:48:09 +00:00
|
|
|
writeVarUInt(this->rows, out);
|
|
|
|
writeVarUInt(this->bytes, out);
|
|
|
|
writeVarUInt(this->total_rows, out);
|
|
|
|
if (client_revision >= DBMS_MIN_REVISION_WITH_CLIENT_WRITE_INFO)
|
|
|
|
{
|
|
|
|
writeVarUInt(this->write_rows, out);
|
|
|
|
writeVarUInt(this->write_bytes, out);
|
|
|
|
}
|
2017-01-22 15:03:55 +00:00
|
|
|
}
|
|
|
|
|
2019-05-13 04:48:09 +00:00
|
|
|
void ProgressValues::writeJSON(WriteBuffer & out) const
|
2017-01-22 15:03:55 +00:00
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
/// Numbers are written in double quotes (as strings) to avoid loss of precision
|
|
|
|
/// of 64-bit integers after interpretation by JavaScript.
|
|
|
|
|
|
|
|
writeCString("{\"read_rows\":\"", out);
|
2019-05-13 04:48:09 +00:00
|
|
|
writeText(this->rows, out);
|
2017-04-01 07:20:54 +00:00
|
|
|
writeCString("\",\"read_bytes\":\"", out);
|
2019-05-13 04:48:09 +00:00
|
|
|
writeText(this->bytes, out);
|
|
|
|
writeCString("\",\"written_rows\":\"", out);
|
|
|
|
writeText(this->write_rows, out);
|
|
|
|
writeCString("\",\"written_bytes\":\"", out);
|
|
|
|
writeText(this->write_bytes, out);
|
2019-05-13 04:56:08 +00:00
|
|
|
writeCString("\",\"rows_in_set\":\"", out);
|
2019-05-13 04:48:09 +00:00
|
|
|
writeText(this->total_rows, out);
|
2017-04-01 07:20:54 +00:00
|
|
|
writeCString("\"}", out);
|
2017-01-22 15:03:55 +00:00
|
|
|
}
|
|
|
|
|
2019-05-13 04:48:09 +00:00
|
|
|
void Progress::read(ReadBuffer & in, UInt64 server_revision)
|
2019-05-06 06:14:37 +00:00
|
|
|
{
|
2019-05-13 04:48:09 +00:00
|
|
|
ProgressValues values;
|
|
|
|
values.read(in, server_revision);
|
|
|
|
|
|
|
|
rows.store(values.rows, std::memory_order_relaxed);
|
|
|
|
bytes.store(values.bytes, std::memory_order_relaxed);
|
|
|
|
total_rows.store(values.total_rows, std::memory_order_relaxed);
|
|
|
|
write_rows.store(values.write_rows, std::memory_order_relaxed);
|
|
|
|
write_bytes.store(values.write_bytes, std::memory_order_relaxed);
|
2019-05-06 06:57:48 +00:00
|
|
|
}
|
2019-05-06 06:14:37 +00:00
|
|
|
|
2019-05-13 04:48:09 +00:00
|
|
|
void Progress::write(WriteBuffer & out, UInt64 client_revision) const
|
2019-05-06 06:14:37 +00:00
|
|
|
{
|
2019-05-13 04:48:09 +00:00
|
|
|
getValues().write(out, client_revision);
|
2019-05-06 06:14:37 +00:00
|
|
|
}
|
|
|
|
|
2019-05-13 04:48:09 +00:00
|
|
|
void Progress::writeJSON(WriteBuffer & out) const
|
2019-04-26 03:38:39 +00:00
|
|
|
{
|
2019-05-13 04:48:09 +00:00
|
|
|
getValues().writeJSON(out);
|
2018-03-09 23:04:26 +00:00
|
|
|
}
|
|
|
|
|
2017-01-22 15:03:55 +00:00
|
|
|
}
|