| 240 | } |
| 241 | |
| 242 | Result<std::shared_ptr<RecordBatch>> ExecBatch::ToRecordBatch( |
| 243 | std::shared_ptr<Schema> schema, MemoryPool* pool) const { |
| 244 | if (static_cast<size_t>(schema->num_fields()) > values.size()) { |
| 245 | return Status::Invalid("ExecBatch::ToRecordBatch mismatching schema size"); |
| 246 | } |
| 247 | ArrayVector columns(schema->num_fields()); |
| 248 | |
| 249 | for (size_t i = 0; i < columns.size(); ++i) { |
| 250 | const Datum& value = values[i]; |
| 251 | if (value.is_array()) { |
| 252 | columns[i] = value.make_array(); |
| 253 | continue; |
| 254 | } else if (value.is_scalar()) { |
| 255 | ARROW_ASSIGN_OR_RAISE(columns[i], |
| 256 | MakeArrayFromScalar(*value.scalar(), length, pool)); |
| 257 | } else { |
| 258 | return Status::TypeError("ExecBatch::ToRecordBatch value ", i, " with unsupported ", |
| 259 | "value kind ", ::arrow::ToString(value.kind())); |
| 260 | } |
| 261 | } |
| 262 | |
| 263 | return RecordBatch::Make(std::move(schema), length, std::move(columns)); |
| 264 | } |
| 265 | |
| 266 | namespace { |
| 267 | |