Method hasKeys

src/Dictionaries/IPAddressDictionary.cpp:370–414

}

ColumnUInt8::Ptr IPAddressDictionary::hasKeys(const Columns & key_columns, const DataTypes & key_types) const
{
    validateKeyTypes(key_types);

    const auto & first_column = key_columns.front();
    const size_t rows = first_column->size();

    auto result = ColumnUInt8::create(rows);
    auto & out = result->getData();

    size_t keys_found = 0;

    TypeIndex type_id = first_column->getDataType();

    if (type_id == TypeIndex::IPv4 || type_id == TypeIndex::UInt32)
    {
        uint8_t addrv6_buf[IPV6_BINARY_LENGTH];
        for (const auto i : collections::range(0, rows))
        {
            auto addrv4 = *reinterpret_cast<const UInt32 *>(first_column->getDataAt(i).data());
            auto found = tryLookupIPv4(addrv4, addrv6_buf);
            out[i] = (found != ipNotFound());
            keys_found += out[i];
        }
    }
    else if (type_id == TypeIndex::IPv6 || type_id == TypeIndex::FixedString)
    {
        for (const auto i : collections::range(0, rows))
        {
            auto addr = first_column->getDataAt(i);
            if (addr.size() != IPV6_BINARY_LENGTH)
                throw Exception(ErrorCodes::TYPE_MISMATCH, "Expected key FixedString(16)");
            auto found = tryLookupIPv6(reinterpret_cast<const uint8_t *>(addr.data()));
            out[i] = (found != ipNotFound());
            keys_found += out[i];
        }
    }
    else
        throw Exception(ErrorCodes::TYPE_MISMATCH, "Expected key to be IPv4 (or UInt32) or IPv6 (or FixedString(16))");

    query_count.fetch_add(rows, std::memory_order_relaxed);
    found_count.fetch_add(keys_found, std::memory_order_relaxed);

    return result;
}

void IPAddressDictionary::createAttributes()
{