thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-24 16:42:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated script [#METR-21075].

Browse Source
This commit is contained in:
Alexey Milovidov 2016-04-27 07:13:52 +03:00
parent 388d3c5605
commit 4e62da8abc
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
utils/compressor/zstd_test.cpp
View File

@ -12,11 +12,18 @@ int main(int argc, char ** argv)
std::vector<char> src_buf(size);
std::vector<char> dst_buf;
ssize_t read_res = read(STDIN_FILENO, &src_buf[0], size);
if (read_res <= 0)
size_t pos = 0;
while (true)
{
ssize_t read_res = read(STDIN_FILENO, &src_buf[pos], size - pos);
if (read_res < 0)
throw std::runtime_error("Cannot read from stdin");
if (read_res == 0)
break;
pos += read_res;
}
src_buf.resize(read_res);
src_buf.resize(pos);
size_t zstd_res;
@ -47,9 +54,14 @@ int main(int argc, char ** argv)
dst_buf.resize(zstd_res);
ssize_t write_res = write(STDOUT_FILENO, &dst_buf[0], dst_buf.size());
if (write_res != static_cast<ssize_t>(dst_buf.size()))
pos = 0;
while (pos < dst_buf.size())
{
ssize_t write_res = write(STDOUT_FILENO, &dst_buf[pos], dst_buf.size());
if (write_res <= 0)
throw std::runtime_error("Cannot write to stdout");
pos += write_res;
}
return 0;
}