mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-11 17:02:25 +00:00
87 lines
2.1 KiB
C++
87 lines
2.1 KiB
C++
#include <base/types.h>
|
|
#include <Common/Exception.h>
|
|
#include <IO/ReadBuffer.h>
|
|
#include <IO/WriteBuffer.h>
|
|
#include <IO/VarInt.h>
|
|
#include <IO/ReadHelpers.h>
|
|
#include <IO/WriteHelpers.h>
|
|
#include <Core/BlockInfo.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int UNKNOWN_BLOCK_INFO_FIELD;
|
|
}
|
|
|
|
|
|
/// Write values in binary form. NOTE: You could use protobuf, but it would be overkill for this case.
|
|
void BlockInfo::write(WriteBuffer & out) const
|
|
{
|
|
/// Set of pairs `FIELD_NUM`, value in binary form. Then 0.
|
|
#define WRITE_FIELD(TYPE, NAME, DEFAULT, FIELD_NUM) \
|
|
writeVarUInt(FIELD_NUM, out); \
|
|
writeBinary(NAME, out);
|
|
|
|
APPLY_FOR_BLOCK_INFO_FIELDS(WRITE_FIELD)
|
|
|
|
#undef WRITE_FIELD
|
|
writeVarUInt(0, out);
|
|
}
|
|
|
|
/// Read values in binary form.
|
|
void BlockInfo::read(ReadBuffer & in)
|
|
{
|
|
UInt64 field_num = 0;
|
|
|
|
while (true)
|
|
{
|
|
readVarUInt(field_num, in);
|
|
if (field_num == 0)
|
|
break;
|
|
|
|
switch (field_num)
|
|
{
|
|
#define READ_FIELD(TYPE, NAME, DEFAULT, FIELD_NUM) \
|
|
case FIELD_NUM: \
|
|
readBinary(NAME, in); \
|
|
break;
|
|
|
|
APPLY_FOR_BLOCK_INFO_FIELDS(READ_FIELD)
|
|
|
|
#undef READ_FIELD
|
|
default:
|
|
throw Exception("Unknown BlockInfo field number: " + toString(field_num), ErrorCodes::UNKNOWN_BLOCK_INFO_FIELD);
|
|
}
|
|
}
|
|
}
|
|
|
|
void BlockMissingValues::setBit(size_t column_idx, size_t row_idx)
|
|
{
|
|
RowsBitMask & mask = rows_mask_by_column_id[column_idx];
|
|
mask.resize(row_idx + 1);
|
|
mask[row_idx] = true;
|
|
}
|
|
|
|
const BlockMissingValues::RowsBitMask & BlockMissingValues::getDefaultsBitmask(size_t column_idx) const
|
|
{
|
|
static RowsBitMask none;
|
|
auto it = rows_mask_by_column_id.find(column_idx);
|
|
if (it != rows_mask_by_column_id.end())
|
|
return it->second;
|
|
return none;
|
|
}
|
|
|
|
bool BlockMissingValues::hasDefaultBits(size_t column_idx) const
|
|
{
|
|
auto it = rows_mask_by_column_id.find(column_idx);
|
|
if (it == rows_mask_by_column_id.end())
|
|
return false;
|
|
|
|
const auto & col_mask = it->second;
|
|
return std::find(col_mask.begin(), col_mask.end(), true) != col_mask.end();
|
|
}
|
|
}
|