Function reserve_range_capacity

datafusion/functions-nested/src/range.rs:544–555

Reserve space for `count` more elements, returning an error when the allocation would overflow `Vec`'s capacity limit or the allocator rejects it, rather than panicking on user-supplied SQL.

(values: &mut Vec<i64>, count: u64)

/// allocation would overflow `Vec`'s capacity limit or the allocator
/// rejects it, rather than panicking on user-supplied SQL.
fn reserve_range_capacity(values: &mut Vec<i64>, count: u64) -> Result<()> {
    let count_usize = usize::try_from(count).map_err(|_| {
        exec_datafusion_err!(
            "Range too large to materialize: would produce {count} elements"
        )
    })?;
    values.try_reserve(count_usize).map_err(|e| {
        exec_datafusion_err!(
            "Range too large to materialize: failed to allocate {count} elements: {e}"
        )
    })
}

/// Generate integer range values directly into the provided buffer.
#[inline]