mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-13 19:14:30 +00:00
801 lines
29 KiB
C++
801 lines
29 KiB
C++
#include "HashedDictionary.h"
|
|
#include <ext/size.h>
|
|
#include "DictionaryBlockInputStream.h"
|
|
#include "DictionaryFactory.h"
|
|
|
|
namespace
|
|
{
|
|
|
|
/// NOTE: Trailing return type is explicitly specified for SFINAE.
|
|
|
|
/// google::sparse_hash_map
|
|
template <typename T> auto first(const T & value) -> decltype(value.first) { return value.first; } // NOLINT
|
|
template <typename T> auto second(const T & value) -> decltype(value.second) { return value.second; } // NOLINT
|
|
|
|
/// HashMap
|
|
template <typename T> auto first(const T & value) -> decltype(value.getKey()) { return value.getKey(); } // NOLINT
|
|
template <typename T> auto second(const T & value) -> decltype(value.getMapped()) { return value.getMapped(); } // NOLINT
|
|
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int TYPE_MISMATCH;
|
|
extern const int BAD_ARGUMENTS;
|
|
extern const int DICTIONARY_IS_EMPTY;
|
|
extern const int UNSUPPORTED_METHOD;
|
|
}
|
|
|
|
|
|
HashedDictionary::HashedDictionary(
|
|
const std::string & database_,
|
|
const std::string & name_,
|
|
const DictionaryStructure & dict_struct_,
|
|
DictionarySourcePtr source_ptr_,
|
|
const DictionaryLifetime dict_lifetime_,
|
|
bool require_nonempty_,
|
|
bool sparse_,
|
|
BlockPtr saved_block_)
|
|
: database(database_)
|
|
, name(name_)
|
|
, full_name{database_.empty() ? name_ : (database_ + "." + name_)}
|
|
, dict_struct(dict_struct_)
|
|
, source_ptr{std::move(source_ptr_)}
|
|
, dict_lifetime(dict_lifetime_)
|
|
, require_nonempty(require_nonempty_)
|
|
, sparse(sparse_)
|
|
, saved_block{std::move(saved_block_)}
|
|
{
|
|
createAttributes();
|
|
loadData();
|
|
calculateBytesAllocated();
|
|
}
|
|
|
|
|
|
void HashedDictionary::toParent(const PaddedPODArray<Key> & ids, PaddedPODArray<Key> & out) const
|
|
{
|
|
const auto null_value = std::get<UInt64>(hierarchical_attribute->null_values);
|
|
|
|
getItemsImpl<UInt64, UInt64>(
|
|
*hierarchical_attribute,
|
|
ids,
|
|
[&](const size_t row, const UInt64 value) { out[row] = value; },
|
|
[&](const size_t) { return null_value; });
|
|
}
|
|
|
|
|
|
/// Allow to use single value in same way as array.
|
|
static inline HashedDictionary::Key getAt(const PaddedPODArray<HashedDictionary::Key> & arr, const size_t idx)
|
|
{
|
|
return arr[idx];
|
|
}
|
|
static inline HashedDictionary::Key getAt(const HashedDictionary::Key & value, const size_t)
|
|
{
|
|
return value;
|
|
}
|
|
|
|
template <typename AttrType, typename ChildType, typename AncestorType>
|
|
void HashedDictionary::isInAttrImpl(const AttrType & attr, const ChildType & child_ids, const AncestorType & ancestor_ids, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
const auto null_value = std::get<UInt64>(hierarchical_attribute->null_values);
|
|
const auto rows = out.size();
|
|
|
|
for (const auto row : ext::range(0, rows))
|
|
{
|
|
auto id = getAt(child_ids, row);
|
|
const auto ancestor_id = getAt(ancestor_ids, row);
|
|
|
|
while (id != null_value && id != ancestor_id)
|
|
{
|
|
auto it = attr.find(id);
|
|
if (it != std::end(attr))
|
|
id = second(*it);
|
|
else
|
|
break;
|
|
}
|
|
|
|
out[row] = id != null_value && id == ancestor_id;
|
|
}
|
|
|
|
query_count.fetch_add(rows, std::memory_order_relaxed);
|
|
}
|
|
template <typename ChildType, typename AncestorType>
|
|
void HashedDictionary::isInImpl(const ChildType & child_ids, const AncestorType & ancestor_ids, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
if (!sparse)
|
|
return isInAttrImpl(*std::get<CollectionPtrType<Key>>(hierarchical_attribute->maps), child_ids, ancestor_ids, out);
|
|
return isInAttrImpl(*std::get<SparseCollectionPtrType<Key>>(hierarchical_attribute->sparse_maps), child_ids, ancestor_ids, out);
|
|
}
|
|
|
|
void HashedDictionary::isInVectorVector(
|
|
const PaddedPODArray<Key> & child_ids, const PaddedPODArray<Key> & ancestor_ids, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
isInImpl(child_ids, ancestor_ids, out);
|
|
}
|
|
|
|
void HashedDictionary::isInVectorConstant(const PaddedPODArray<Key> & child_ids, const Key ancestor_id, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
isInImpl(child_ids, ancestor_id, out);
|
|
}
|
|
|
|
void HashedDictionary::isInConstantVector(const Key child_id, const PaddedPODArray<Key> & ancestor_ids, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
isInImpl(child_id, ancestor_ids, out);
|
|
}
|
|
|
|
|
|
#define DECLARE(TYPE) \
|
|
void HashedDictionary::get##TYPE(const std::string & attribute_name, const PaddedPODArray<Key> & ids, ResultArrayType<TYPE> & out) \
|
|
const \
|
|
{ \
|
|
const auto & attribute = getAttribute(attribute_name); \
|
|
checkAttributeType(full_name, attribute_name, attribute.type, AttributeUnderlyingType::ut##TYPE); \
|
|
\
|
|
const auto null_value = std::get<TYPE>(attribute.null_values); \
|
|
\
|
|
getItemsImpl<TYPE, TYPE>( \
|
|
attribute, ids, [&](const size_t row, const auto value) { out[row] = value; }, [&](const size_t) { return null_value; }); \
|
|
}
|
|
DECLARE(UInt8)
|
|
DECLARE(UInt16)
|
|
DECLARE(UInt32)
|
|
DECLARE(UInt64)
|
|
DECLARE(UInt128)
|
|
DECLARE(Int8)
|
|
DECLARE(Int16)
|
|
DECLARE(Int32)
|
|
DECLARE(Int64)
|
|
DECLARE(Float32)
|
|
DECLARE(Float64)
|
|
DECLARE(Decimal32)
|
|
DECLARE(Decimal64)
|
|
DECLARE(Decimal128)
|
|
#undef DECLARE
|
|
|
|
void HashedDictionary::getString(const std::string & attribute_name, const PaddedPODArray<Key> & ids, ColumnString * out) const
|
|
{
|
|
const auto & attribute = getAttribute(attribute_name);
|
|
checkAttributeType(full_name, attribute_name, attribute.type, AttributeUnderlyingType::utString);
|
|
|
|
const auto & null_value = StringRef{std::get<String>(attribute.null_values)};
|
|
|
|
getItemsImpl<StringRef, StringRef>(
|
|
attribute,
|
|
ids,
|
|
[&](const size_t, const StringRef value) { out->insertData(value.data, value.size); },
|
|
[&](const size_t) { return null_value; });
|
|
}
|
|
|
|
#define DECLARE(TYPE) \
|
|
void HashedDictionary::get##TYPE( \
|
|
const std::string & attribute_name, \
|
|
const PaddedPODArray<Key> & ids, \
|
|
const PaddedPODArray<TYPE> & def, \
|
|
ResultArrayType<TYPE> & out) const \
|
|
{ \
|
|
const auto & attribute = getAttribute(attribute_name); \
|
|
checkAttributeType(full_name, attribute_name, attribute.type, AttributeUnderlyingType::ut##TYPE); \
|
|
\
|
|
getItemsImpl<TYPE, TYPE>( \
|
|
attribute, ids, [&](const size_t row, const auto value) { out[row] = value; }, [&](const size_t row) { return def[row]; }); \
|
|
}
|
|
DECLARE(UInt8)
|
|
DECLARE(UInt16)
|
|
DECLARE(UInt32)
|
|
DECLARE(UInt64)
|
|
DECLARE(UInt128)
|
|
DECLARE(Int8)
|
|
DECLARE(Int16)
|
|
DECLARE(Int32)
|
|
DECLARE(Int64)
|
|
DECLARE(Float32)
|
|
DECLARE(Float64)
|
|
DECLARE(Decimal32)
|
|
DECLARE(Decimal64)
|
|
DECLARE(Decimal128)
|
|
#undef DECLARE
|
|
|
|
void HashedDictionary::getString(
|
|
const std::string & attribute_name, const PaddedPODArray<Key> & ids, const ColumnString * const def, ColumnString * const out) const
|
|
{
|
|
const auto & attribute = getAttribute(attribute_name);
|
|
checkAttributeType(full_name, attribute_name, attribute.type, AttributeUnderlyingType::utString);
|
|
|
|
getItemsImpl<StringRef, StringRef>(
|
|
attribute,
|
|
ids,
|
|
[&](const size_t, const StringRef value) { out->insertData(value.data, value.size); },
|
|
[&](const size_t row) { return def->getDataAt(row); });
|
|
}
|
|
|
|
#define DECLARE(TYPE) \
|
|
void HashedDictionary::get##TYPE( \
|
|
const std::string & attribute_name, const PaddedPODArray<Key> & ids, const TYPE & def, ResultArrayType<TYPE> & out) const \
|
|
{ \
|
|
const auto & attribute = getAttribute(attribute_name); \
|
|
checkAttributeType(full_name, attribute_name, attribute.type, AttributeUnderlyingType::ut##TYPE); \
|
|
\
|
|
getItemsImpl<TYPE, TYPE>( \
|
|
attribute, ids, [&](const size_t row, const auto value) { out[row] = value; }, [&](const size_t) { return def; }); \
|
|
}
|
|
DECLARE(UInt8)
|
|
DECLARE(UInt16)
|
|
DECLARE(UInt32)
|
|
DECLARE(UInt64)
|
|
DECLARE(UInt128)
|
|
DECLARE(Int8)
|
|
DECLARE(Int16)
|
|
DECLARE(Int32)
|
|
DECLARE(Int64)
|
|
DECLARE(Float32)
|
|
DECLARE(Float64)
|
|
DECLARE(Decimal32)
|
|
DECLARE(Decimal64)
|
|
DECLARE(Decimal128)
|
|
#undef DECLARE
|
|
|
|
void HashedDictionary::getString(
|
|
const std::string & attribute_name, const PaddedPODArray<Key> & ids, const String & def, ColumnString * const out) const
|
|
{
|
|
const auto & attribute = getAttribute(attribute_name);
|
|
checkAttributeType(full_name, attribute_name, attribute.type, AttributeUnderlyingType::utString);
|
|
|
|
getItemsImpl<StringRef, StringRef>(
|
|
attribute,
|
|
ids,
|
|
[&](const size_t, const StringRef value) { out->insertData(value.data, value.size); },
|
|
[&](const size_t) { return StringRef{def}; });
|
|
}
|
|
|
|
void HashedDictionary::has(const PaddedPODArray<Key> & ids, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
const auto & attribute = attributes.front();
|
|
|
|
switch (attribute.type)
|
|
{
|
|
case AttributeUnderlyingType::utUInt8:
|
|
has<UInt8>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt16:
|
|
has<UInt16>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt32:
|
|
has<UInt32>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt64:
|
|
has<UInt64>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt128:
|
|
has<UInt128>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utInt8:
|
|
has<Int8>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utInt16:
|
|
has<Int16>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utInt32:
|
|
has<Int32>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utInt64:
|
|
has<Int64>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utFloat32:
|
|
has<Float32>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utFloat64:
|
|
has<Float64>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utString:
|
|
has<StringRef>(attribute, ids, out);
|
|
break;
|
|
|
|
case AttributeUnderlyingType::utDecimal32:
|
|
has<Decimal32>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utDecimal64:
|
|
has<Decimal64>(attribute, ids, out);
|
|
break;
|
|
case AttributeUnderlyingType::utDecimal128:
|
|
has<Decimal128>(attribute, ids, out);
|
|
break;
|
|
}
|
|
}
|
|
|
|
void HashedDictionary::createAttributes()
|
|
{
|
|
const auto size = dict_struct.attributes.size();
|
|
attributes.reserve(size);
|
|
|
|
for (const auto & attribute : dict_struct.attributes)
|
|
{
|
|
attribute_index_by_name.emplace(attribute.name, attributes.size());
|
|
attributes.push_back(createAttributeWithType(attribute.underlying_type, attribute.null_value));
|
|
|
|
if (attribute.hierarchical)
|
|
{
|
|
hierarchical_attribute = &attributes.back();
|
|
|
|
if (hierarchical_attribute->type != AttributeUnderlyingType::utUInt64)
|
|
throw Exception{full_name + ": hierarchical attribute must be UInt64.", ErrorCodes::TYPE_MISMATCH};
|
|
}
|
|
}
|
|
}
|
|
|
|
void HashedDictionary::blockToAttributes(const Block & block)
|
|
{
|
|
const auto & id_column = *block.safeGetByPosition(0).column;
|
|
|
|
for (const size_t attribute_idx : ext::range(0, attributes.size()))
|
|
{
|
|
const IColumn & attribute_column = *block.safeGetByPosition(attribute_idx + 1).column;
|
|
auto & attribute = attributes[attribute_idx];
|
|
|
|
for (const auto row_idx : ext::range(0, id_column.size()))
|
|
if (setAttributeValue(attribute, id_column[row_idx].get<UInt64>(), attribute_column[row_idx]))
|
|
++element_count;
|
|
}
|
|
}
|
|
|
|
void HashedDictionary::updateData()
|
|
{
|
|
if (!saved_block || saved_block->rows() == 0)
|
|
{
|
|
auto stream = source_ptr->loadUpdatedAll();
|
|
stream->readPrefix();
|
|
|
|
while (const auto block = stream->read())
|
|
{
|
|
/// We are using this to keep saved data if input stream consists of multiple blocks
|
|
if (!saved_block)
|
|
saved_block = std::make_shared<DB::Block>(block.cloneEmpty());
|
|
for (const auto attribute_idx : ext::range(0, attributes.size() + 1))
|
|
{
|
|
const IColumn & update_column = *block.getByPosition(attribute_idx).column.get();
|
|
MutableColumnPtr saved_column = saved_block->getByPosition(attribute_idx).column->assumeMutable();
|
|
saved_column->insertRangeFrom(update_column, 0, update_column.size());
|
|
}
|
|
}
|
|
stream->readSuffix();
|
|
}
|
|
else
|
|
{
|
|
auto stream = source_ptr->loadUpdatedAll();
|
|
stream->readPrefix();
|
|
|
|
while (Block block = stream->read())
|
|
{
|
|
const auto & saved_id_column = *saved_block->safeGetByPosition(0).column;
|
|
const auto & update_id_column = *block.safeGetByPosition(0).column;
|
|
|
|
std::unordered_map<Key, std::vector<size_t>> update_ids;
|
|
for (size_t row = 0; row < update_id_column.size(); ++row)
|
|
{
|
|
const auto id = update_id_column.get64(row);
|
|
update_ids[id].push_back(row);
|
|
}
|
|
|
|
const size_t saved_rows = saved_id_column.size();
|
|
IColumn::Filter filter(saved_rows);
|
|
std::unordered_map<Key, std::vector<size_t>>::iterator it;
|
|
|
|
for (size_t row = 0; row < saved_id_column.size(); ++row)
|
|
{
|
|
auto id = saved_id_column.get64(row);
|
|
it = update_ids.find(id);
|
|
|
|
if (it != update_ids.end())
|
|
filter[row] = 0;
|
|
else
|
|
filter[row] = 1;
|
|
}
|
|
|
|
auto block_columns = block.mutateColumns();
|
|
for (const auto attribute_idx : ext::range(0, attributes.size() + 1))
|
|
{
|
|
auto & column = saved_block->safeGetByPosition(attribute_idx).column;
|
|
const auto & filtered_column = column->filter(filter, -1);
|
|
|
|
block_columns[attribute_idx]->insertRangeFrom(*filtered_column.get(), 0, filtered_column->size());
|
|
}
|
|
|
|
saved_block->setColumns(std::move(block_columns));
|
|
}
|
|
stream->readSuffix();
|
|
}
|
|
|
|
if (saved_block)
|
|
blockToAttributes(*saved_block.get());
|
|
}
|
|
|
|
void HashedDictionary::loadData()
|
|
{
|
|
if (!source_ptr->hasUpdateField())
|
|
{
|
|
auto stream = source_ptr->loadAll();
|
|
stream->readPrefix();
|
|
|
|
while (const auto block = stream->read())
|
|
blockToAttributes(block);
|
|
|
|
stream->readSuffix();
|
|
}
|
|
else
|
|
updateData();
|
|
|
|
if (require_nonempty && 0 == element_count)
|
|
throw Exception{full_name + ": dictionary source is empty and 'require_nonempty' property is set.", ErrorCodes::DICTIONARY_IS_EMPTY};
|
|
}
|
|
|
|
template <typename T>
|
|
void HashedDictionary::addAttributeSize(const Attribute & attribute)
|
|
{
|
|
if (!sparse)
|
|
{
|
|
const auto & map_ref = std::get<CollectionPtrType<T>>(attribute.maps);
|
|
bytes_allocated += sizeof(CollectionType<T>) + map_ref->getBufferSizeInBytes();
|
|
bucket_count = map_ref->getBufferSizeInCells();
|
|
}
|
|
else
|
|
{
|
|
const auto & map_ref = std::get<SparseCollectionPtrType<T>>(attribute.sparse_maps);
|
|
bucket_count = map_ref->bucket_count();
|
|
|
|
/** TODO: more accurate calculation */
|
|
bytes_allocated += sizeof(SparseCollectionType<T>);
|
|
bytes_allocated += bucket_count;
|
|
bytes_allocated += map_ref->size() * (sizeof(Key) + sizeof(T));
|
|
}
|
|
}
|
|
|
|
void HashedDictionary::calculateBytesAllocated()
|
|
{
|
|
bytes_allocated += attributes.size() * sizeof(attributes.front());
|
|
|
|
for (const auto & attribute : attributes)
|
|
{
|
|
switch (attribute.type)
|
|
{
|
|
case AttributeUnderlyingType::utUInt8:
|
|
addAttributeSize<UInt8>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt16:
|
|
addAttributeSize<UInt16>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt32:
|
|
addAttributeSize<UInt32>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt64:
|
|
addAttributeSize<UInt64>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt128:
|
|
addAttributeSize<UInt128>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utInt8:
|
|
addAttributeSize<Int8>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utInt16:
|
|
addAttributeSize<Int16>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utInt32:
|
|
addAttributeSize<Int32>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utInt64:
|
|
addAttributeSize<Int64>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utFloat32:
|
|
addAttributeSize<Float32>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utFloat64:
|
|
addAttributeSize<Float64>(attribute);
|
|
break;
|
|
|
|
case AttributeUnderlyingType::utDecimal32:
|
|
addAttributeSize<Decimal32>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utDecimal64:
|
|
addAttributeSize<Decimal64>(attribute);
|
|
break;
|
|
case AttributeUnderlyingType::utDecimal128:
|
|
addAttributeSize<Decimal128>(attribute);
|
|
break;
|
|
|
|
case AttributeUnderlyingType::utString:
|
|
{
|
|
addAttributeSize<StringRef>(attribute);
|
|
bytes_allocated += sizeof(Arena) + attribute.string_arena->size();
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
template <typename T>
|
|
void HashedDictionary::createAttributeImpl(Attribute & attribute, const Field & null_value)
|
|
{
|
|
attribute.null_values = T(null_value.get<NearestFieldType<T>>());
|
|
if (!sparse)
|
|
attribute.maps = std::make_unique<CollectionType<T>>();
|
|
else
|
|
attribute.sparse_maps = std::make_unique<SparseCollectionType<T>>();
|
|
}
|
|
|
|
HashedDictionary::Attribute HashedDictionary::createAttributeWithType(const AttributeUnderlyingType type, const Field & null_value)
|
|
{
|
|
Attribute attr{type, {}, {}, {}, {}};
|
|
|
|
switch (type)
|
|
{
|
|
case AttributeUnderlyingType::utUInt8:
|
|
createAttributeImpl<UInt8>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt16:
|
|
createAttributeImpl<UInt16>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt32:
|
|
createAttributeImpl<UInt32>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt64:
|
|
createAttributeImpl<UInt64>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utUInt128:
|
|
createAttributeImpl<UInt128>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utInt8:
|
|
createAttributeImpl<Int8>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utInt16:
|
|
createAttributeImpl<Int16>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utInt32:
|
|
createAttributeImpl<Int32>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utInt64:
|
|
createAttributeImpl<Int64>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utFloat32:
|
|
createAttributeImpl<Float32>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utFloat64:
|
|
createAttributeImpl<Float64>(attr, null_value);
|
|
break;
|
|
|
|
case AttributeUnderlyingType::utDecimal32:
|
|
createAttributeImpl<Decimal32>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utDecimal64:
|
|
createAttributeImpl<Decimal64>(attr, null_value);
|
|
break;
|
|
case AttributeUnderlyingType::utDecimal128:
|
|
createAttributeImpl<Decimal128>(attr, null_value);
|
|
break;
|
|
|
|
case AttributeUnderlyingType::utString:
|
|
{
|
|
attr.null_values = null_value.get<String>();
|
|
if (!sparse)
|
|
attr.maps = std::make_unique<CollectionType<StringRef>>();
|
|
else
|
|
attr.sparse_maps = std::make_unique<SparseCollectionType<StringRef>>();
|
|
attr.string_arena = std::make_unique<Arena>();
|
|
break;
|
|
}
|
|
}
|
|
|
|
return attr;
|
|
}
|
|
|
|
|
|
template <typename OutputType, typename AttrType, typename ValueSetter, typename DefaultGetter>
|
|
void HashedDictionary::getItemsAttrImpl(
|
|
const AttrType & attr, const PaddedPODArray<Key> & ids, ValueSetter && set_value, DefaultGetter && get_default) const
|
|
{
|
|
const auto rows = ext::size(ids);
|
|
|
|
for (const auto i : ext::range(0, rows))
|
|
{
|
|
const auto it = attr.find(ids[i]);
|
|
set_value(i, it != attr.end() ? static_cast<OutputType>(second(*it)) : get_default(i));
|
|
}
|
|
|
|
query_count.fetch_add(rows, std::memory_order_relaxed);
|
|
}
|
|
template <typename AttributeType, typename OutputType, typename ValueSetter, typename DefaultGetter>
|
|
void HashedDictionary::getItemsImpl(
|
|
const Attribute & attribute, const PaddedPODArray<Key> & ids, ValueSetter && set_value, DefaultGetter && get_default) const
|
|
{
|
|
if (!sparse)
|
|
return getItemsAttrImpl<OutputType>(*std::get<CollectionPtrType<AttributeType>>(attribute.maps), ids, set_value, get_default);
|
|
return getItemsAttrImpl<OutputType>(*std::get<SparseCollectionPtrType<AttributeType>>(attribute.sparse_maps), ids, set_value, get_default);
|
|
}
|
|
|
|
|
|
template <typename T>
|
|
bool HashedDictionary::setAttributeValueImpl(Attribute & attribute, const Key id, const T value)
|
|
{
|
|
if (!sparse)
|
|
{
|
|
auto & map = *std::get<CollectionPtrType<T>>(attribute.maps);
|
|
return map.insert({id, value}).second;
|
|
}
|
|
else
|
|
{
|
|
auto & map = *std::get<SparseCollectionPtrType<T>>(attribute.sparse_maps);
|
|
return map.insert({id, value}).second;
|
|
}
|
|
}
|
|
|
|
bool HashedDictionary::setAttributeValue(Attribute & attribute, const Key id, const Field & value)
|
|
{
|
|
switch (attribute.type)
|
|
{
|
|
case AttributeUnderlyingType::utUInt8:
|
|
return setAttributeValueImpl<UInt8>(attribute, id, value.get<UInt64>());
|
|
case AttributeUnderlyingType::utUInt16:
|
|
return setAttributeValueImpl<UInt16>(attribute, id, value.get<UInt64>());
|
|
case AttributeUnderlyingType::utUInt32:
|
|
return setAttributeValueImpl<UInt32>(attribute, id, value.get<UInt64>());
|
|
case AttributeUnderlyingType::utUInt64:
|
|
return setAttributeValueImpl<UInt64>(attribute, id, value.get<UInt64>());
|
|
case AttributeUnderlyingType::utUInt128:
|
|
return setAttributeValueImpl<UInt128>(attribute, id, value.get<UInt128>());
|
|
case AttributeUnderlyingType::utInt8:
|
|
return setAttributeValueImpl<Int8>(attribute, id, value.get<Int64>());
|
|
case AttributeUnderlyingType::utInt16:
|
|
return setAttributeValueImpl<Int16>(attribute, id, value.get<Int64>());
|
|
case AttributeUnderlyingType::utInt32:
|
|
return setAttributeValueImpl<Int32>(attribute, id, value.get<Int64>());
|
|
case AttributeUnderlyingType::utInt64:
|
|
return setAttributeValueImpl<Int64>(attribute, id, value.get<Int64>());
|
|
case AttributeUnderlyingType::utFloat32:
|
|
return setAttributeValueImpl<Float32>(attribute, id, value.get<Float64>());
|
|
case AttributeUnderlyingType::utFloat64:
|
|
return setAttributeValueImpl<Float64>(attribute, id, value.get<Float64>());
|
|
|
|
case AttributeUnderlyingType::utDecimal32:
|
|
return setAttributeValueImpl<Decimal32>(attribute, id, value.get<Decimal32>());
|
|
case AttributeUnderlyingType::utDecimal64:
|
|
return setAttributeValueImpl<Decimal64>(attribute, id, value.get<Decimal64>());
|
|
case AttributeUnderlyingType::utDecimal128:
|
|
return setAttributeValueImpl<Decimal128>(attribute, id, value.get<Decimal128>());
|
|
|
|
case AttributeUnderlyingType::utString:
|
|
{
|
|
const auto & string = value.get<String>();
|
|
const auto string_in_arena = attribute.string_arena->insert(string.data(), string.size());
|
|
if (!sparse)
|
|
{
|
|
auto & map = *std::get<CollectionPtrType<StringRef>>(attribute.maps);
|
|
return map.insert({id, StringRef{string_in_arena, string.size()}}).second;
|
|
}
|
|
else
|
|
{
|
|
auto & map = *std::get<SparseCollectionPtrType<StringRef>>(attribute.sparse_maps);
|
|
return map.insert({id, StringRef{string_in_arena, string.size()}}).second;
|
|
}
|
|
}
|
|
}
|
|
|
|
throw Exception{"Invalid attribute type", ErrorCodes::BAD_ARGUMENTS};
|
|
}
|
|
|
|
const HashedDictionary::Attribute & HashedDictionary::getAttribute(const std::string & attribute_name) const
|
|
{
|
|
const auto it = attribute_index_by_name.find(attribute_name);
|
|
if (it == std::end(attribute_index_by_name))
|
|
throw Exception{full_name + ": no such attribute '" + attribute_name + "'", ErrorCodes::BAD_ARGUMENTS};
|
|
|
|
return attributes[it->second];
|
|
}
|
|
|
|
template <typename T>
|
|
void HashedDictionary::has(const Attribute & attribute, const PaddedPODArray<Key> & ids, PaddedPODArray<UInt8> & out) const
|
|
{
|
|
const auto & attr = *std::get<CollectionPtrType<T>>(attribute.maps);
|
|
const auto rows = ext::size(ids);
|
|
|
|
for (const auto i : ext::range(0, rows))
|
|
out[i] = attr.find(ids[i]) != nullptr;
|
|
|
|
query_count.fetch_add(rows, std::memory_order_relaxed);
|
|
}
|
|
|
|
template <typename T, typename AttrType>
|
|
PaddedPODArray<HashedDictionary::Key> HashedDictionary::getIdsAttrImpl(const AttrType & attr) const
|
|
{
|
|
PaddedPODArray<Key> ids;
|
|
ids.reserve(attr.size());
|
|
for (const auto & value : attr)
|
|
ids.push_back(first(value));
|
|
|
|
return ids;
|
|
}
|
|
template <typename T>
|
|
PaddedPODArray<HashedDictionary::Key> HashedDictionary::getIds(const Attribute & attribute) const
|
|
{
|
|
if (!sparse)
|
|
return getIdsAttrImpl<T>(*std::get<CollectionPtrType<T>>(attribute.maps));
|
|
return getIdsAttrImpl<T>(*std::get<SparseCollectionPtrType<T>>(attribute.sparse_maps));
|
|
}
|
|
|
|
PaddedPODArray<HashedDictionary::Key> HashedDictionary::getIds() const
|
|
{
|
|
const auto & attribute = attributes.front();
|
|
|
|
switch (attribute.type)
|
|
{
|
|
case AttributeUnderlyingType::utUInt8:
|
|
return getIds<UInt8>(attribute);
|
|
case AttributeUnderlyingType::utUInt16:
|
|
return getIds<UInt16>(attribute);
|
|
case AttributeUnderlyingType::utUInt32:
|
|
return getIds<UInt32>(attribute);
|
|
case AttributeUnderlyingType::utUInt64:
|
|
return getIds<UInt64>(attribute);
|
|
case AttributeUnderlyingType::utUInt128:
|
|
return getIds<UInt128>(attribute);
|
|
case AttributeUnderlyingType::utInt8:
|
|
return getIds<Int8>(attribute);
|
|
case AttributeUnderlyingType::utInt16:
|
|
return getIds<Int16>(attribute);
|
|
case AttributeUnderlyingType::utInt32:
|
|
return getIds<Int32>(attribute);
|
|
case AttributeUnderlyingType::utInt64:
|
|
return getIds<Int64>(attribute);
|
|
case AttributeUnderlyingType::utFloat32:
|
|
return getIds<Float32>(attribute);
|
|
case AttributeUnderlyingType::utFloat64:
|
|
return getIds<Float64>(attribute);
|
|
case AttributeUnderlyingType::utString:
|
|
return getIds<StringRef>(attribute);
|
|
|
|
case AttributeUnderlyingType::utDecimal32:
|
|
return getIds<Decimal32>(attribute);
|
|
case AttributeUnderlyingType::utDecimal64:
|
|
return getIds<Decimal64>(attribute);
|
|
case AttributeUnderlyingType::utDecimal128:
|
|
return getIds<Decimal128>(attribute);
|
|
}
|
|
return PaddedPODArray<Key>();
|
|
}
|
|
|
|
BlockInputStreamPtr HashedDictionary::getBlockInputStream(const Names & column_names, size_t max_block_size) const
|
|
{
|
|
using BlockInputStreamType = DictionaryBlockInputStream<HashedDictionary, Key>;
|
|
return std::make_shared<BlockInputStreamType>(shared_from_this(), max_block_size, getIds(), column_names);
|
|
}
|
|
|
|
void registerDictionaryHashed(DictionaryFactory & factory)
|
|
{
|
|
auto create_layout = [=](const std::string & full_name,
|
|
const DictionaryStructure & dict_struct,
|
|
const Poco::Util::AbstractConfiguration & config,
|
|
const std::string & config_prefix,
|
|
DictionarySourcePtr source_ptr,
|
|
bool sparse) -> DictionaryPtr
|
|
{
|
|
if (dict_struct.key)
|
|
throw Exception{"'key' is not supported for dictionary of layout 'hashed'", ErrorCodes::UNSUPPORTED_METHOD};
|
|
|
|
if (dict_struct.range_min || dict_struct.range_max)
|
|
throw Exception{full_name
|
|
+ ": elements .structure.range_min and .structure.range_max should be defined only "
|
|
"for a dictionary of layout 'range_hashed'",
|
|
ErrorCodes::BAD_ARGUMENTS};
|
|
|
|
const String database = config.getString(config_prefix + ".database", "");
|
|
const String name = config.getString(config_prefix + ".name");
|
|
const DictionaryLifetime dict_lifetime{config, config_prefix + ".lifetime"};
|
|
const bool require_nonempty = config.getBool(config_prefix + ".require_nonempty", false);
|
|
return std::make_unique<HashedDictionary>(database, name, dict_struct, std::move(source_ptr), dict_lifetime, require_nonempty, sparse);
|
|
};
|
|
using namespace std::placeholders;
|
|
factory.registerLayout("hashed", std::bind(create_layout, _1, _2, _3, _4, _5, /* sparse = */ false), false);
|
|
factory.registerLayout("sparse_hashed", std::bind(create_layout, _1, _2, _3, _4, _5, /* sparse = */ true), false);
|
|
}
|
|
|
|
}
|