rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-18 14:00:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add impl of arrow array access

Browse Source
This commit is contained in:
liangxingjian
2022-10-19 10:25:58 +08:00
parent 35afa9dc74
commit 3bf2c9840d
2 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/datatypes/src/arrow_array.rs
View File

@@ -8,7 +8,7 @@ use snafu::OptionExt;
use crate::error::{ConversionSnafu, Result};
use crate::prelude::ConcreteDataType;
use crate::value::Value;
use crate::value::{GeometryValue, Value};
pub type BinaryArray = ArrowBinaryArray<i64>;
pub type MutableBinaryArray = ArrowMutableBinaryArray<i64>;
@@ -70,8 +70,26 @@ pub fn arrow_array_get(array: &dyn Array, idx: usize) -> Result<Value> {
Value::Timestamp(Timestamp::new(value, unit))
}
ArrowDataType::Struct(_) => {
let k = cast_array!(array, StructArray).values().get(0).unwrap();
Value::Geometry(crate::value::GeometryValue::new_point(0.into(), 0.into()))
let struct_array = array.as_any().downcast_ref::<StructArray>().unwrap();
let ref_x_array = struct_array
.values()
.get(0)
.unwrap()
.as_any()
.downcast_ref::<PrimitiveArray<f64>>()
.unwrap();
let ref_y_array = struct_array
.values()
.get(1)
.unwrap()
.as_any()
.downcast_ref::<PrimitiveArray<f64>>()
.unwrap();
let (x, y) = (ref_x_array.value(idx), ref_y_array.value(idx));
Value::Geometry(GeometryValue::new_point(x, y))
}
// TODO(sunng87): List
_ => unimplemented!("Arrow array datatype: {:?}", array.data_type()),

4
src/datatypes/src/vectors/operations.rs
View File

@@ -130,9 +130,9 @@ impl VectorOp for GeometryVector {
todo!()
}
fn filter(&self, filter: &BooleanVector) -> Result<VectorRef> {
fn filter(&self, _filter: &BooleanVector) -> Result<VectorRef> {
let array = self.to_arrow_array();
let v = GeometryVector::try_from_arrow_array(array)?;
return Ok(Arc::new(v));
Ok(Arc::new(v))
}
}