Function generate_range_values

datafusion/functions-nested/src/range.rs:559–610

(
    start: i64,
    stop: i64,
    step: i64,
    include_upper: bool,
    values: &mut Vec<i64>,
)

/// Generate integer range values directly into the provided buffer.
#[inline]
fn generate_range_values(
    start: i64,
    stop: i64,
    step: i64,
    include_upper: bool,
    values: &mut Vec<i64>,
) -> Result<()> {
    if !include_upper && start == stop {
        return Ok(());
    }

    if step > 0 {
        let limit = if include_upper {
            stop
        } else {
            stop.saturating_sub(1)
        };
        if start > limit {
            return Ok(());
        }
        let count = (start.abs_diff(limit) / step.unsigned_abs()).saturating_add(1);
        reserve_range_capacity(values, count)?;
        let mut current = start;
        while current <= limit {
            values.push(current);
            match current.checked_add(step) {
                Some(next) => current = next,
                None => break,
            }
        }
    } else if step < 0 {
        let limit = if include_upper {
            stop
        } else {
            stop.saturating_add(1)
        };
        if start < limit {
            return Ok(());
        }
        let count = (start.abs_diff(limit) / step.unsigned_abs()).saturating_add(1);
        reserve_range_capacity(values, count)?;
        let mut current = start;
        while current >= limit {
            values.push(current);
            match current.checked_add(step) {
                Some(next) => current = next,
                None => break,
            }
        }
    }
    Ok(())
}

fn parse_tz(tz: &Option<&str>) -> Result<Tz> {
    let tz = tz.unwrap_or_else(|| "+00");