Method isArraySetCheck

src/Storages/MergeTree/MergeTreeWhereOptimizer.cpp:298–355

A expression is arraySetCheck if 1. single arraySetCheck function with column argument is a BLOOM column 2. not arraySetCheck function

// 1. single arraySetCheck function with column argument is a BLOOM column
// 2. not arraySetCheck function
bool MergeTreeWhereOptimizer::isArraySetCheck(const ASTPtr & condition, bool) const
{
    if (!condition)
        return false;

    const auto * const function = typeid_cast<const ASTFunction *>(condition.get());

    if (function)
    {
        if (BitmapIndexHelper::isArraySetFunctions(function->name))
        {
            size_t arg_size = function->arguments->children.size();
            if (arg_size % 2)
                throw Exception("Wrong number of arguments of arraySetCheck", ErrorCodes::LOGICAL_ERROR);

            for (size_t i = 0; i < arg_size; i += 2)
            {
                auto * left_arg = function->arguments->children.at(i).get();
                auto * right_arg = function->arguments->children.at(i + 1).get();
                auto * identifier = left_arg->as<ASTIdentifier>();
                if (!identifier || right_arg->as<ASTIdentifier>())
                    return false;

                String identifier_name = identifier->getColumnName();
                if (isMapImplicitKey(identifier_name))
                    identifier_name = parseMapNameFromImplicitFileName(identifier_name);

                auto columns = metadata_snapshot->getColumns();
                if (!columns.has(identifier_name))
                    return false;

                // unlikely
                {
                    // Cannot handle column like 'map.key'
                    auto [maybe_reserved_map_keys, name_without_suffix] = mayBeMapKVReservedKeys(identifier_name);
                    if (maybe_reserved_map_keys)
                    {
                        if (!columns.has(name_without_suffix))
                            return false;

                        const ColumnDescription & column = columns.get(name_without_suffix);
                        if (column.type->isMap())
                            return false;
                    }
                }

                const ColumnDescription & column = columns.get(identifier_name);

                if (!column.type->isBitmapIndex() && !column.type->isSegmentBitmapIndex())
                    return false;
            }

            return true;
        }
    }

    return false;
}