Function vortex_array_to_record_batch

crates/paimon/src/arrow/format/vortex.rs:470–506

Convert a Vortex ArrayRef to an Arrow RecordBatch.

(
    vortex_array: ArrayRef,
    schema: &SchemaRef,
)

/// Convert a Vortex ArrayRef to an Arrow RecordBatch.
fn vortex_array_to_record_batch(
    vortex_array: ArrayRef,
    schema: &SchemaRef,
) -> crate::Result<RecordBatch> {
    let arrow_array = vortex_array
        .into_arrow(&ArrowDataType::Struct(schema.fields().clone()))
        .map_err(|e| Error::DataInvalid {
            message: format!("Failed to convert Vortex array to Arrow: {e}"),
            source: None,
        })?;

    let struct_array = arrow_array
        .as_any()
        .downcast_ref::<arrow_array::StructArray>()
        .ok_or_else(|| Error::DataInvalid {
            message: "Vortex array did not convert to Arrow StructArray".to_string(),
            source: None,
        })?;

    if struct_array.columns().len() != schema.fields().len() {
        return Err(Error::DataInvalid {
            message: format!(
                "Vortex column count {} does not match target schema column count {}",
                struct_array.columns().len(),
                schema.fields().len()
            ),
            source: None,
        });
    }

    RecordBatch::try_new(schema.clone(), struct_array.columns().to_vec()).map_err(|e| {
        Error::DataInvalid {
            message: format!("Failed to build RecordBatch from Vortex data: {e}"),
            source: None,
        }
    })
}

// ---------------------------------------------------------------------------
// VortexWrite adapter