Method setGetNumberImpl

src/Functions/array/arraySetGet.cpp:184–220

    template <bool Nullable, typename Method, typename DataType>
    inline void setGetNumberImpl(Method & method, const ColumnArray & array, ColumnArray & result) const
    {
        ConstNullMapPtr null_map{};
        ColumnRawPtrs key_cols;
        key_cols.push_back(&array.getData());
        if(Nullable) extractNestedColumnsAndNullMap(key_cols, null_map);

        if (auto * result_column = typeid_cast<ColumnVector<DataType> *>(&result.getData()))
        {
            typename Method::State state(key_cols, {}, nullptr);
            typename ColumnVector<DataType>::Container & result_container = result_column->getData();
            size_t rsize = array.size();
            const auto & offsets = array.getOffsets();
            auto & res_offsets = result.getOffsets();
            size_t pre_offset = 0, cur_offset = 0;

            Arena arena(0);
            for (size_t i = 0; i<rsize; ++i)
            {
                // get current elems in this array input
                cur_offset = offsets[i];
                size_t element_size = 0;
                for (size_t j = pre_offset; j < cur_offset; ++j)
                {
                    if (Nullable && (*null_map)[j]) continue;
                    auto key_holder = state.getKeyHolder(j, arena);
                    if (method.data.has(key_holder))
                    {
                        result_container.push_back(key_holder);
                        element_size++;
                    }
                }
                res_offsets.push_back(res_offsets.back() + element_size);
                pre_offset = offsets[i];
            }
        }
    }

};