Method to_record_batch

nodedb/src/engine/graph/algo/result.rs:131–149

Convert to an Arrow `RecordBatch`. Consumes the batch. Returns `Err` if column lengths are inconsistent (should never happen with correct `push_*` usage).

(self)

    /// Consumes the batch. Returns `Err` if column lengths are inconsistent
    /// (should never happen with correct `push_*` usage).
    pub fn to_record_batch(self) -> Result<RecordBatch, crate::Error> {
        let schema = Arc::new(self.arrow_schema());
        let mut columns: Vec<ArrayRef> = Vec::with_capacity(self.column_map.len());

        for &(col_type, idx) in &self.column_map {
            let array: ArrayRef = match col_type {
                AlgoColumnType::Text => Arc::new(StringArray::from(self.text_columns[idx].clone())),
                AlgoColumnType::Float64 => {
                    Arc::new(Float64Array::from(self.f64_columns[idx].clone()))
                }
                AlgoColumnType::Int64 => Arc::new(Int64Array::from(self.i64_columns[idx].clone())),
            };
            columns.push(array);
        }

        RecordBatch::try_new(schema, columns).map_err(|e| crate::Error::Internal {
            detail: format!("arrow result batch: {e}"),
        })
    }

    /// Serialize the result batch as MessagePack bytes.
    ///