Function createStructArray

test/include/arrow_test_utils.h:576–609

Build a struct ArrowArray whose children are filled by the given builders.

// Build a struct ArrowArray whose children are filled by the given builders.
inline ArrowArrayWrapper createStructArray(int64_t length,
    const std::vector<std::function<void(ArrowArray*)>>& childBuilders) {
    ArrowArrayWrapper array;
    array.length = length;
    array.null_count = 0;
    array.offset = 0;
    array.n_buffers = 1;
    array.n_children = static_cast<int64_t>(childBuilders.size());
    array.buffers = static_cast<const void**>(malloc(sizeof(void*)));
    array.buffers[0] = nullptr;
    array.children = static_cast<ArrowArray**>(malloc(sizeof(ArrowArray*) * childBuilders.size()));
    for (size_t i = 0; i < childBuilders.size(); ++i) {
        array.children[i] = static_cast<ArrowArray*>(malloc(sizeof(ArrowArray)));
        childBuilders[i](array.children[i]);
    }
    array.dictionary = nullptr;
    array.release = [](ArrowArray* arr) {
        if (arr->children) {
            for (int64_t i = 0; i < arr->n_children; ++i) {
                if (arr->children[i]->release) {
                    arr->children[i]->release(arr->children[i]);
                }
                free(arr->children[i]);
            }
            free(arr->children);
        }
        if (arr->buffers) {
            free(const_cast<void**>(arr->buffers));
        }
        arr->release = nullptr;
    };
    array.private_data = nullptr;
    return array;
}