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

tabs -> spaces

Browse Source
This commit is contained in:
Vladislav Smirnov 2019-03-13 14:05:52 +03:00
parent 22f4852830
commit 7d822d8081
1 changed files with 33 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
dbms/src/Formats/CapnProtoRowInputStream.cpp
View File

@ -221,39 +221,38 @@ bool CapnProtoRowInputStream::read(MutableColumns & columns, RowReadExtension &)
kj::ArrayPtr<const capnp::word> array;
size_t bytesSize = buf.size() - istr.offset();
size_t remainingBytes = 0;
if (bytesSize < sizeof(capnp::word)) // case when we read less than 8 bytes (capnp::word)
{
char edgeBytes[sizeof(capnp::word)];
while (bytesSize + remainingBytes < sizeof(capnp::word))
{
istr.readStrict(edgeBytes + remainingBytes, bytesSize);
remainingBytes += bytesSize;
istr.next();
bytesSize = buf.size();
}
size_t remainingBytes = 0;
if (bytesSize < sizeof(capnp::word)) // case when we read less than 8 bytes (capnp::word)
{
char edgeBytes[sizeof(capnp::word)];
while (bytesSize + remainingBytes < sizeof(capnp::word))
{
istr.readStrict(edgeBytes + remainingBytes, bytesSize);
remainingBytes += bytesSize;
istr.next();
bytesSize = buf.size();
}
auto wordsSize = bytesSize / sizeof(capnp::word) + 1;
heap_array = kj::heapArray<capnp::word>(wordsSize + 1);
auto chars_heap_array = heap_array.asChars();
::memcpy(chars_heap_array.begin(), edgeBytes, remainingBytes);
::memcpy(chars_heap_array.begin() + remainingBytes, buf.begin(), buf.size());
array = heap_array.asPtr();
}
else
{
auto wordsSize = bytesSize / sizeof(capnp::word);
array = kj::arrayPtr(base, wordsSize);
auto expected_words = capnp::expectedSizeInWordsFromPrefix(array);
if (expected_words * sizeof(capnp::word) > array.size())
{
// We'll need to reassemble the message in a contiguous buffer
heap_array = kj::heapArray<capnp::word>(expected_words);
istr.readStrict(heap_array.asChars().begin(), heap_array.asChars().size());
array = heap_array.asPtr();
}
}
auto wordsSize = bytesSize / sizeof(capnp::word) + 1;
heap_array = kj::heapArray<capnp::word>(wordsSize + 1);
auto chars_heap_array = heap_array.asChars();
::memcpy(chars_heap_array.begin(), edgeBytes, remainingBytes);
::memcpy(chars_heap_array.begin() + remainingBytes, buf.begin(), buf.size());
array = heap_array.asPtr();
}
else
{
auto wordsSize = bytesSize / sizeof(capnp::word);
array = kj::arrayPtr(base, wordsSize);
auto expected_words = capnp::expectedSizeInWordsFromPrefix(array);
if (expected_words * sizeof(capnp::word) > array.size())
{
// We'll need to reassemble the message in a contiguous buffer
heap_array = kj::heapArray<capnp::word>(expected_words);
istr.readStrict(heap_array.asChars().begin(), heap_array.asChars().size());
array = heap_array.asPtr();
}
}
#if CAPNP_VERSION >= 8000
capnp::UnalignedFlatArrayMessageReader msg(array);
@ -316,8 +315,8 @@ bool CapnProtoRowInputStream::read(MutableColumns & columns, RowReadExtension &)
// Advance buffer position if used with remaining bytes from previous buffer
else if (remainingBytes != 0)
{
auto parsed = (msg.getEnd() - heap_array.begin()) * sizeof(capnp::word) - remainingBytes;
istr.position() += parsed;
auto parsed = (msg.getEnd() - heap_array.begin()) * sizeof(capnp::word) - remainingBytes;
istr.position() += parsed;
}
return true;