Method setCheckImpl

src/Functions/array/arraySetCheck.cpp:185–226

    template <typename Method, bool Nullable, bool StringType = false>
    inline void setCheckImpl(Method& method,
                             const ColumnArray& keys,
                             ColumnUInt8::Container& result,
                             bool first) const
    {
        size_t rsize = result.size();
        ConstNullMapPtr null_map{};

        ColumnRawPtrs key_cols;
        key_cols.push_back(&keys.getData());

        if(Nullable) extractNestedColumnsAndNullMap(key_cols, null_map);

        Sizes key_sizes;
        ClearableSetVariants::chooseMethod(key_cols, key_sizes);
        typename Method::State state(key_cols, key_sizes, nullptr);
        size_t pre_offset = 0, cur_offset = 0;
        auto& offsets = keys.getOffsets();
        Arena arena(0);
        for (size_t i = 0; i<rsize; ++i)
        {
            // get current elems in this array input
            cur_offset = offsets[i];

            if (first || result[i])
            {
                bool has_key = false;
                for (size_t j = pre_offset; (!has_key) && j < cur_offset; ++j)
                {
                    if (Nullable && (*null_map)[j]) continue; // skip null input
                    auto key_holder = state.getKeyHolder(j, arena);
                    if constexpr (StringType)
                        has_key = method.data.has(key_holder.key);
                    else
                        has_key = method.data.has(key_holder);
                }
                result[i] = has_key;
            }

            pre_offset = offsets[i];
        }
    }

};